From 004d1935c8d0b1470bc4352c05c3d848dc4db5bc Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Sat, 9 May 2026 14:35:32 +0100
Subject: [PATCH] test: tighten before tool call extension assertion

---
 .../pi-tools.before-tool-call.integration.e2e.test.ts       | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/agents/pi-tools.before-tool-call.integration.e2e.test.ts b/src/agents/pi-tools.before-tool-call.integration.e2e.test.ts
index ad446753f3a..c083c16c8d3 100644
--- a/src/agents/pi-tools.before-tool-call.integration.e2e.test.ts
+++ b/src/agents/pi-tools.before-tool-call.integration.e2e.test.ts
@@ -503,7 +503,11 @@ describe("before_tool_call hook integration for client tools", () => {
           namespace: "policy",
           value: { gate: "client" },
         }),
-      ).resolves.toMatchObject({ ok: true });
+      ).resolves.toEqual({
+        ok: true,
+        key: "agent:main:client",
+        value: { gate: "client" },
+      });
 
       const [tool] = toClientToolDefinitions(
         [