feat: retry google.auth TransportError by default (#624)

plamut · web-flow · commit 34ecc3f1ca0f · 2021-04-23T15:54:32.000-05:00
diff --git a/google/cloud/bigquery/retry.py b/google/cloud/bigquery/retry.py
@@ -14,6 +14,7 @@
 
 from google.api_core import exceptions
 from google.api_core import retry
+from google.auth import exceptions as auth_exceptions
 import requests.exceptions
 
 
@@ -27,6 +28,7 @@
     exceptions.InternalServerError,
     exceptions.BadGateway,
     requests.exceptions.ConnectionError,
+    auth_exceptions.TransportError,
 )
 
 
diff --git a/tests/unit/test_retry.py b/tests/unit/test_retry.py
@@ -51,6 +51,12 @@ def test_w_unstructured_requests_connectionerror(self):
         exc = requests.exceptions.ConnectionError()
         self.assertTrue(self._call_fut(exc))
 
+    def test_w_auth_transporterror(self):
+        from google.auth.exceptions import TransportError
+
+        exc = TransportError("testing")
+        self.assertTrue(self._call_fut(exc))
+
     def test_w_unstructured_too_many_requests(self):
         from google.api_core.exceptions import TooManyRequests
 
