From fcdc10c86c2c1a5440029731b9e70c97d7b1ccf8 Mon Sep 17 00:00:00 2001 From: Peter Lamut Date: Thu, 30 Jan 2020 06:59:24 +0000 Subject: [PATCH] chore(bigquery): adjust test assertions to new default timeout (#10222) --- bigquery/tests/unit/test__http.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) 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(), )