diff --git a/script/seed_sample.py b/script/seed_sample.py index 8fced973..2ed03d90 100644 --- a/script/seed_sample.py +++ b/script/seed_sample.py @@ -164,6 +164,7 @@ def add_task_orders_to_portfolio(portfolio): today = date.today() future = today + timedelta(days=100) yesterday = today - timedelta(days=1) + five_days = timedelta(days=5) draft_to = TaskOrderFactory.build(portfolio=portfolio, pdf=None) unsigned_to = TaskOrderFactory.build(portfolio=portfolio) @@ -172,10 +173,14 @@ def add_task_orders_to_portfolio(portfolio): active_to = TaskOrderFactory.build(portfolio=portfolio, signed_at=yesterday) clins = [ - CLINFactory.build(task_order=unsigned_to, start_date=today, end_date=today), - CLINFactory.build(task_order=upcoming_to, start_date=future, end_date=future), CLINFactory.build( - task_order=expired_to, start_date=yesterday, end_date=yesterday + task_order=unsigned_to, start_date=(today - five_days), end_date=today + ), + CLINFactory.build( + task_order=upcoming_to, start_date=future, end_date=(today + five_days) + ), + CLINFactory.build( + task_order=expired_to, start_date=(today - five_days), end_date=yesterday ), CLINFactory.build(task_order=active_to, start_date=yesterday, end_date=future), ] diff --git a/tests/factories.py b/tests/factories.py index 3bbee856..f5d04a12 100644 --- a/tests/factories.py +++ b/tests/factories.py @@ -30,7 +30,7 @@ def random_phone_number(): def random_task_order_number(): - return "-".join([str(random.randint(100, 999)) for _ in range(4)]) + return "".join(random.choices(string.digits, k=10)) def random_past_date(year_min=1, year_max=5): @@ -261,7 +261,7 @@ class AttachmentFactory(Base): class Meta: model = Attachment - filename = factory.Faker("domain_word") + filename = factory.Faker("file_name", extension="pdf") object_name = factory.LazyFunction(lambda *args: uuid4().hex)