From 4ff637772994518fba53c6b3b2ef3f7c2c9073b4 Mon Sep 17 00:00:00 2001 From: Tres Seaver Date: Tue, 11 Sep 2018 15:37:03 -0400 Subject: [PATCH] Retry '502 Bad Gateway' errors by default. Closes #5918. --- bigquery/google/cloud/bigquery/retry.py | 1 + bigquery/tests/unit/test_retry.py | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/bigquery/google/cloud/bigquery/retry.py b/bigquery/google/cloud/bigquery/retry.py index fc2eebeecfad..be279171a479 100644 --- a/bigquery/google/cloud/bigquery/retry.py +++ b/bigquery/google/cloud/bigquery/retry.py @@ -20,6 +20,7 @@ 'backendError', 'rateLimitExceeded', 'internalError', + 'badGateway', ]) diff --git a/bigquery/tests/unit/test_retry.py b/bigquery/tests/unit/test_retry.py index 4c93b3b1cbdd..9f4987fe5a54 100644 --- a/bigquery/tests/unit/test_retry.py +++ b/bigquery/tests/unit/test_retry.py @@ -51,3 +51,8 @@ def test_w_internalError(self): exc = mock.Mock( errors=[{'reason': 'internalError'}], spec=['errors']) self.assertTrue(self._call_fut(exc)) + + def test_w_badGateway(self): + exc = mock.Mock( + errors=[{'reason': 'badGateway'}], spec=['errors']) + self.assertTrue(self._call_fut(exc))