Create CLINs when creating a TO

This commit is contained in:
leigh-mil
2019-06-04 11:42:14 -04:00
parent 633e1b6a37
commit 36200e01ad
8 changed files with 161 additions and 33 deletions

View File

@@ -41,6 +41,10 @@ def random_future_date(year_min=1, year_max=5):
return _random_date(year_min, year_max, operator.add)
def random_loa_numbers():
return ["".join(random.choices(string.digits, k=43))]
def _random_date(year_min, year_max, operation):
if year_min == year_max:
inc = year_min
@@ -267,6 +271,17 @@ class TaskOrderFactory(Base):
portfolio = factory.SubFactory(PortfolioFactory)
number = factory.LazyFunction(random_task_order_number)
creator = factory.SubFactory(UserFactory)
@classmethod
def _create(cls, model_class, *args, **kwargs):
with_clins = kwargs.pop("clins", [])
task_order = super()._create(model_class, *args, **kwargs)
for clin in with_clins:
CLINFactory.create(task_order=task_order, number=clin)
return task_order
class CLINFactory(Base):