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


def test_should_render_forgot_password(notifications_admin, notifications_admin_db, notify_db_session):
    with notifications_admin.test_request_context():
        response = notifications_admin.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(notifications_admin,
                                                                  notifications_admin_db,
                                                                  mocker,
                                                                  notify_db_session):
    mocker.patch("app.admin_api_client.send_email")
    with notifications_admin.test_request_context():
        user = create_test_user('active')
        response = notifications_admin.test_client().post(url_for('.forgot_password'),
                                                          data={'email_address': user.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 users_dao.get_user_by_id(user.id).state == 'request_password_reset'