mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 10:53:28 -05:00
47 lines
2.0 KiB
Python
47 lines
2.0 KiB
Python
from flask import url_for, Response
|
|
from notifications_python_client.errors import HTTPError
|
|
|
|
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 'If you have forgotten your password, we can send you an email to create a new password.' \
|
|
in response.get_data(as_text=True)
|
|
|
|
|
|
def test_should_redirect_to_password_reset_sent_for_valid_email(
|
|
app_,
|
|
api_user_active,
|
|
mocker):
|
|
with app_.test_request_context():
|
|
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': api_user_active.email_address})
|
|
assert response.status_code == 200
|
|
assert (
|
|
'You have been sent an email containing a link'
|
|
' 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)
|
|
|
|
|
|
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 (
|
|
'You have been sent an email containing a link'
|
|
' 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)
|