mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-26 01:04:00 -04:00
Logout functionality and test added.
This commit is contained in:
43
tests/app/main/views/test_sign_out.py
Normal file
43
tests/app/main/views/test_sign_out.py
Normal file
@@ -0,0 +1,43 @@
|
||||
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(notifications_admin):
|
||||
with notifications_admin.test_request_context():
|
||||
response = notifications_admin.test_client().get(
|
||||
url_for('main.sign_out'))
|
||||
assert response.status_code == 302
|
||||
assert response.location == url_for(
|
||||
'main.render_sign_in', _external=True, next=url_for('main.sign_out'))
|
||||
|
||||
|
||||
def test_sign_out_user(notifications_admin,
|
||||
notifications_admin_db,
|
||||
notify_db_session,
|
||||
mocker):
|
||||
with notifications_admin.test_request_context():
|
||||
_set_up_mocker(mocker)
|
||||
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 notifications_admin.test_client() as client:
|
||||
client.login(user)
|
||||
# Check we are logged in
|
||||
response = client.get(
|
||||
url_for('main.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.render_sign_in', _external=True)
|
||||
Reference in New Issue
Block a user