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

import json
import uuid

import pytest
from flask import url_for
from notifications_python_client.errors import HTTPError
from notifications_utils.url_safe_token import generate_token

from app.models.webauthn_credential import (
    WebAuthnCredential,
    WebAuthnCredentials,
)
from tests.conftest import (
    create_api_user_active,
    normalize_spaces,
    url_for_endpoint_with_token,
)


def test_should_show_overview_page(
    client_request,
):
    page = client_request.get('main.user_profile')
    assert page.select_one('h1').text.strip() == 'Your profile'
    assert 'Use platform admin view' not in page
    assert 'Security keys' not in page


def test_overview_page_shows_disable_for_platform_admin(
    client_request,
    platform_admin_user,
    mocker
):
    mocker.patch('app.models.webauthn_credential.WebAuthnCredentials.client_method')
    client_request.login(platform_admin_user)
    page = client_request.get('main.user_profile')
    assert page.select_one('h1').text.strip() == 'Your profile'
    disable_platform_admin_row = page.select_one('#disable-platform-admin')
    assert ' '.join(disable_platform_admin_row.text.split()) == \
        'Use platform admin view Yes Change whether to use platform admin view'


@pytest.mark.parametrize('key_count, expected_row_text', [
    (0, 'Security keys None registered Change security keys'),
    (1, 'Security keys 1 registered Change security keys'),
    (2, 'Security keys 2 registered Change security keys'),
])
def test_overview_page_shows_security_keys_if_user_they_can_use_webauthn(
    mocker,
    client_request,
    platform_admin_user,
    webauthn_credential,
    key_count,
    expected_row_text,
):
    client_request.login(platform_admin_user)
    credentials = [webauthn_credential for _ in range(key_count)]
    mocker.patch(
        'app.models.webauthn_credential.WebAuthnCredentials.client_method',
        return_value=credentials,
    )
    page = client_request.get('main.user_profile')
    security_keys_row = page.select_one('#security-keys')
    assert ' '.join(security_keys_row.text.split()) == expected_row_text


def test_should_show_name_page(
    client_request
):
    page = client_request.get(('main.user_profile_name'))
    assert page.select_one('h1').text.strip() == 'Change your name'


def test_should_redirect_after_name_change(
    client_request,
    mock_update_user_attribute,
    mock_email_is_not_already_in_use
):
    client_request.post(
        'main.user_profile_name',
        _data={'new_name': 'New Name'},
        _expected_status=302,
        _expected_redirect=url_for('main.user_profile', _external=True),
    )
    assert mock_update_user_attribute.called is True


def test_should_show_email_page(
    client_request,
):
    page = client_request.get(
        'main.user_profile_email'
    )
    assert page.select_one('h1').text.strip() == 'Change your email address'


def test_should_redirect_after_email_change(
    client_request,
    mock_login,
    mock_email_is_not_already_in_use,
):
    client_request.post(
        'main.user_profile_email',
        _data={'email_address': 'new_notify@notify.gov.uk'},
        _expected_status=302,
        _expected_redirect=url_for(
            'main.user_profile_email_authenticate',
            _external=True,
        )
    )


def test_should_show_authenticate_after_email_change(
    client_request,
):
    with client_request.session_transaction() as session:
        session['new-email'] = 'new_notify@notify.gov.uk'

    page = client_request.get('main.user_profile_email_authenticate')

    assert 'Change your email address' in page.text
    assert 'Confirm' in page.text


def test_should_render_change_email_continue_after_authenticate_email(
    client_request,
    mock_verify_password,
    mock_send_change_email_verification,
):
    with client_request.session_transaction() as session:
        session['new-email'] = 'new_notify@notify.gov.uk'
    page = client_request.post(
        'main.user_profile_email_authenticate',
        _data={'password': '12345'},
        _expected_status=200,
    )
    assert 'Click the link in the email to confirm the change to your email address.' in page.text


def test_should_redirect_to_user_profile_when_user_confirms_email_link(
    notify_admin,
    logged_in_client,
    api_user_active,
    mock_update_user_attribute,
):

    token = generate_token(payload=json.dumps({'user_id': api_user_active['id'], 'email': 'new_email@gov.uk'}),
                           secret=notify_admin.config['SECRET_KEY'], salt=notify_admin.config['DANGEROUS_SALT'])
    response = logged_in_client.get(url_for_endpoint_with_token('main.user_profile_email_confirm',
                                                                token=token))

    assert response.status_code == 302
    assert response.location == url_for('main.user_profile', _external=True)


def test_should_show_mobile_number_page(
    client_request,
):
    page = client_request.get(('main.user_profile_mobile_number'))
    assert 'Change your mobile number' in page.text


@pytest.mark.parametrize('phone_number_to_register_with', [
    '+4407700900460',
    '+1800-555-555',
])
def test_should_redirect_after_mobile_number_change(
    client_request,
    phone_number_to_register_with,
):
    client_request.post(
        'main.user_profile_mobile_number',
        _data={'mobile_number': phone_number_to_register_with},
        _expected_status=302,
        _expected_redirect=url_for(
            'main.user_profile_mobile_number_authenticate',
            _external=True,
        )
    )
    with client_request.session_transaction() as session:
        assert session['new-mob'] == phone_number_to_register_with


def test_should_show_authenticate_after_mobile_number_change(
    client_request,
):
    with client_request.session_transaction() as session:
        session['new-mob'] = '+441234123123'

    page = client_request.get(
        'main.user_profile_mobile_number_authenticate',
    )

    assert 'Change your mobile number' in page.text
    assert 'Confirm' in page.text


def test_should_redirect_after_mobile_number_authenticate(
    client_request,
    mock_verify_password,
    mock_send_verify_code,
):
    with client_request.session_transaction() as session:
        session['new-mob'] = '+441234123123'

    client_request.post(
        'main.user_profile_mobile_number_authenticate',
        _data={'password': '12345667'},
        _expected_status=302,
        _expected_redirect=url_for(
            'main.user_profile_mobile_number_confirm',
            _external=True,
        )
    )


def test_should_show_confirm_after_mobile_number_change(
    client_request,
):
    with client_request.session_transaction() as session:
        session['new-mob-password-confirmed'] = True
    page = client_request.get(
        'main.user_profile_mobile_number_confirm'
    )

    assert 'Change your mobile number' in page.text
    assert 'Confirm' in page.text


@pytest.mark.parametrize('phone_number_to_register_with', [
    '+4407700900460',
    '+1800-555-555',
])
def test_should_redirect_after_mobile_number_confirm(
    client_request,
    mocker,
    mock_update_user_attribute,
    mock_check_verify_code,
    phone_number_to_register_with,
):
    user_before = create_api_user_active(with_unique_id=True)
    user_after = create_api_user_active(with_unique_id=True)
    user_before['current_session_id'] = str(uuid.UUID(int=1))
    user_after['current_session_id'] = str(uuid.UUID(int=2))

    # first time (login decorator) return normally, second time (after 2FA return with new session id)
    mocker.patch('app.user_api_client.get_user', side_effect=[user_before, user_after])

    with client_request.session_transaction() as session:
        session['new-mob-password-confirmed'] = True
        session['new-mob'] = phone_number_to_register_with
        session['current_session_id'] = user_before['current_session_id']

    client_request.post(
        'main.user_profile_mobile_number_confirm',
        _data={'sms_code': '12345'},
        _expected_status=302,
        _expected_redirect=url_for(
            'main.user_profile',
            _external=True,
        )
    )

    # make sure the current_session_id has changed to what the API returned
    with client_request.session_transaction() as session:
        assert session['current_session_id'] == user_after['current_session_id']


def test_should_show_password_page(
    client_request,
):
    page = client_request.get(('main.user_profile_password'))

    assert page.select_one('h1').text.strip() == 'Change your password'


def test_should_redirect_after_password_change(
    client_request,
    mock_update_user_password,
    mock_verify_password,
):
    client_request.post(
        'main.user_profile_password',
        _data={
            'new_password': 'the new password',
            'old_password': 'the old password',
        },
        _expected_status=302,
        _expected_redirect=url_for(
            'main.user_profile',
            _external=True,
        ),
    )


def test_non_gov_user_cannot_see_change_email_link(
    client_request,
    api_nongov_user_active,
    mock_get_organisations,
):
    client_request.login(api_nongov_user_active)
    page = client_request.get('main.user_profile')
    assert not page.find('a', {'href': url_for('main.user_profile_email')})
    assert page.select_one('h1').text.strip() == 'Your profile'


def test_non_gov_user_cannot_access_change_email_page(
    client_request,
    api_nongov_user_active,
    mock_get_organisations,
):
    client_request.login(api_nongov_user_active)
    client_request.get('main.user_profile_email', _expected_status=403)


def test_normal_user_doesnt_see_disable_platform_admin(client_request):
    client_request.get('main.user_profile_disable_platform_admin_view', _expected_status=403)


def test_platform_admin_can_see_disable_platform_admin_page(client_request, platform_admin_user):
    client_request.login(platform_admin_user)
    page = client_request.get('main.user_profile_disable_platform_admin_view')

    assert page.select_one('h1').text.strip() == 'Use platform admin view'
    assert page.select_one('input[checked]')['value'] == 'True'


def test_can_disable_platform_admin(client_request, platform_admin_user):
    client_request.login(platform_admin_user)

    with client_request.session_transaction() as session:
        assert 'disable_platform_admin_view' not in session

    client_request.post(
        'main.user_profile_disable_platform_admin_view',
        _data={'enabled': False},
        _expected_status=302,
        _expected_redirect=url_for('main.user_profile', _external=True),
    )

    with client_request.session_transaction() as session:
        assert session['disable_platform_admin_view'] is True


def test_can_reenable_platform_admin(client_request, platform_admin_user):
    client_request.login(platform_admin_user)

    with client_request.session_transaction() as session:
        session['disable_platform_admin_view'] = True

    client_request.post(
        'main.user_profile_disable_platform_admin_view',
        _data={'enabled': True},
        _expected_status=302,
        _expected_redirect=url_for('main.user_profile', _external=True),
    )

    with client_request.session_transaction() as session:
        assert session['disable_platform_admin_view'] is False


def test_user_doesnt_see_security_keys_unless_they_can_use_webauthn(
    client_request,
    platform_admin_user
):
    platform_admin_user['can_use_webauthn'] = False
    client_request.login(platform_admin_user)

    client_request.get(
        '.user_profile_security_keys',
        _expected_status=403,
    )


def test_should_show_security_keys_page(
    mocker,
    client_request,
    platform_admin_user,
    webauthn_credential,
):
    client_request.login(platform_admin_user)

    mocker.patch(
        'app.models.webauthn_credential.WebAuthnCredentials.client_method',
        return_value=[webauthn_credential],
    )

    page = client_request.get('.user_profile_security_keys')
    assert page.select_one('h1').text.strip() == 'Security keys'

    credential_row = page.select('tr')[-1]
    assert 'Test credential' in credential_row.text
    assert "Manage" in credential_row.find('a').text
    assert credential_row.find('a')["href"] == url_for(
        '.user_profile_manage_security_key',
        key_id=webauthn_credential['id']
    )

    register_button = page.select_one("[data-module='register-security-key']")
    assert register_button.text.strip() == 'Register a key'


def test_get_key_from_list_of_keys(
    mocker,
    webauthn_credential,
    webauthn_credential_2,
    fake_uuid,
):
    mocker.patch(
        'app.models.webauthn_credential.WebAuthnCredentials.client_method',
        return_value=[webauthn_credential, webauthn_credential_2],
    )
    assert WebAuthnCredentials(fake_uuid).by_id(webauthn_credential["id"]) == WebAuthnCredential(webauthn_credential)


def test_should_show_manage_security_key_page(
    mocker,
    client_request,
    platform_admin_user,
    webauthn_credential,
):
    client_request.login(platform_admin_user)

    mocker.patch(
        'app.models.webauthn_credential.WebAuthnCredentials.client_method',
        return_value=[webauthn_credential],
    )

    page = client_request.get('.user_profile_manage_security_key', key_id=webauthn_credential['id'])
    assert page.select_one('h1').text.strip() == f'Manage ‘{webauthn_credential["name"]}’'

    assert page.select_one('.govuk-back-link').text.strip() == 'Back'
    assert page.select_one('.govuk-back-link')['href'] == url_for('.user_profile_security_keys')

    assert page.select_one('#security_key_name')["value"] == webauthn_credential["name"]


def test_manage_security_key_page_404s_when_key_not_found(
    mocker,
    client_request,
    platform_admin_user,
    webauthn_credential,
    webauthn_credential_2
):
    client_request.login(platform_admin_user)

    mocker.patch(
        'app.models.webauthn_credential.WebAuthnCredentials.client_method',
        return_value=[webauthn_credential_2],
    )
    client_request.get(
        '.user_profile_manage_security_key',
        key_id=webauthn_credential['id'],
        _expected_status=404,
    )


@pytest.mark.parametrize('endpoint,method', [
    (".user_profile_manage_security_key", "get"),
    (".user_profile_manage_security_key", "post"),
    (".user_profile_confirm_delete_security_key", "get"),
    (".user_profile_confirm_delete_security_key", "post"),
    (".user_profile_delete_security_key", "post"),
])
def test_cant_manage_security_keys_unless_can_use_webauthn(
    client_request,
    platform_admin_user,
    webauthn_credential,
    endpoint,
    method
):
    platform_admin_user['can_use_webauthn'] = False
    client_request.login(platform_admin_user)

    if method == "get":
        client_request.get(
            endpoint,
            key_id=webauthn_credential['id'],
            _expected_status=403,
        )

    else:
        client_request.post(
            endpoint,
            key_id=webauthn_credential['id'],
            _expected_status=403,
        )


def test_should_redirect_after_change_of_security_key_name(
    client_request,
    platform_admin_user,
    webauthn_credential,
    mocker
):
    client_request.login(platform_admin_user)

    mocker.patch(
        'app.models.webauthn_credential.WebAuthnCredentials.client_method',
        return_value=[webauthn_credential],
    )

    mock_update = mocker.patch('app.user_api_client.update_webauthn_credential_name_for_user')

    client_request.post(
        'main.user_profile_manage_security_key',
        key_id=webauthn_credential['id'],
        _data={'security_key_name': "new name"},
        _expected_status=302,
        _expected_redirect=url_for(
            'main.user_profile_security_keys',
            _external=True,
        )
    )

    mock_update.assert_called_once_with(
        credential_id=webauthn_credential['id'],
        new_name_for_credential="new name",
        user_id=platform_admin_user["id"]
    )


def test_user_profile_manage_security_key_should_not_call_api_if_key_name_stays_the_same(
    client_request,
    platform_admin_user,
    webauthn_credential,
    mocker
):
    client_request.login(platform_admin_user)

    mocker.patch(
        'app.models.webauthn_credential.WebAuthnCredentials.client_method',
        return_value=[webauthn_credential],
    )

    mock_update = mocker.patch('app.user_api_client.update_webauthn_credential_name_for_user')

    client_request.post(
        'main.user_profile_manage_security_key',
        key_id=webauthn_credential['id'],
        _data={'security_key_name': webauthn_credential['name']},
        _expected_status=302,
        _expected_redirect=url_for(
            'main.user_profile_security_keys',
            _external=True,
        )
    )

    assert not mock_update.called


def test_shows_delete_link_for_security_key(
    mocker,
    client_request,
    platform_admin_user,
    webauthn_credential,
):
    client_request.login(platform_admin_user)

    mocker.patch(
        'app.models.webauthn_credential.WebAuthnCredentials.client_method',
        return_value=[webauthn_credential],
    )

    page = client_request.get('.user_profile_manage_security_key', key_id=webauthn_credential['id'])
    assert page.select_one('h1').text.strip() == f'Manage ‘{webauthn_credential["name"]}’'

    link = page.select_one('.page-footer a')
    assert normalize_spaces(link.text) == 'Delete'
    assert link['href'] == url_for('.user_profile_confirm_delete_security_key', key_id=webauthn_credential['id'])


def test_confirm_delete_security_key(
    client_request,
    platform_admin_user,
    webauthn_credential,
    mocker
):
    client_request.login(platform_admin_user)

    mocker.patch(
        'app.models.webauthn_credential.WebAuthnCredentials.client_method',
        return_value=[webauthn_credential],
    )

    page = client_request.get(
        '.user_profile_confirm_delete_security_key',
        key_id=webauthn_credential['id'],
        _test_page_title=False,
    )

    assert normalize_spaces(page.select_one('.banner-dangerous').text) == (
        'Are you sure you want to delete this security key? '
        'Yes, delete'
    )
    assert 'action' not in page.select_one('.banner-dangerous form')
    assert page.select_one('.banner-dangerous form')['method'] == 'post'


def test_delete_security_key(
    client_request,
    platform_admin_user,
    webauthn_credential,
    mocker
):
    client_request.login(platform_admin_user)
    mock_delete = mocker.patch('app.user_api_client.delete_webauthn_credential_for_user')

    client_request.post(
        '.user_profile_delete_security_key',
        key_id=webauthn_credential['id'],
        _expected_redirect=url_for(
            '.user_profile_security_keys',
            _external=True,
        )
    )
    mock_delete.assert_called_once_with(
        credential_id=webauthn_credential['id'],
        user_id=platform_admin_user["id"]
    )


def test_delete_security_key_handles_last_credential_error(
    client_request,
    platform_admin_user,
    webauthn_credential,
    mocker,
):
    client_request.login(platform_admin_user)
    mocker.patch(
        'app.models.webauthn_credential.WebAuthnCredentials.client_method',
        return_value=[webauthn_credential],
    )

    mocker.patch(
        'app.user_api_client.delete_webauthn_credential_for_user',
        side_effect=HTTPError(
            response={},
            message='Cannot delete last remaining webauthn credential for user'
        )
    )

    page = client_request.post(
        '.user_profile_delete_security_key',
        key_id=webauthn_credential['id'],
        _follow_redirects=True
    )
    assert 'Manage ‘Test credential’' in page.find('h1').text
    expected_message = "You cannot delete your last security key."
    assert expected_message in page.find('div', class_="banner-dangerous").text