1 files changed, 2 insertions, 6 deletions
diff --git a/tests/api/test_errors.py b/tests/api/test_errors.py
index 25fa93b9d8..efa3addf00 100644
--- a/tests/api/test_errors.py
+++ b/tests/api/test_errors.py
@@ -33,18 +33,14 @@ class LimitExceededErrorTestCase(unittest.TestCase):
self.assertIn("needle", err.debug_context)
self.assertNotIn("needle", serialised)
- # Create a sub-class to avoid mutating the class-level property.
- class LimitExceededErrorHeaders(LimitExceededError):
- include_retry_after_header = True
-
def test_limit_exceeded_header(self) -> None:
- err = self.LimitExceededErrorHeaders(limiter_name="test", retry_after_ms=100)
+ err = LimitExceededError(limiter_name="test", retry_after_ms=100)
self.assertEqual(err.error_dict(None).get("retry_after_ms"), 100)
assert err.headers is not None
self.assertEqual(err.headers.get("Retry-After"), "1")
def test_limit_exceeded_rounding(self) -> None:
- err = self.LimitExceededErrorHeaders(limiter_name="test", retry_after_ms=3001)
+ err = LimitExceededError(limiter_name="test", retry_after_ms=3001)
self.assertEqual(err.error_dict(None).get("retry_after_ms"), 3001)
assert err.headers is not None
self.assertEqual(err.headers.get("Retry-After"), "4")
|