darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-08-25 08:44:23 -04:00
Code Issues Packages Projects Releases Wiki Activity

Create unit tests that test the forms.

Browse Source
This commit is contained in:
Rebecca Law
2015-12-14 10:52:00 +00:00
parent c0550d2c61
commit 295dbeb7d1
6 changed files with 141 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/app/main/views/test_two_factor.py
View File

@@ -58,5 +58,4 @@ def test_should_return_400_when_sms_code_is_too_short(notifications_admin, notif
assert response.status_code == 400
data = json.loads(response.get_data(as_text=True))
assert len(data.keys()) == 1
assert 'sms_code' in data
assert set(data['sms_code']) == ['Code must be 5 digits', 'Code does not match'].sort()
assert set(data) == set({'sms_code': ['Code must be 5 digits', 'Code does not match']})

11
tests/app/main/views/test_verify.py
View File

@@ -1,7 +1,4 @@
from datetime import datetime, timedelta
from flask import json
from app.main.dao import users_dao, verify_codes_dao
from tests.app.main import create_test_user
@@ -55,11 +52,7 @@ def test_should_return_400_when_codes_are_wrong(notifications_admin, notificatio
'email_code': '23456'})
assert response.status_code == 400
expected = {'sms_code': ['Code must be 5 digits', 'Code does not match'],
'email_code': ['Code must be 5 digits', 'Code does not match']}
'email_code': ['Code must be 5 digits', 'Code does not match']}
errors = json.loads(response.get_data(as_text=True))
assert len(errors) == 2
assert 'sms_code' in errors
assert errors['sms_code'] == expected['sms_code']
assert 'email_code' in errors
assert set(errors['email_code']) in set(expected['email_code'])
assert set(errors) == set(expected)