From 44916fc9b2958a941168e8982016ac160327c8a8 Mon Sep 17 00:00:00 2001
From: William Valentin <william.valentin.info@gmail.com>
Date: Mon, 16 Feb 2026 22:27:27 -0800
Subject: [PATCH] test(companion): cover cancel wait snapshot updates

---
 docs/plans/state.json               | 11 +++++++++++
 src/companion/runtimeClient.test.ts | 21 +++++++++++++++++++++
 2 files changed, 32 insertions(+)

diff --git a/docs/plans/state.json b/docs/plans/state.json
index f8df4d9..2a19ead 100644
--- a/docs/plans/state.json
+++ b/docs/plans/state.json
@@ -1173,6 +1173,17 @@
       ],
       "test_status": "pnpm test:run src/companion/platformClients.integration.test.ts src/companion/platformClients.test.ts src/companion/runtimeClient.test.ts src/companion/heartbeatLoop.test.ts + pnpm typecheck passing"
     },
+    "companion-runtime-cancel-waits-snapshot-coverage": {
+      "status": "completed",
+      "date": "2026-02-17",
+      "updated": "2026-02-17",
+      "summary": "Added runtime regression coverage proving `cancelPendingEventWaits()` immediately updates pending-work snapshots (`pendingEventWaitCount`, `hasPendingWork`).",
+      "files_modified": [
+        "src/companion/runtimeClient.test.ts",
+        "docs/plans/state.json"
+      ],
+      "test_status": "pnpm test:run src/companion/runtimeClient.test.ts src/companion/platformClients.test.ts src/companion/heartbeatLoop.test.ts src/companion/platformClients.integration.test.ts + pnpm typecheck passing"
+    },
     "companion-runtime-wait-for-idle-nonfinite-validation-coverage": {
       "status": "completed",
       "date": "2026-02-17",
diff --git a/src/companion/runtimeClient.test.ts b/src/companion/runtimeClient.test.ts
index ca26478..1224c42 100644
--- a/src/companion/runtimeClient.test.ts
+++ b/src/companion/runtimeClient.test.ts
@@ -420,6 +420,27 @@ describe('CompanionRuntimeClient', () => {
     expect(client.cancelPendingEventWaits()).toBe(0);
   });
 
+  it('cancelPendingEventWaits updates pending work snapshot immediately', async () => {
+    const client = new CompanionRuntimeClient({
+      url: 'ws://127.0.0.1:1',
+    });
+    const pendingWait = client.waitForEvent('agent.stream', { timeoutMs: 10_000 }).catch(() => undefined);
+
+    expect(client.getPendingWorkSnapshot()).toEqual({
+      pendingRequestCount: 0,
+      pendingEventWaitCount: 1,
+      hasPendingWork: true,
+    });
+
+    expect(client.cancelPendingEventWaits('snapshot cancel')).toBe(1);
+    expect(client.getPendingWorkSnapshot()).toEqual({
+      pendingRequestCount: 0,
+      pendingEventWaitCount: 0,
+      hasPendingWork: false,
+    });
+    await pendingWait;
+  });
+
   it('waitForEvent rejects immediately on disconnect', async () => {
     const client = new CompanionRuntimeClient({
       url: 'ws://127.0.0.1:1',