fix: add timeouts to unit tests

googleapis · Dec 11, 2019 · c10002e · c10002e
1 parent f499e7e
commit c10002e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git 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
@@ -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)