use user factory for testing

This commit is contained in:
dandds 2018-08-06 14:52:23 -04:00
parent 2fd34194b3
commit 61270fbb40
4 changed files with 25 additions and 20 deletions

View File

@ -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

View File

@ -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)

View File

@ -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",

View File

@ -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()