mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-15 07:18:58 -04:00
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from datetime import datetime
|
|
from flask import url_for
|
|
from app.main.dao import users_dao
|
|
from app.models import User
|
|
|
|
|
|
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,
|
|
mock_get_service):
|
|
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(
|
|
url_for('main.service_dashboard', service_id="123"))
|
|
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)
|