Better ergonomics for creating factory portfolios w/ TOs

This commit is contained in:
richard-dds
2019-09-10 14:33:35 -04:00
parent f6cb6f2a31
commit 6b7db2ca46
4 changed files with 17 additions and 8 deletions

View File

@@ -163,12 +163,12 @@ class EnvQueryTest:
}
],
task_orders=[
TaskOrderFactory.create(
clins=[
CLINFactory.create(start_date=start_date, end_date=end_date)
{
"create_clins": [
{"start_date": start_date, "end_date": end_date}
for (start_date, end_date) in start_and_end_dates
]
)
}
],
)

View File

@@ -135,7 +135,7 @@ def test_update_adds_clins():
def test_update_does_not_duplicate_clins():
task_order = TaskOrderFactory.create(
number="3453453456", create_clins=["123", "456"]
number="3453453456", create_clins=[{"number": "123"}, {"number": "456"}]
)
clins = [
{
@@ -165,7 +165,9 @@ def test_update_does_not_duplicate_clins():
def test_delete_task_order_with_clins(session):
task_order = TaskOrderFactory.create(create_clins=[1, 2, 3])
task_order = TaskOrderFactory.create(
create_clins=[{"number": 1}, {"number": 2}, {"number": 3}]
)
TaskOrders.delete(task_order.id)
assert not session.query(