mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 02:42:26 -05:00
Code checks and all tests passing.
This commit is contained in:
@@ -10,4 +10,3 @@ from app.main import main
|
||||
def sign_out():
|
||||
logout_user()
|
||||
return redirect(url_for('main.render_sign_in'))
|
||||
|
||||
|
||||
@@ -1,6 +1,17 @@
|
||||
def test_should_show_recent_jobs_on_dashboard(notifications_admin):
|
||||
response = notifications_admin.test_client().get('/dashboard')
|
||||
from tests.app.main import create_test_user
|
||||
from flask import url_for
|
||||
|
||||
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)
|
||||
|
||||
def test_should_show_recent_jobs_on_dashboard(notifications_admin,
|
||||
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
|
||||
|
||||
|
||||
class TestClient(FlaskClient):
|
||||
def login(self, user):
|
||||
# Skipping authentication here and just log them in
|
||||
with self.session_transaction() as session:
|
||||
session['user_id'] = user.id
|
||||
verify_codes_dao.add_code(user_id=user.id, code='12345', code_type='sms')
|
||||
response = self.post('/two-factor',
|
||||
data={'sms_code': '12345'})
|
||||
response = self.post(
|
||||
url_for('main.process_two_factor'), data={'sms_code': '12345'})
|
||||
|
||||
def logout(self, user):
|
||||
self.get(url_for("main.logout"))
|
||||
|
||||
Reference in New Issue
Block a user