Fix auth route response injection crash
This commit is contained in:
@@ -784,6 +784,15 @@ class AuthLoginRouteThrottleTests(unittest.TestCase):
|
||||
|
||||
self.commit_count += 1
|
||||
|
||||
@staticmethod
|
||||
def _response_stub() -> SimpleNamespace:
|
||||
"""Builds a minimal response object for direct route invocation."""
|
||||
|
||||
return SimpleNamespace(
|
||||
set_cookie=lambda *_args, **_kwargs: None,
|
||||
delete_cookie=lambda *_args, **_kwargs: None,
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def _request_stub(ip_address: str = "203.0.113.2", user_agent: str = "unit-test") -> SimpleNamespace:
|
||||
"""Builds request-like object containing client host and user-agent header fields."""
|
||||
@@ -810,6 +819,7 @@ class AuthLoginRouteThrottleTests(unittest.TestCase):
|
||||
auth_routes_module.login(
|
||||
payload=payload,
|
||||
request=self._request_stub(),
|
||||
response=self._response_stub(),
|
||||
session=session,
|
||||
)
|
||||
self.assertEqual(raised.exception.status_code, 429)
|
||||
@@ -839,6 +849,7 @@ class AuthLoginRouteThrottleTests(unittest.TestCase):
|
||||
auth_routes_module.login(
|
||||
payload=payload,
|
||||
request=self._request_stub(),
|
||||
response=self._response_stub(),
|
||||
session=session,
|
||||
)
|
||||
self.assertEqual(raised.exception.status_code, 429)
|
||||
@@ -876,6 +887,7 @@ class AuthLoginRouteThrottleTests(unittest.TestCase):
|
||||
response = auth_routes_module.login(
|
||||
payload=payload,
|
||||
request=self._request_stub(),
|
||||
response=self._response_stub(),
|
||||
session=session,
|
||||
)
|
||||
self.assertEqual(response.access_token, "session-token")
|
||||
@@ -893,6 +905,7 @@ class AuthLoginRouteThrottleTests(unittest.TestCase):
|
||||
auth_routes_module.login(
|
||||
payload=payload,
|
||||
request=self._request_stub(),
|
||||
response=self._response_stub(),
|
||||
session=session,
|
||||
)
|
||||
self.assertEqual(raised.exception.status_code, 503)
|
||||
|
||||
Reference in New Issue
Block a user