Reject throttling status code (#123)

anna-money · Feb 9, 2022 · 008a5ac · 008a5ac
1 parent 286b701
commit 008a5ac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/aio_request/response_classifier.py b/aio_request/response_classifier.py
@@ -40,4 +40,6 @@ def classify(self, response: Response) -> ResponseVerdict:
             return ResponseVerdict.REJECT
         if response.status == 408:
             return ResponseVerdict.REJECT
+        if response.status == 429:
+            return ResponseVerdict.REJECT
         return ResponseVerdict.ACCEPT
diff --git a/tests/test_response_classifier.py b/tests/test_response_classifier.py
@@ -10,8 +10,8 @@
     [
         (408, None, aio_request.ResponseVerdict.REJECT),
         (400, None, aio_request.ResponseVerdict.ACCEPT),
-        (429, None, aio_request.ResponseVerdict.ACCEPT),
-        (429, {429: aio_request.ResponseVerdict.REJECT}, aio_request.ResponseVerdict.REJECT),
+        (429, None, aio_request.ResponseVerdict.REJECT),
+        (429, {429: aio_request.ResponseVerdict.ACCEPT}, aio_request.ResponseVerdict.ACCEPT),
     ],
 )
 def test_default_response_classifier(