mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-04-28 13:11:05 -04:00
53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
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)
|