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

Updated code used flow

Browse Source
This commit is contained in:
chrisw
2017-11-09 17:06:24 +00:00
parent 888ac5bf85
commit 04f6d0a5e0
2 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -276,7 +276,9 @@ def test_two_factor_email_link_is_invalid(
def test_two_factor_email_link_is_already_used(
client,
valid_token,
mocker
mocker,
mock_send_verify_code
):
mocker.patch('app.user_api_client.check_verify_code', return_value=(False, 'Code has expired'))
@@ -288,14 +290,15 @@ def test_two_factor_email_link_is_already_used(
page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser')
assert normalize_spaces(
page.select_one('.banner-dangerous').text
) == "There’s something wrong with the code"
) == "This link has already been used"
assert response.status_code == 200
def test_two_factor_email_link_when_user_is_locked_out(
client,
valid_token,
mocker
mocker,
mock_send_verify_code
):
mocker.patch('app.user_api_client.check_verify_code', return_value=(False, 'Code not found'))
@@ -307,7 +310,7 @@ def test_two_factor_email_link_when_user_is_locked_out(
page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser')
assert normalize_spaces(
page.select_one('.banner-dangerous').text
) == "There’s something wrong with the code"
) == "This link has already been used"
assert response.status_code == 200