mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-05-05 16:38:59 -04:00
54 lines
2.1 KiB
Python
54 lines
2.1 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(app_):
|
||
with app_.test_request_context():
|
||
response = app_.test_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(
|
||
app_,
|
||
fake_uuid,
|
||
email_address,
|
||
mocker):
|
||
with app_.test_request_context():
|
||
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 = app_.test_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(
|
||
app_,
|
||
api_user_active,
|
||
mocker):
|
||
with app_.test_request_context():
|
||
|
||
mocker.patch('app.user_api_client.send_reset_password_url', side_effect=HTTPError(Response(status=404),
|
||
'Not found'))
|
||
response = app_.test_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)
|