Add another route for new TOs

This commit is contained in:
richard-dds 2019-07-30 11:06:59 -04:00
parent 3c56f96fa2
commit ec6494c6c3
2 changed files with 10 additions and 1 deletions

View File

@ -118,9 +118,10 @@ def submit_form_step_one_add_pdf(portfolio_id=None, task_order_id=None):
) )
@task_orders_bp.route("/portfolios/<portfolio_id>/task_orders/form/cancel", methods=["POST"])
@task_orders_bp.route("/task_orders/<task_order_id>/form/cancel", methods=["POST"]) @task_orders_bp.route("/task_orders/<task_order_id>/form/cancel", methods=["POST"])
@user_can(Permissions.CREATE_TASK_ORDER, message="cancel task order form") @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) save = http_request.args.get("save", False)
if save: if save:
form_data = {**http_request.form} form_data = {**http_request.form}

View File

@ -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) updated_task_order = session.query(TaskOrder).get(task_order.id)
assert updated_task_order.number == "7896564324567" 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): def test_cancel_edit_does_not_save_invalid_form_input(client, user_session, session):
task_order = TaskOrderFactory.create() task_order = TaskOrderFactory.create()