diff --git a/atst/routes/task_orders/new.py b/atst/routes/task_orders/new.py index 8c5d4fd3..ee639690 100644 --- a/atst/routes/task_orders/new.py +++ b/atst/routes/task_orders/new.py @@ -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) ) diff --git a/tests/routes/task_orders/test_new.py b/tests/routes/task_orders/test_new.py index 0dfa0b23..feef2844 100644 --- a/tests/routes/task_orders/test_new.py +++ b/tests/routes/task_orders/test_new.py @@ -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"),