notifications-admin/tests/app/main/views/test_sign_out.py

from flask import url_for

from tests.conftest import SERVICE_ONE_ID


def test_render_sign_out_redirects_to_sign_in(
    logged_in_client_with_session
):
    with logged_in_client_with_session.session_transaction() as session:
        assert session
    response = logged_in_client_with_session.get(
        url_for('main.sign_out'))
    assert response.status_code == 302
    assert response.location == url_for(
        'main.index', _external=True)
    with logged_in_client_with_session.session_transaction() as session:
        assert not session


def test_sign_out_user(
    client_request,
    mock_get_service,
    api_user_active,
    mock_get_user,
    mock_get_user_by_email,
    mock_login,
    mock_get_service_templates,
    mock_has_no_jobs,
    mock_has_permissions,
    mock_get_template_statistics,
    mock_get_service_statistics,
    mock_get_usage,
    mock_get_free_sms_fragment_limit,
    mock_get_inbound_sms_summary,
    mock_get_returned_letter_statistics_with_no_returned_letters
):
    with client_request.session_transaction() as session:
        assert session.get('user_id') is not None
    # Check we are logged in
    client_request.get(
        'main.service_dashboard',
        service_id=SERVICE_ONE_ID,
    )
    client_request.get(
        'main.sign_out',
        _expected_status=302,
        _expected_redirect=url_for(
            'main.index',
            _external=True,
        )
    )
    with client_request.session_transaction() as session:
        assert session.get('user_id') is None


def test_sign_out_of_two_sessions(
    logged_in_client_with_session
):
    logged_in_client_with_session.get(
        url_for('main.sign_out'))
    with logged_in_client_with_session.session_transaction() as session:
        assert not session
    response = logged_in_client_with_session.get(
        url_for('main.sign_out'))

    assert response.status_code == 302