Better ergonomics for creating factory portfolios w/ TOs
This commit is contained in:
@@ -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
|
||||
]
|
||||
)
|
||||
}
|
||||
],
|
||||
)
|
||||
|
||||
|
@@ -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(
|
||||
|
Reference in New Issue
Block a user