fix: add 'requests.exceptions.ChunkedEncodingError' to retryable exce…

…ptions (#237) Closes #235.
googleapis · Jul 23, 2021 · 5e540f2 · 5e540f2
1 parent 3d4cab2
commit 5e540f2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/google/api_core/retry.py b/google/api_core/retry.py
@@ -104,6 +104,7 @@ def if_exception_type_predicate(exception):
     exceptions.TooManyRequests,
     exceptions.ServiceUnavailable,
     requests.exceptions.ConnectionError,
+    requests.exceptions.ChunkedEncodingError,
     auth_exceptions.TransportError,
 )
 """A predicate that checks if an exception is a transient API error.

diff --git a/tests/unit/test_retry.py b/tests/unit/test_retry.py
@@ -45,6 +45,7 @@ def test_if_transient_error():
     assert retry.if_transient_error(exceptions.TooManyRequests(""))
     assert retry.if_transient_error(exceptions.ServiceUnavailable(""))
     assert retry.if_transient_error(requests.exceptions.ConnectionError(""))
+    assert retry.if_transient_error(requests.exceptions.ChunkedEncodingError(""))
     assert retry.if_transient_error(auth_exceptions.TransportError(""))
     assert not retry.if_transient_error(exceptions.InvalidArgument(""))