Merge pull request #548 from dod-ccpo/to-congrats-msg

Add "Congrats" message after completing TO form

tests/routes/task_orders/test_invite.py

@@ -1,11 +1,17 @@
 import pytest
 from flask import url_for
 
-from tests.factories import TaskOrderFactory
+from tests.factories import PortfolioFactory, TaskOrderFactory
 
 
-def test_invite(client):
-    to = TaskOrderFactory.create()
+def test_invite(client, user_session):
+    portfolio = PortfolioFactory.create()
+    user_session(portfolio.owner)
+    to = TaskOrderFactory.create(portfolio=portfolio)
     response = client.post(
         url_for("task_orders.invite", task_order_id=to.id), follow_redirects=False
     )
+    redirect = url_for(
+        "portfolios.view_task_order", portfolio_id=to.portfolio_id, task_order_id=to.id
+    )
+    assert redirect in response.headers["Location"]