Code checks and all tests passing.

This commit is contained in:
Nicholas Staples
2016-01-06 16:51:35 +00:00
parent 5905229714
commit 79c15ec9cf
3 changed files with 19 additions and 8 deletions

View File

@@ -10,4 +10,3 @@ from app.main import main
def sign_out():
logout_user()
return redirect(url_for('main.render_sign_in'))

View File

@@ -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)

View File

@@ -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"))