Merge pull request d3v-null#12 from december1981/master

derwentx · web-flow · commit bfeb9832dad7 · 2019-05-05T12:19:21.000+10:00
Allow a wordpress api request to specify certain status codes it want…
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -438,6 +438,24 @@ def test_APIPostBadData(self):
         with self.assertRaises(UserWarning):
             self.wpapi.post('posts', data)
 
+    def test_APIPostBadDataHandleBadStatus(self):
+        """
+        Test handling explicitly a bad status code for a request.
+        """
+        nonce = "%f\u00ae" % random.random()
+
+        data = {
+            'a': nonce
+        }
+
+        response = self.wpapi.post('posts', data, handle_status_codes=[400])
+        self.assertEqual(response.status_code, 400)
+
+        # If we don't specify a correct status code to handle we should
+        # still expect an exception
+        with self.assertRaises(UserWarning):
+            self.wpapi.post('posts', data, handle_status_codes=[404])
+
     def test_APIPostMedia(self):
         img_path = 'tests/data/test.jpg'
         with open(img_path, 'rb') as test_file:
diff --git a/wordpress/api.py b/wordpress/api.py
@@ -219,6 +219,8 @@ def __request(self, method, endpoint, data, **kwargs):
             # enforce utf-8 encoded binary
             data = StrUtils.to_binary(data)
 
+        handle_status_codes = kwargs.pop('handle_status_codes', [])
+
         response = self.requester.request(
             method=method,
             url=endpoint_url,
@@ -227,7 +229,7 @@ def __request(self, method, endpoint, data, **kwargs):
             **kwargs
         )
 
-        if response.status_code not in [200, 201, 202]:
+        if response.status_code not in [200, 201, 202] + handle_status_codes:
             self.request_post_mortem(response)
 
         return response
