Fix test for task_orders.invite route

This commit is contained in:
Patrick Smith 2019-01-17 09:03:51 -05:00
parent a8ac5de968
commit 87cd100c51

View File

@ -1,11 +1,17 @@
import pytest import pytest
from flask import url_for from flask import url_for
from tests.factories import TaskOrderFactory from tests.factories import PortfolioFactory, TaskOrderFactory
def test_invite(client): def test_invite(client, user_session):
to = TaskOrderFactory.create() portfolio = PortfolioFactory.create()
user_session(portfolio.owner)
to = TaskOrderFactory.create(portfolio=portfolio)
response = client.post( response = client.post(
url_for("task_orders.invite", task_order_id=to.id), follow_redirects=False 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"]