Mock external calls in one of BigQuery unit tests

googleapis · Jul 22, 2019 · e1192c2 · e1192c2
1 parent 951b9c6
commit e1192c2
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/bigquery/tests/unit/test_magics.py b/bigquery/tests/unit/test_magics.py
@@ -559,9 +559,17 @@ def test_bigquery_magic_w_maximum_bytes_billed_invalid():
     ip.extension_manager.load_extension("google.cloud.bigquery")
     magics.context._project = None
 
+    credentials_mock = mock.create_autospec(
+        google.auth.credentials.Credentials, instance=True
+    )
+    default_patch = mock.patch(
+        "google.auth.default", return_value=(credentials_mock, "general-project")
+    )
+    client_query_patch = mock.patch("google.cloud.bigquery.client.Client.query")
+
     sql = "SELECT 17 AS num"
 
-    with pytest.raises(ValueError):
+    with pytest.raises(ValueError), default_patch, client_query_patch:
         ip.run_cell_magic("bigquery", "--maximum_bytes_billed=abc", sql)