Merge pull request #3415 from alphagov/normalise-2fa-code-input

Allow spaces and dashes in the two factor code
This commit is contained in:
Chris Hill-Scott
2020-04-21 10:14:19 +01:00
committed by GitHub
2 changed files with 11 additions and 0 deletions

View File

@@ -11,6 +11,8 @@ def _check_code(code):
@pytest.mark.parametrize('post_data', [
{'sms_code': '12345'},
{'sms_code': ' 12345 '},
{'sms_code': '12 34 5'},
{'sms_code': '1-23-45'},
])
def test_form_is_valid_returns_no_errors(
app_,
@@ -21,6 +23,7 @@ def test_form_is_valid_returns_no_errors(
form = TwoFactorForm(_check_code)
assert form.validate() is True
assert form.errors == {}
mock_check_verify_code.assert_called_once_with('1', '12345', 'sms')
@pytest.mark.parametrize('post_data, expected_error', (
@@ -40,6 +43,10 @@ def test_form_is_valid_returns_no_errors(
{'sms_code': '12E45'},
'Numbers only',
),
(
{'sms_code': ' ! 2 3 4 5'},
'Numbers only',
),
))
def test_check_verify_code_returns_errors(
app_,