Merge branch 'connection-426' of github.com:googleapis/python-storage…

… into connection-426
googleapis · May 19, 2021 · 581fdce · 581fdce
2 parents 1416ce7 + bcfe2be
commit 581fdce
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/tests/unit/test_retry.py b/tests/unit/test_retry.py
@@ -65,16 +65,13 @@ def test_miss_w_stdlib_error(self):
         exc = ValueError("testing")
         self.assertFalse(self._call_fut(exc))
 
+    @unittest.skipUnless(
+        "ConnectionError" in __builtins__.__dict__,
+        "No builtin 'ConnectionError' in Python 2",
+    )
     def test_w_stdlib_connection_error(self):
-        from google.cloud.storage import retry
-
-        try:
-            exc = ConnectionError()
-            self.assertTrue(self._call_fut(exc))
-            self.assertTrue(ConnectionError in retry._RETRYABLE_TYPES)
-        except NameError:
-            pass
-
+        exc = ConnectionError()
+        self.assertTrue(self._call_fut(exc))
 
 class TestConditionalRetryPolicy(unittest.TestCase):
     def _make_one(self, retry_policy, conditional_predicate, required_kwargs):