mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-05-05 16:38:59 -04:00
Add test for redirecting to service in session
If you click GOV.UK and have: - multiple services - a service in your session …we take you to the dashboard for that service. This worked great, but wasn’t tested. This commit adds a test for it.
This commit is contained in:
@@ -1,4 +1,4 @@
|
|||||||
from flask import url_for
|
from flask import url_for, session
|
||||||
|
|
||||||
|
|
||||||
def test_should_show_choose_services_page(app_,
|
def test_should_show_choose_services_page(app_,
|
||||||
@@ -54,6 +54,29 @@ def test_redirect_if_multiple_services(
|
|||||||
assert response.location == url_for('main.choose_service', _external=True)
|
assert response.location == url_for('main.choose_service', _external=True)
|
||||||
|
|
||||||
|
|
||||||
|
def test_redirect_if_service_in_session(
|
||||||
|
app_,
|
||||||
|
mock_login,
|
||||||
|
mock_get_user,
|
||||||
|
api_user_active,
|
||||||
|
mock_get_services,
|
||||||
|
mock_get_service
|
||||||
|
):
|
||||||
|
with app_.test_request_context():
|
||||||
|
with app_.test_client() as client:
|
||||||
|
client.login(api_user_active)
|
||||||
|
with client.session_transaction() as session:
|
||||||
|
session['service_id'] = '147ad62a-2951-4fa1-9ca0-093cd1a52c52'
|
||||||
|
response = client.get(url_for('main.show_all_services_or_dashboard'))
|
||||||
|
|
||||||
|
assert response.status_code == 302
|
||||||
|
assert response.location == url_for(
|
||||||
|
'main.service_dashboard',
|
||||||
|
service_id='147ad62a-2951-4fa1-9ca0-093cd1a52c52',
|
||||||
|
_external=True
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def test_should_redirect_if_not_logged_in(app_):
|
def test_should_redirect_if_not_logged_in(app_):
|
||||||
with app_.test_request_context():
|
with app_.test_request_context():
|
||||||
with app_.test_client() as client:
|
with app_.test_client() as client:
|
||||||
|
|||||||
Reference in New Issue
Block a user