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

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