From d7f6bf3693e589d5edbae2ff2fd580180ce9afe2 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Tue, 12 May 2026 05:33:01 +0100
Subject: [PATCH] test: guard gateway status mock calls

---
 src/commands/gateway-status/output.test.ts | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/src/commands/gateway-status/output.test.ts b/src/commands/gateway-status/output.test.ts
index 0aa6f457348..73805f9034c 100644
--- a/src/commands/gateway-status/output.test.ts
+++ b/src/commands/gateway-status/output.test.ts
@@ -33,6 +33,15 @@ function createRuntimeCapture(): RuntimeEnv {
   } as unknown as RuntimeEnv;
 }
 
+function requireRuntimeJsonPayload(runtime: RuntimeEnv, index = 0): unknown {
+  const call = mocks.writeRuntimeJson.mock.calls.at(index);
+  if (!call) {
+    throw new Error(`expected writeRuntimeJson call ${index}`);
+  }
+  expect(call[0]).toBe(runtime);
+  return call[1];
+}
+
 function createProbe(
   capability: GatewayProbeResult["auth"]["capability"],
   params: {
@@ -144,10 +153,7 @@ describe("gateway status output", () => {
     });
 
     expect(mocks.writeRuntimeJson).toHaveBeenCalledOnce();
-    expect(mocks.writeRuntimeJson.mock.calls[0]?.[0]).toBe(runtime);
-    const payload = mocks.writeRuntimeJson.mock.calls[0]?.[1] as
-      | { ok?: unknown; capability?: unknown }
-      | undefined;
+    const payload = requireRuntimeJsonPayload(runtime) as { ok?: unknown; capability?: unknown };
     expect(payload?.ok).toBe(true);
     expect(payload?.capability).toBe("read_only");
   });
@@ -218,8 +224,7 @@ describe("gateway status output", () => {
     });
 
     expect(mocks.writeRuntimeJson).toHaveBeenCalledOnce();
-    expect(mocks.writeRuntimeJson.mock.calls[0]?.[0]).toBe(runtime);
-    const payload = mocks.writeRuntimeJson.mock.calls[0]?.[1];
+    const payload = requireRuntimeJsonPayload(runtime);
     expect(payload).toStrictEqual({
       ok: true,
       degraded: true,