From c107b9d574ad4df36396db05cc1aede417ede963 Mon Sep 17 00:00:00 2001
From: Chisanan232 <chi10211201@cycu.org.tw>
Date: Wed, 20 May 2026 23:57:55 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20(adapters/google=5Fadk):=20Add=20Go?=
 =?UTF-8?q?ogle=20ADK=20interception=20integration=20test?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Two tests:

* A two-tool async flow under a `monkeypatch`ed `google.adk.tools.BaseTool`
  that verifies a denied tool raises `PolicyViolationError("blocked by
  governance policy")` while a subsequent safe tool still executes its
  patched `run_async` path through `record_result`.
* A real-`BaseTool` patch-path test guarded by `pytest.importorskip` that
  asserts the patched flag is set after `apply()` and cleared after
  `revert()` against the live `google.adk.tools.BaseTool` class.

Test marked `@pytest.mark.integration`; skips cleanly without `google-adk`.

Refs: AAASM-1678
---
 ...est_google_adk_interception_integration.py | 73 +++++++++++++++++++
 1 file changed, 73 insertions(+)
 create mode 100644 test/integration/test_google_adk_interception_integration.py

diff --git a/test/integration/test_google_adk_interception_integration.py b/test/integration/test_google_adk_interception_integration.py
new file mode 100644
index 0000000..6ee06ca
--- /dev/null
+++ b/test/integration/test_google_adk_interception_integration.py
@@ -0,0 +1,73 @@
+from __future__ import annotations
+
+from types import SimpleNamespace
+from typing import Any
+
+import pytest
+
+from agent_assembly.adapters.google_adk import patch as google_adk_patch
+from agent_assembly.exceptions import PolicyViolationError
+
+
+@pytest.mark.asyncio
+@pytest.mark.integration
+async def test_google_adk_two_tool_flow_continues_after_blocked_tool(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    class FakeBaseTool:
+        def __init__(self, name: str, result: str) -> None:
+            self.name = name
+            self._result = result
+
+        async def run_async(self, *, args: Any, tool_context: Any, **kwargs: Any) -> str:
+            del args, tool_context, kwargs
+            return self._result
+
+    fake_google_adk_tools = SimpleNamespace(BaseTool=FakeBaseTool)
+
+    def fake_import_module(module_name: str) -> object:
+        if module_name == "google.adk.tools":
+            return fake_google_adk_tools
+        raise ImportError(module_name)
+
+    monkeypatch.setattr(google_adk_patch.importlib, "import_module", fake_import_module)
+
+    class Interceptor:
+        async def check_tool_start(self, **kwargs: object) -> dict[str, str]:
+            if kwargs.get("tool_name") == "blocked_tool":
+                return {"status": "deny", "reason": "blocked by policy"}
+            return {"status": "allow"}
+
+    patcher = google_adk_patch.GoogleADKPatch(Interceptor())
+    assert patcher.apply() is True
+
+    tool_context = SimpleNamespace(
+        invocation_context=SimpleNamespace(assembly_agent_id="agent-1", invocation_id="run-main"),
+    )
+    blocked_tool = FakeBaseTool("blocked_tool", "should-not-run")
+    safe_tool = FakeBaseTool("safe_tool", "ok:safe_tool")
+
+    with pytest.raises(PolicyViolationError, match="blocked by governance policy: blocked by policy"):
+        await blocked_tool.run_async(args={"step": 1}, tool_context=tool_context)
+
+    safe_result = await safe_tool.run_async(args={"step": 2}, tool_context=tool_context)
+    assert safe_result == "ok:safe_tool"
+
+
+@pytest.mark.integration
+def test_google_adk_real_base_tool_class_patch_path_when_available() -> None:
+    google_adk_tools = pytest.importorskip("google.adk.tools")
+    tool_cls = google_adk_tools.BaseTool
+
+    class Interceptor:
+        async def check_tool_start(self, **kwargs: object) -> dict[str, str]:
+            del kwargs
+            return {"status": "allow"}
+
+    patcher = google_adk_patch.GoogleADKPatch(Interceptor())
+    try:
+        assert patcher.apply() is True
+        assert getattr(tool_cls, google_adk_patch._TOOLS_PATCHED_FLAG, False) is True
+    finally:
+        patcher.revert()
+    assert getattr(tool_cls, google_adk_patch._TOOLS_PATCHED_FLAG, False) is False