diff --git a/bigquery/tests/unit/test__http.py b/bigquery/tests/unit/test__http.py index 1d8313776a0e..4da805d48c78 100644 --- 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(), )