atst/tests/factories.py
2018-08-06 10:44:00 -04:00

22 lines
452 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
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