41 lines
865 B
Python
41 lines
865 B
Python
import factory
|
|
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):
|
|
class Meta:
|
|
model = Request
|
|
|
|
id = factory.Sequence(lambda x: uuid4())
|
|
|
|
class PENumberFactory(factory.Factory):
|
|
class Meta:
|
|
model = PENumber
|
|
|
|
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)
|
|
|