From 87cd100c515d43a18253485f2c60f2231286e8a1 Mon Sep 17 00:00:00 2001 From: Patrick Smith Date: Thu, 17 Jan 2019 09:03:51 -0500 Subject: [PATCH] Fix test for task_orders.invite route --- tests/routes/task_orders/test_invite.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tests/routes/task_orders/test_invite.py b/tests/routes/task_orders/test_invite.py index 39cfeb76..6ee9fa30 100644 --- a/tests/routes/task_orders/test_invite.py +++ b/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"]