From e06025366aca9a3fff12d239f01bca09db8a69f5 Mon Sep 17 00:00:00 2001 From: richard-dds Date: Mon, 8 Oct 2018 10:35:36 -0400 Subject: [PATCH] Allow RequestFactory to generate task order --- tests/factories.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/factories.py b/tests/factories.py index b8ab9b47..6844006d 100644 --- a/tests/factories.py +++ b/tests/factories.py @@ -100,11 +100,16 @@ class RequestFactory(Base): new_status=RequestStatus.STARTED, revision=factory.LazyAttribute(lambda se: se.factory_parent.revisions[-1]), ) - task_order = factory.SubFactory("tests.factories.TaskOrderFactory") class Params: 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 def create_initial_status_event(cls, request): return RequestStatusEventFactory(