From ac9d4ff2f0e72d564dfa54050321cdd0c1da112d Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Sat, 6 Jun 2026 08:53:01 -0700
Subject: [PATCH] test(gateway): include aborted chat run state in mock

---
 src/gateway/server-methods/chat.error-broadcast.test.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/gateway/server-methods/chat.error-broadcast.test.ts b/src/gateway/server-methods/chat.error-broadcast.test.ts
index 409b7c936458..62db20b706f4 100644
--- a/src/gateway/server-methods/chat.error-broadcast.test.ts
+++ b/src/gateway/server-methods/chat.error-broadcast.test.ts
@@ -8,6 +8,7 @@ function createMockContext() {
   const broadcast = vi.fn();
   const nodeSendToSession = vi.fn();
   const chatAbortControllers = new Map();
+  const chatAbortedRuns = new Map();
   const agentRunSeq = new Map<string, number>();
   const dedupe = new Map();
 
@@ -15,6 +16,7 @@ function createMockContext() {
     broadcast,
     nodeSendToSession,
     chatAbortControllers,
+    chatAbortedRuns,
     agentRunSeq,
     dedupe,
     getRuntimeConfig: () => ({ agents: { list: [{ id: "main", default: true }] } }),