diff --git a/atst/routes/task_orders/signing.py b/atst/routes/task_orders/signing.py index 3c0458cd..595c184b 100644 --- a/atst/routes/task_orders/signing.py +++ b/atst/routes/task_orders/signing.py @@ -1,4 +1,4 @@ -from flask import render_template, g, request as http_request +from flask import url_for, redirect, render_template, g, request as http_request import datetime @@ -53,7 +53,14 @@ def record_signature(task_order_id): signed_at=datetime.datetime.now(), **form.data, ) - return render_template("task_orders/signing/success.html"), 201 + + return redirect( + url_for( + "portfolios.view_task_order", + portfolio_id=task_order.portfolio_id, + task_order_id=task_order.id, + ) + ) else: return ( render_template( diff --git a/tests/routes/task_orders/test_sign.py b/tests/routes/task_orders/test_sign.py index 590cf280..0196c4dd 100644 --- a/tests/routes/task_orders/test_sign.py +++ b/tests/routes/task_orders/test_sign.py @@ -89,7 +89,15 @@ def test_signing_a_task_order(client, user_session): data={"signature": "y", "level_of_warrant": "33.33"}, ) - assert response.status_code == 201 + assert ( + url_for( + "portfolios.view_task_order", + portfolio_id=task_order.portfolio_id, + task_order_id=task_order.id, + ) + in response.headers["Location"] + ) + assert task_order.signer_dod_id == contracting_officer.dod_id assert task_order.signed_at is not None @@ -122,7 +130,15 @@ def test_signing_a_task_order_unlimited_level_of_warrant(client, user_session): }, ) - assert response.status_code == 201 + assert ( + url_for( + "portfolios.view_task_order", + portfolio_id=task_order.portfolio_id, + task_order_id=task_order.id, + ) + in response.headers["Location"] + ) + assert task_order.signed_at is not None assert task_order.signer_dod_id == contracting_officer.dod_id assert task_order.unlimited_level_of_warrant == True