diff --git a/src/acp/session-mapper.test.ts b/src/acp/session-mapper.test.ts index 859b1da7380..ac06dcf4b89 100644 --- a/src/acp/session-mapper.test.ts +++ b/src/acp/session-mapper.test.ts @@ -1,6 +1,7 @@ -import { describe, expect, it, vi } from "vitest"; +import { afterEach, describe, expect, it, vi } from "vitest"; import type { GatewayClient } from "../gateway/client.js"; import { parseSessionMeta, resolveSessionKey } from "./session-mapper.js"; +import { createInMemorySessionStore } from "./session.js"; function createGateway(resolveLabelKey = "agent:main:label"): { gateway: GatewayClient; @@ -54,3 +55,26 @@ describe("acp session mapper", () => { expect(request).not.toHaveBeenCalled(); }); }); + +describe("acp session manager", () => { + const store = createInMemorySessionStore(); + + afterEach(() => { + store.clearAllSessionsForTest(); + }); + + it("tracks active runs and clears on cancel", () => { + const session = store.createSession({ + sessionKey: "acp:test", + cwd: "/tmp", + }); + const controller = new AbortController(); + store.setActiveRun(session.sessionId, "run-1", controller); + + expect(store.getSessionByRunId("run-1")?.sessionId).toBe(session.sessionId); + + const cancelled = store.cancelActiveRun(session.sessionId); + expect(cancelled).toBe(true); + expect(store.getSessionByRunId("run-1")).toBeUndefined(); + }); +}); diff --git a/src/acp/session.test.ts b/src/acp/session.test.ts deleted file mode 100644 index a38b58f1703..00000000000 --- a/src/acp/session.test.ts +++ /dev/null @@ -1,25 +0,0 @@ -import { describe, expect, it, afterEach } from "vitest"; -import { createInMemorySessionStore } from "./session.js"; - -describe("acp session manager", () => { - const store = createInMemorySessionStore(); - - afterEach(() => { - store.clearAllSessionsForTest(); - }); - - it("tracks active runs and clears on cancel", () => { - const session = store.createSession({ - sessionKey: "acp:test", - cwd: "/tmp", - }); - const controller = new AbortController(); - store.setActiveRun(session.sessionId, "run-1", controller); - - expect(store.getSessionByRunId("run-1")?.sessionId).toBe(session.sessionId); - - const cancelled = store.cancelActiveRun(session.sessionId); - expect(cancelled).toBe(true); - expect(store.getSessionByRunId("run-1")).toBeUndefined(); - }); -});