Redirect on successful save

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(

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