From ec6494c6c352f7a5ceec6b9ce9dfd5d4d6d1eb3e Mon Sep 17 00:00:00 2001 From: richard-dds Date: Tue, 30 Jul 2019 11:06:59 -0400 Subject: [PATCH] Add another route for new TOs --- atst/routes/task_orders/new.py | 3 ++- tests/routes/task_orders/test_new.py | 8 ++++++++ 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/atst/routes/task_orders/new.py b/atst/routes/task_orders/new.py index 4496a21a..c3259902 100644 --- a/atst/routes/task_orders/new.py +++ b/atst/routes/task_orders/new.py @@ -118,9 +118,10 @@ def submit_form_step_one_add_pdf(portfolio_id=None, task_order_id=None): ) +@task_orders_bp.route("/portfolios//task_orders/form/cancel", methods=["POST"]) @task_orders_bp.route("/task_orders//form/cancel", methods=["POST"]) @user_can(Permissions.CREATE_TASK_ORDER, message="cancel task order form") -def cancel_edit(task_order_id): +def cancel_edit(task_order_id=None, portfolio_id=None): save = http_request.args.get("save", False) if save: form_data = {**http_request.form} diff --git a/tests/routes/task_orders/test_new.py b/tests/routes/task_orders/test_new.py index cc22094d..cee4c26d 100644 --- a/tests/routes/task_orders/test_new.py +++ b/tests/routes/task_orders/test_new.py @@ -329,6 +329,14 @@ def test_can_cancel_edit_and_save_task_order(client, user_session, task_order, s updated_task_order = session.query(TaskOrder).get(task_order.id) assert updated_task_order.number == "7896564324567" +def test_cancel_can_create_new_to(client, user_session, portfolio): + user_session(portfolio.owner) + response = client.post( + url_for("task_orders.cancel_edit", portfolio_id=portfolio.id), + data={"number": "7643906432984"}, + ) + assert response.status_code == 302 + def test_cancel_edit_does_not_save_invalid_form_input(client, user_session, session): task_order = TaskOrderFactory.create()