mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-09-06 07:58:24 -04:00
Code checks and all tests passing.
This commit is contained in:
@@ -10,4 +10,3 @@ from app.main import main
|
|||||||
def sign_out():
|
def sign_out():
|
||||||
logout_user()
|
logout_user()
|
||||||
return redirect(url_for('main.render_sign_in'))
|
return redirect(url_for('main.render_sign_in'))
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,17 @@
|
|||||||
def test_should_show_recent_jobs_on_dashboard(notifications_admin):
|
from tests.app.main import create_test_user
|
||||||
response = notifications_admin.test_client().get('/dashboard')
|
from flask import url_for
|
||||||
|
|
||||||
assert response.status_code == 200
|
|
||||||
assert 'Test message 1' in response.get_data(as_text=True)
|
def test_should_show_recent_jobs_on_dashboard(notifications_admin,
|
||||||
assert 'Asdfgg' in response.get_data(as_text=True)
|
notifications_admin_db,
|
||||||
|
notify_db_session):
|
||||||
|
with notifications_admin.test_request_context():
|
||||||
|
with notifications_admin.test_client() as client:
|
||||||
|
with client.session_transaction() as session:
|
||||||
|
user = create_test_user('active')
|
||||||
|
client.login(user)
|
||||||
|
response = client.get(url_for('main.dashboard'))
|
||||||
|
|
||||||
|
assert response.status_code == 200
|
||||||
|
assert 'Test message 1' in response.get_data(as_text=True)
|
||||||
|
assert 'Asdfgg' in response.get_data(as_text=True)
|
||||||
|
|||||||
@@ -13,14 +13,15 @@ from app.main.dao import verify_codes_dao
|
|||||||
|
|
||||||
from app import create_app, db
|
from app import create_app, db
|
||||||
|
|
||||||
|
|
||||||
class TestClient(FlaskClient):
|
class TestClient(FlaskClient):
|
||||||
def login(self, user):
|
def login(self, user):
|
||||||
# Skipping authentication here and just log them in
|
# Skipping authentication here and just log them in
|
||||||
with self.session_transaction() as session:
|
with self.session_transaction() as session:
|
||||||
session['user_id'] = user.id
|
session['user_id'] = user.id
|
||||||
verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='sms')
|
verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='sms')
|
||||||
response = self.post('/two-factor',
|
response = self.post(
|
||||||
data={'sms_code': '12345'})
|
url_for('main.process_two_factor'), data={'sms_code': '12345'})
|
||||||
|
|
||||||
def logout(self, user):
|
def logout(self, user):
|
||||||
self.get(url_for("main.logout"))
|
self.get(url_for("main.logout"))
|
||||||
|
|||||||
Reference in New Issue
Block a user