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

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