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

Future proof edit route

Browse Source
This commit is contained in:
richard-dds 2019-07-30 14:07:04 -04:00
parent 136b97b23f
commit 43ed257f10
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
atst/routes/task_orders/new.py
View File

@ -60,7 +60,11 @@ def update_task_order(
def edit(task_order_id):
task_order = TaskOrders.get(task_order_id)
if not task_order.number:
if not task_order.pdf:
return redirect(
url_for("task_orders.form_step_one_add_pdf", task_order_id=task_order_id)
)
elif not task_order.number:
return redirect(
url_for("task_orders.form_step_two_add_number", task_order_id=task_order_id)
)

1
tests/routes/task_orders/test_new.py
View File

@ -307,6 +307,7 @@ def test_task_orders_submit_task_order(client, user_session, task_order):
@pytest.mark.parametrize(
"to_factory_args,expected_step",
[
({"pdf": None, "number": "", "clins": []}, "step_1"),
({"number": "", "clins": []}, "step_2"),
({"number": "1234567890123", "clins": []}, "step_3"),
({"number": "1234567890123", "create_clins": [1]}, "step_4"),