from flask import url_for
from app.main.dao import users_dao
from tests import create_test_user


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_and_state_updated(app_,
                                                                  mock_send_email,
                                                                  api_user_active,
                                                                  mock_user_dao_get_by_email,
                                                                  mock_user_dao_password_reset):
    with app_.test_request_context():
        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)
        assert api_user_active.state == 'request_password_reset'