mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-05-15 09:07:02 +02:00
change StateMismatch to inherit AuthenticationFailed
This commit is contained in:
parent
3ea0d90853
commit
d2c6a2df8b
2 changed files with 8 additions and 6 deletions
|
@ -33,7 +33,7 @@ class AuthenticationFailed(OIDCException):
|
|||
friendly_message = "This login attempt didn't work."
|
||||
|
||||
|
||||
class StateMismatch(OIDCException):
|
||||
class StateMismatch(AuthenticationFailed):
|
||||
friendly_message = "State mismatch. This login attempt didn't work."
|
||||
|
||||
|
||||
|
|
|
@ -132,10 +132,11 @@ class ViewsTest(TestCase):
|
|||
"""If the local session does not match the OP session,
|
||||
we do not throw an exception. Rather, we attempt to login again."""
|
||||
with less_console_noise():
|
||||
# MOCK
|
||||
# MOCK get_default_acr_value and the callback to raise StateMismatch
|
||||
# error when called
|
||||
mock_client.get_default_acr_value.side_effect = self.create_acr
|
||||
mock_client.callback.side_effect = StateMismatch()
|
||||
# TEST
|
||||
# TEST receiving a response from login.gov
|
||||
response = self.client.get(reverse("openid_login_callback"))
|
||||
# ASSERT
|
||||
self.assertEqual(response.status_code, 302)
|
||||
|
@ -146,16 +147,17 @@ class ViewsTest(TestCase):
|
|||
def test_login_callback_with_no_session_state_attempt_again_only_once(self, mock_client):
|
||||
"""We only attempt to relogin once. After that, it's the error page for you."""
|
||||
with less_console_noise():
|
||||
# MOCK
|
||||
# MOCK get_default_acr_value, redirect_attempted to True and the callback
|
||||
# to raise StateMismatch error when called
|
||||
mock_client.get_default_acr_value.side_effect = self.create_acr
|
||||
mock_client.callback.side_effect = StateMismatch()
|
||||
session = self.client.session
|
||||
session["redirect_attempted"] = True
|
||||
session.save()
|
||||
# TEST
|
||||
# TEST receiving a response from login.gov
|
||||
response = self.client.get(reverse("openid_login_callback"))
|
||||
# ASSERT
|
||||
self.assertEqual(response.status_code, 500)
|
||||
self.assertEqual(response.status_code, 401)
|
||||
|
||||
def test_login_callback_reads_next(self, mock_client):
|
||||
"""If the next value is set in the session, test that login_callback returns
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue