Copybara import of the project:

thiagoneves · copybara-github · commit 5fd6a448de61 · 2025-05-08T11:02:13.000-07:00
-- 0723b09 by Thiago Neves <thiagohneves@gmail.com>: fix(tests): use mock GCS client in artifact service tests to avoid real credentials COPYBARA_INTEGRATE_REVIEW=google#601 from thiagoneves:fix/mock-gcs-client-in-tests e7d1671 PiperOrigin-RevId: 756381115
diff --git a/tests/unittests/artifacts/test_artifact_service.py b/tests/unittests/artifacts/test_artifact_service.py
@@ -21,6 +21,8 @@
 from google.adk.artifacts import GcsArtifactService
 from google.adk.artifacts import InMemoryArtifactService
 from google.genai import types
+
+from unittest import mock
 import pytest
 
 Enum = enum.Enum
@@ -136,11 +138,10 @@ def list_blobs(self, bucket: MockBucket, prefix: Optional[str] = None):
 
 
 def mock_gcs_artifact_service():
-  """Creates a mock GCS artifact service for testing."""
-  service = GcsArtifactService(bucket_name="test_bucket")
-  service.storage_client = MockClient()
-  service.bucket = service.storage_client.bucket("test_bucket")
-  return service
+  with mock.patch("google.cloud.storage.Client", return_value=MockClient()):
+    service = GcsArtifactService(bucket_name="test_bucket")
+    service.bucket = service.storage_client.bucket("test_bucket")
+    return service
 
 
 def get_artifact_service(
