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

vertex-sdk-bot · copybara-github · commit 945b9e483514 · 2024-06-04T16:05:36.000-07:00
PiperOrigin-RevId: 640313157
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
