test(gateway): use real dedupe map in abort helpers

This commit is contained in:
Kaspre
2026-05-20 09:09:42 -04:00
committed by Peter Steinberger
parent 01fce88082
commit fd293bd2a7
2 changed files with 2 additions and 12 deletions

View File

@@ -279,9 +279,6 @@ describe("chat abort transcript persistence", () => {
chatDeltaSentAt: new Map([["run-stop-1", Date.now()]]),
removeChatRun: vi.fn().mockReturnValue({ sessionKey: "main", clientRunId: "client-stop-1" }),
agentRunSeq: new Map<string, number>([["run-stop-1", 1]]),
dedupe: {
get: vi.fn(),
},
});
await chatHandlers["chat.send"]({
@@ -321,9 +318,6 @@ describe("chat abort transcript persistence", () => {
sessionKey: "main",
clientRunId: "run-stop-canonical",
}),
dedupe: {
get: vi.fn(),
},
});
await chatHandlers["chat.send"]({
@@ -356,9 +350,6 @@ describe("chat abort transcript persistence", () => {
sessionKey: "alias-main",
clientRunId: "run-stop-raw-alias",
}),
dedupe: {
get: vi.fn(),
},
});
await chatHandlers["chat.send"]({
@@ -387,9 +378,6 @@ describe("chat abort transcript persistence", () => {
const active = createActiveRun("third-session", { sessionId });
const context = createChatAbortContext({
chatAbortControllers: new Map([["run-stop-client-session", active]]),
dedupe: {
get: vi.fn(),
},
});
await chatHandlers["chat.send"]({

View File

@@ -27,6 +27,7 @@ type ChatAbortTestContext = Record<string, unknown> & {
chatDeltaSentAt: Map<string, number>;
chatDeltaLastBroadcastLen: Map<string, number>;
chatDeltaLastBroadcastText: Map<string, string>;
dedupe: Map<string, unknown>;
agentDeltaSentAt: Map<string, number>;
bufferedAgentEvents: Map<string, unknown>;
chatAbortedRuns: Map<string, number>;
@@ -48,6 +49,7 @@ export function createChatAbortContext(
chatDeltaSentAt: new Map(),
chatDeltaLastBroadcastLen: new Map(),
chatDeltaLastBroadcastText: new Map(),
dedupe: new Map(),
agentDeltaSentAt: new Map(),
bufferedAgentEvents: new Map(),
chatAbortedRuns: new Map<string, number>(),