use user factory for testing
This commit is contained in:
parent
2fd34194b3
commit
61270fbb40
@ -124,5 +124,5 @@ class JEDIRequestFlow(object):
|
||||
if self.request_id:
|
||||
Requests.update(self.request_id, request_data)
|
||||
else:
|
||||
request = Requests.create(self.current_user["id"], request_data)
|
||||
request = Requests.create(self.current_user.id, request_data)
|
||||
self.request_id = request.id
|
||||
|
@ -4,6 +4,8 @@ from uuid import uuid4
|
||||
from atst.models import Request
|
||||
from atst.models.pe_number import PENumber
|
||||
from atst.models.task_order import TaskOrder
|
||||
from atst.models.user import User
|
||||
from atst.models.role import Role
|
||||
|
||||
|
||||
class RequestFactory(factory.Factory):
|
||||
@ -19,3 +21,20 @@ class PENumberFactory(factory.Factory):
|
||||
class TaskOrderFactory(factory.Factory):
|
||||
class Meta:
|
||||
model = TaskOrder
|
||||
|
||||
class RoleFactory(factory.Factory):
|
||||
class Meta:
|
||||
model = Role
|
||||
|
||||
permissions = []
|
||||
|
||||
class UserFactory(factory.Factory):
|
||||
class Meta:
|
||||
model = User
|
||||
|
||||
id = factory.Sequence(lambda x: uuid4())
|
||||
email = "fake.user@mail.com"
|
||||
first_name = "Fake"
|
||||
last_name = "User"
|
||||
atat_role = factory.SubFactory(RoleFactory)
|
||||
|
||||
|
@ -2,18 +2,12 @@ import tornado.gen
|
||||
from tornado.httpclient import HTTPRequest, HTTPResponse
|
||||
|
||||
from atst.api_client import ApiClient
|
||||
from tests.factories import RequestFactory
|
||||
from tests.factories import RequestFactory, UserFactory
|
||||
|
||||
|
||||
MOCK_USER = {
|
||||
"id": "9cb348f0-8102-4962-88c4-dac8180c904c",
|
||||
"email": "fake.user@mail.com",
|
||||
"first_name": "Fake",
|
||||
"last_name": "User",
|
||||
"atat_permissions": []
|
||||
}
|
||||
MOCK_USER = UserFactory.create()
|
||||
MOCK_REQUEST = RequestFactory.create(
|
||||
creator=MOCK_USER["id"],
|
||||
creator=MOCK_USER.id,
|
||||
body={
|
||||
"financial_verification": {
|
||||
"pe_id": "0203752A",
|
||||
|
@ -1,19 +1,11 @@
|
||||
import re
|
||||
import pytest
|
||||
import urllib
|
||||
from tests.mocks import MOCK_USER
|
||||
from tests.mocks import MOCK_USER, MOCK_REQUEST
|
||||
from tests.factories import RequestFactory
|
||||
|
||||
ERROR_CLASS = "alert--error"
|
||||
MOCK_REQUEST = RequestFactory.create(
|
||||
creator=MOCK_USER["id"],
|
||||
body={
|
||||
"financial_verification": {
|
||||
"pe_id": "0203752A",
|
||||
},
|
||||
}
|
||||
)
|
||||
|
||||
ERROR_CLASS = "alert--error"
|
||||
|
||||
def test_submit_invalid_request_form(monkeypatch, client, user_session):
|
||||
user_session()
|
||||
|
Loading…
x
Reference in New Issue
Block a user