http: handle ChunkedEncodingError

src/surykatka/http.py

@@ -74,7 +74,10 @@ def request(url, timeout=TIMEOUT, headers=None, session=requests, version=0):
     except requests.exceptions.ConnectionError:
         response = requests.models.Response()
         response.status_code = 523
-    except requests.exceptions.Timeout:
+    except (
+        requests.exceptions.Timeout,
+        requests.exceptions.ChunkedEncodingError,
+    ):
         response = requests.models.Response()
         response.status_code = 524
     except requests.exceptions.TooManyRedirects:

tests/test_http.py

@@ -152,6 +152,19 @@ class SurykatkaHttpTestCase(unittest.TestCase):
             assert mock_request.call_count == 1
             assert response.status_code == 524, response.status_code
 
+    def test_request_ChunkedEncodingError(self):
+        url_to_proxy = "http://example.org/"
+        httpretty.register_uri(httpretty.GET, url_to_proxy)
+        with mock.patch("surykatka.http.requests.request") as mock_request:
+
+            def sideEffect(*args, **kw):
+                raise surykatka.http.requests.exceptions.ChunkedEncodingError()
+
+            mock_request.side_effect = sideEffect
+            response = request(url_to_proxy)
+            assert mock_request.call_count == 1
+            assert response.status_code == 524, response.status_code
+
     def test_request_tooManyRedirect(self):
         url_to_proxy = "http://example.org/"
         httpretty.register_uri(httpretty.GET, url_to_proxy)