mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-19 14:09:20 -04:00
Code checks and all tests passing.
This commit is contained in:
@@ -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