notifications-admin/tests/app/main/views/test_sign_out.py

from datetime import datetime
from flask import url_for
from app.main.dao import users_dao
from app.models import User

from .test_sign_in import _set_up_mocker


def test_render_sign_out_redirects_to_sign_in(app_):
    with app_.test_request_context():
        response = app_.test_client().get(
            url_for('main.sign_out'))
        assert response.status_code == 302
        assert response.location == url_for(
            'main.sign_in', _external=True, next=url_for('main.sign_out'))


def test_sign_out_user(app_,
                       db_,
                       db_session,
                       mock_send_sms,
                       mock_send_email):
    with app_.test_request_context():
        email = 'valid@example.gov.uk'
        password = 'val1dPassw0rd!'
        user = User(email_address=email,
                    password=password,
                    mobile_number='+441234123123',
                    name='valid',
                    created_at=datetime.now(),
                    role_id=1,
                    state='active')
        users_dao.insert_user(user)
        with app_.test_client() as client:
            client.login(user)
            # Check we are logged in
            response = client.get('/services/123/dashboard')
            assert response.status_code == 200
            response = client.get(url_for('main.sign_out'))
            assert response.status_code == 302
            assert response.location == url_for(
                'main.index', _external=True)