pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow RequestFactory to generate task order

Browse Source
This commit is contained in:
richard-dds 2018-10-08 10:35:36 -04:00
parent 9061c3515c
commit e06025366a
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/factories.py
View File

@ -100,11 +100,16 @@ class RequestFactory(Base):
new_status=RequestStatus.STARTED, new_status=RequestStatus.STARTED,
revision=factory.LazyAttribute(lambda se: se.factory_parent.revisions[-1]), revision=factory.LazyAttribute(lambda se: se.factory_parent.revisions[-1]),
) )
task_order = factory.SubFactory("tests.factories.TaskOrderFactory")
class Params: class Params:
initial_revision = None initial_revision = None
@classmethod
def _adjust_kwargs(cls, **kwargs):
if kwargs.pop("with_task_order", False) and "task_order" not in kwargs:
kwargs["task_order"] = TaskOrderFactory.build()
return kwargs
@classmethod @classmethod
def create_initial_status_event(cls, request): def create_initial_status_event(cls, request):
return RequestStatusEventFactory( return RequestStatusEventFactory(