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

import pytest

from flask import url_for, Response
from notifications_python_client.errors import HTTPError
from tests.conftest import api_user_active as create_active_user

import app


def test_should_render_forgot_password(client):
    response = client.get(url_for('.forgot_password'))
    assert response.status_code == 200
    assert 'We’ll send you an email to create a new password.' \
           in response.get_data(as_text=True)


@pytest.mark.parametrize('email_address', [
    'test@user.gov.uk',
    'someuser@notonwhitelist.com'
])
def test_should_redirect_to_password_reset_sent_for_valid_email(
    client,
    fake_uuid,
    email_address,
    mocker,
):
    sample_user = create_active_user(fake_uuid, email_address=email_address)
    mocker.patch('app.user_api_client.send_reset_password_url', return_value=None)
    response = client.post(
        url_for('.forgot_password'),
        data={'email_address': sample_user.email_address})
    assert response.status_code == 200
    assert 'Click the link in the email to reset your password.' \
           in response.get_data(as_text=True)
    app.user_api_client.send_reset_password_url.assert_called_once_with(sample_user.email_address)


def test_should_redirect_to_password_reset_sent_for_missing_email(
    client,
    api_user_active,
    mocker,
):

    mocker.patch('app.user_api_client.send_reset_password_url', side_effect=HTTPError(Response(status=404),
                                                                                      'Not found'))
    response = client.post(
        url_for('.forgot_password'),
        data={'email_address': api_user_active.email_address})
    assert response.status_code == 200
    assert 'Click the link in the email to reset your password.' \
           in response.get_data(as_text=True)
    app.user_api_client.send_reset_password_url.assert_called_once_with(api_user_active.email_address)