fix: retry ChunkedEncodingError by default (#802)

googleapis · Jul 23, 2021 · 419d36d · 419d36d
1 parent 7e0e2ba
commit 419d36d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/google/cloud/bigquery/retry.py b/google/cloud/bigquery/retry.py
@@ -27,6 +27,7 @@
     exceptions.TooManyRequests,
     exceptions.InternalServerError,
     exceptions.BadGateway,
+    requests.exceptions.ChunkedEncodingError,
     requests.exceptions.ConnectionError,
     auth_exceptions.TransportError,
 )

diff --git a/tests/unit/test_retry.py b/tests/unit/test_retry.py
@@ -51,6 +51,10 @@ def test_w_unstructured_requests_connectionerror(self):
         exc = requests.exceptions.ConnectionError()
         self.assertTrue(self._call_fut(exc))
 
+    def test_w_unstructured_requests_chunked_encoding_error(self):
+        exc = requests.exceptions.ChunkedEncodingError()
+        self.assertTrue(self._call_fut(exc))
+
     def test_w_auth_transporterror(self):
         from google.auth.exceptions import TransportError