If user is logged in and visits / /sign-in or /register they will

be redirected to choose service.
This commit is contained in:
Adam Shimali
2016-01-22 17:24:14 +00:00
parent f11f50b4a5
commit f8fc8e951a
6 changed files with 110 additions and 49 deletions

View File

@@ -0,0 +1,17 @@
from flask import url_for
def test_logged_in_user_redirects_to_choose_service(app_,
db_,
db_session,
mock_api_user,
mock_user_loader,
mock_user_dao_get_by_email):
with app_.test_request_context():
with app_.test_client() as client:
client.login(mock_api_user)
response = client.get(url_for('main.index'))
assert response.status_code == 302
response = client.get(url_for('main.sign_in', follow_redirects=True))
assert response.location == url_for('main.choose_service', _external=True)