diff --git a/bigquery/tests/unit/test__http.py b/bigquery/tests/unit/test__http.py index 629e8031a52f..1d8313776a0e 100644 --- a/bigquery/tests/unit/test__http.py +++ b/bigquery/tests/unit/test__http.py @@ -75,7 +75,11 @@ def test_user_agent(self): } expected_uri = conn.build_api_url("/rainbow") http.request.assert_called_once_with( - data=req_data, headers=expected_headers, method="GET", url=expected_uri + data=req_data, + headers=expected_headers, + method="GET", + url=expected_uri, + timeout=None, ) self.assertIn("my-application/1.2.3", conn.user_agent) @@ -104,5 +108,9 @@ def test_extra_headers_replace(self): } expected_uri = conn.build_api_url("/rainbow") http.request.assert_called_once_with( - data=req_data, headers=expected_headers, method="GET", url=expected_uri + data=req_data, + headers=expected_headers, + method="GET", + url=expected_uri, + timeout=None, ) diff --git a/bigquery/tests/unit/test_client.py b/bigquery/tests/unit/test_client.py index 8a82b2e19aa8..e661c86970db 100644 --- a/bigquery/tests/unit/test_client.py +++ b/bigquery/tests/unit/test_client.py @@ -1552,6 +1552,7 @@ def test_get_table_sets_user_agent(self): "User-Agent": expected_user_agent, }, data=mock.ANY, + timeout=None, ) self.assertIn("my-application/1.2.3", expected_user_agent)