No public description

Co-authored-by: Yifan Wang <wanyif@google.com>
PiperOrigin-RevId: 927466558

Author: wyf7107

tests/unittests/test_optional_dependencies.py

@@ -20,6 +20,7 @@
 
 from __future__ import annotations
 
+import importlib.util
 import os
 from pathlib import Path
 import subprocess
@@ -167,6 +168,24 @@ def test_vertex_ai_session_service_fails_on_creation():
     assert "google-cloud-aiplatform" in str(exc_info.value)
 
 
+def test_vertexai_dependency_shim_raises_clear_importerror():
+  """Verify that the Vertex AI dependency shim points users to the dependency."""
+  with mock.patch.dict("sys.modules", {"google.cloud.aiplatform": None}):
+    module_path = _REPO_ROOT / "dependencies_internal/vertexai.py"
+    spec = importlib.util.spec_from_file_location(
+        "_test_google_adk_dependencies_vertexai", module_path
+    )
+    assert spec is not None
+    assert spec.loader is not None
+    module = importlib.util.module_from_spec(spec)
+
+    with pytest.raises(ImportError) as exc_info:
+      spec.loader.exec_module(module)
+
+    message = str(exc_info.value)
+    assert "//third_party/py/google/cloud/aiplatform" in message
+
+
 # =============================================================================
 # Approach 2: High-Fidelity Integration Tests (Clean Venv, Skipped by Default)
 # =============================================================================