From a4b3aeeefab8cb78c40c24d4114e4a8bcc91d601 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Sat, 21 Feb 2026 22:07:20 +0000
Subject: [PATCH] test(gateway): reuse last agent command assertion helper

---
 src/gateway/server-methods/agent.test.ts | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/src/gateway/server-methods/agent.test.ts b/src/gateway/server-methods/agent.test.ts
index 262000578..c1e36a99e 100644
--- a/src/gateway/server-methods/agent.test.ts
+++ b/src/gateway/server-methods/agent.test.ts
@@ -139,6 +139,17 @@ async function runMainAgent(message: string, idempotencyKey: string) {
   return respond;
 }
 
+function readLastAgentCommandCall():
+  | {
+      message?: string;
+      sessionId?: string;
+    }
+  | undefined {
+  return mocks.agentCommand.mock.calls.at(-1)?.[0] as
+    | { message?: string; sessionId?: string }
+    | undefined;
+}
+
 async function invokeAgent(
   params: AgentParams,
   options?: {
@@ -338,9 +349,7 @@ describe("gateway agent handler", () => {
 
     await vi.waitFor(() => expect(mocks.agentCommand).toHaveBeenCalled());
     expect(mocks.sessionsResetHandler).toHaveBeenCalledTimes(1);
-    const call = mocks.agentCommand.mock.calls.at(-1)?.[0] as
-      | { message?: string; sessionId?: string }
-      | undefined;
+    const call = readLastAgentCommandCall();
     expect(call?.message).toBe(BARE_SESSION_RESET_PROMPT);
     expect(call?.message).toContain("Execute your Session Startup sequence now");
     expect(call?.sessionId).toBe("reset-session-id");
@@ -388,9 +397,7 @@ describe("gateway agent handler", () => {
 
     await vi.waitFor(() => expect(mocks.agentCommand).toHaveBeenCalled());
     expect(mocks.sessionsResetHandler).toHaveBeenCalledTimes(1);
-    const call = mocks.agentCommand.mock.calls.at(-1)?.[0] as
-      | { message?: string; sessionId?: string }
-      | undefined;
+    const call = readLastAgentCommandCall();
     expect(call?.message).toBe("[Wed 2026-01-28 20:30 EST] check status");
     expect(call?.sessionId).toBe("reset-session-id");