From 60214e3963ea655839a225da67cfbfdd74dfe75d Mon Sep 17 00:00:00 2001
From: Shakker <shakkerdroid@gmail.com>
Date: Mon, 11 May 2026 01:51:37 +0100
Subject: [PATCH] test: tighten exec approval followup handoff assertion

---
 .../bash-tools.exec-approval-followup.test.ts | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/src/agents/bash-tools.exec-approval-followup.test.ts b/src/agents/bash-tools.exec-approval-followup.test.ts
index 4d1d69051c1..da42ba65b09 100644
--- a/src/agents/bash-tools.exec-approval-followup.test.ts
+++ b/src/agents/bash-tools.exec-approval-followup.test.ts
@@ -43,6 +43,7 @@ function expectGatewayAgentFollowup(expected: Record<string, unknown>) {
     expect(params[key]).toBe(value);
   }
   expect(call[3]).toEqual({ expectFinal: true });
+  return params;
 }
 
 function expectDirectSend(expected: Record<string, unknown>) {
@@ -293,18 +294,12 @@ describe("exec approval followup", () => {
       idempotencyKey: "exec-approval-followup:req-elevated-75832:nonce:nonce-75832",
     });
 
-    expect(callGatewayTool).toHaveBeenCalledWith(
-      "agent",
-      expect.any(Object),
-      expect.objectContaining({
-        sessionKey: "agent:main:telegram:direct:123",
-        channel: "telegram",
-        idempotencyKey: "exec-approval-followup:req-elevated-75832:nonce:nonce-75832",
-        internalRuntimeHandoffId: "handoff-75832",
-      }),
-      { expectFinal: true },
-    );
-    const [, , agentArgs] = vi.mocked(callGatewayTool).mock.calls[0] ?? [];
+    const agentArgs = expectGatewayAgentFollowup({
+      sessionKey: "agent:main:telegram:direct:123",
+      channel: "telegram",
+      idempotencyKey: "exec-approval-followup:req-elevated-75832:nonce:nonce-75832",
+      internalRuntimeHandoffId: "handoff-75832",
+    });
     expect(agentArgs).not.toHaveProperty("bashElevated");
     expect(agentArgs).not.toHaveProperty("execApprovalFollowupToken");
   });