fix: Fix failed unit tests due to google-cloud-storage upgrade.

PiperOrigin-RevId: 640313157
googleapis · Jun 4, 2024 · 945b9e4 · 945b9e4
1 parent f9d4b51
commit 945b9e4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/tests/unit/aiplatform/test_metadata.py b/tests/unit/aiplatform/test_metadata.py
@@ -91,7 +91,10 @@
 _TEST_RUN = "run-1"
 _TEST_OTHER_RUN = "run-2"
 _TEST_DISPLAY_NAME = "test-display-name"
-_TEST_CREDENTIALS = mock.Mock(spec=credentials.AnonymousCredentials())
+_TEST_CREDENTIALS = mock.Mock(
+    spec=credentials.AnonymousCredentials(),
+    universe_domain="googleapis.com",
+)
 _TEST_BUCKET_NAME = "gs://test-bucket"
 
 # resource attributes

diff --git a/tests/unit/aiplatform/test_metadata_models.py b/tests/unit/aiplatform/test_metadata_models.py
@@ -50,7 +50,10 @@
 _TEST_PARENT = (
     f"projects/{_TEST_PROJECT}/locations/{_TEST_LOCATION}/metadataStores/default"
 )
-_TEST_CREDENTIALS = mock.Mock(spec=auth_credentials.AnonymousCredentials())
+_TEST_CREDENTIALS = mock.Mock(
+    spec=auth_credentials.AnonymousCredentials(),
+    universe_domain="googleapis.com",
+)
 
 
 # artifact