Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
feat: retry google.auth TransportError by default (#624)
  • Loading branch information
plamut committed Apr 23, 2021
1 parent 6ee5824 commit 34ecc3f
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 0 deletions.
2 changes: 2 additions & 0 deletions google/cloud/bigquery/retry.py
Expand Up @@ -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


Expand All @@ -27,6 +28,7 @@
exceptions.InternalServerError,
exceptions.BadGateway,
requests.exceptions.ConnectionError,
auth_exceptions.TransportError,
)


Expand Down
6 changes: 6 additions & 0 deletions tests/unit/test_retry.py
Expand Up @@ -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

Expand Down

0 comments on commit 34ecc3f

Please sign in to comment.