chore(bigquery): adjust test assertions to new default timeout (#10222)

googleapis · Jan 30, 2020 · fcdc10c · fcdc10c
1 parent 79d8e00
commit fcdc10c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/bigquery/tests/unit/test__http.py b/bigquery/tests/unit/test__http.py
@@ -19,6 +19,12 @@
 
 
 class TestConnection(unittest.TestCase):
+    @staticmethod
+    def _get_default_timeout():
+        from google.cloud.bigquery._http import _http
+
+        return _http._DEFAULT_TIMEOUT
+
     @staticmethod
     def _get_target_class():
         from google.cloud.bigquery._http import Connection
@@ -79,7 +85,7 @@ def test_user_agent(self):
             headers=expected_headers,
             method="GET",
             url=expected_uri,
-            timeout=None,
+            timeout=self._get_default_timeout(),
         )
         self.assertIn("my-application/1.2.3", conn.user_agent)
 
@@ -112,5 +118,5 @@ def test_extra_headers_replace(self):
             headers=expected_headers,
             method="GET",
             url=expected_uri,
-            timeout=None,
+            timeout=self._get_default_timeout(),
         )