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

More realistic seeds

Browse Source
This commit is contained in:
George Drummond 2019-06-12 10:28:11 -04:00
parent e554845407
commit 9edf816765
No known key found for this signature in database
GPG Key ID: 296DD6077123BF17
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
script/seed_sample.py
View File

@ -164,6 +164,7 @@ def add_task_orders_to_portfolio(portfolio):
today = date.today() today = date.today()
future = today + timedelta(days=100) future = today + timedelta(days=100)
yesterday = today - timedelta(days=1) yesterday = today - timedelta(days=1)
five_days = timedelta(days=5)
draft_to = TaskOrderFactory.build(portfolio=portfolio, pdf=None) draft_to = TaskOrderFactory.build(portfolio=portfolio, pdf=None)
unsigned_to = TaskOrderFactory.build(portfolio=portfolio) 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) active_to = TaskOrderFactory.build(portfolio=portfolio, signed_at=yesterday)
clins = [ 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( 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), CLINFactory.build(task_order=active_to, start_date=yesterday, end_date=future),
] ]

4
tests/factories.py
View File

@ -30,7 +30,7 @@ def random_phone_number():
def random_task_order_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): def random_past_date(year_min=1, year_max=5):
@ -261,7 +261,7 @@ class AttachmentFactory(Base):
class Meta: class Meta:
model = Attachment model = Attachment
filename = factory.Faker("domain_word") filename = factory.Faker("file_name", extension="pdf")
object_name = factory.LazyFunction(lambda *args: uuid4().hex) object_name = factory.LazyFunction(lambda *args: uuid4().hex)