Fix auth route response injection crash

This commit is contained in:
2026-03-01 21:43:09 -03:00
parent 26eae1a09b
commit 3cccf2e0e8
2 changed files with 16 additions and 4 deletions

View File

@@ -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)