From f2b560ff5dff2b0a10400a876b6e7d5bde2f082b Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Wed, 13 May 2026 00:16:04 +0100
Subject: [PATCH] test: dedupe session restart mock reads

---
 src/auto-reply/reply/commands-session-restart.test.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/auto-reply/reply/commands-session-restart.test.ts b/src/auto-reply/reply/commands-session-restart.test.ts
index 08087e83ab0..19472412b88 100644
--- a/src/auto-reply/reply/commands-session-restart.test.ts
+++ b/src/auto-reply/reply/commands-session-restart.test.ts
@@ -107,6 +107,10 @@ function restartCommandParams(overrides?: Partial<HandleCommandsParams>): Handle
   } as HandleCommandsParams;
 }
 
+function firstRestartSentinelPayload() {
+  return mocks.writeRestartSentinel.mock.calls[0]?.[0];
+}
+
 describe("handleRestartCommand", () => {
   beforeEach(() => {
     mocks.isRestartEnabled.mockReset();
@@ -128,7 +132,7 @@ describe("handleRestartCommand", () => {
 
     expect(result?.shouldContinue).toBe(false);
     expect(mocks.writeRestartSentinel).toHaveBeenCalledOnce();
-    const sentinelPayload = mocks.writeRestartSentinel.mock.calls.at(0)?.[0];
+    const sentinelPayload = firstRestartSentinelPayload();
     expect(sentinelPayload?.kind).toBe("restart");
     expect(sentinelPayload?.status).toBe("ok");
     expect(typeof sentinelPayload?.ts).toBe("number");
@@ -166,7 +170,7 @@ describe("handleRestartCommand", () => {
       await scheduledArgs?.emitHooks?.beforeEmit?.();
 
       expect(mocks.writeRestartSentinel).toHaveBeenCalledOnce();
-      const sentinelPayload = mocks.writeRestartSentinel.mock.calls.at(0)?.[0];
+      const sentinelPayload = firstRestartSentinelPayload();
       expect(sentinelPayload?.kind).toBe("restart");
       expect(sentinelPayload?.status).toBe("ok");
       expect(sentinelPayload?.sessionKey).toBe("agent:main:telegram:direct:123:thread:thread-1");