fix(test): add _resetForTesting for proper test isolation

This commit is contained in:
justsisyphus
2026-01-17 16:57:31 +09:00
parent 86711a7b5b
commit c1e42d71b3
2 changed files with 11 additions and 11 deletions
@@ -6,17 +6,16 @@ import {
updateSessionAgent, updateSessionAgent,
setMainSession, setMainSession,
getMainSessionID, getMainSessionID,
subagentSessions, _resetForTesting,
} from "./state" } from "./state"
describe("claude-code-session-state", () => { describe("claude-code-session-state", () => {
beforeEach(() => { beforeEach(() => {
// #given - clean state before each test // #given - clean state before each test
_resetForTesting()
clearSessionAgent("test-session-1") clearSessionAgent("test-session-1")
clearSessionAgent("test-session-2") clearSessionAgent("test-session-2")
clearSessionAgent("test-prometheus-session") clearSessionAgent("test-prometheus-session")
setMainSession(undefined)
subagentSessions.clear()
}) })
describe("setSessionAgent", () => { describe("setSessionAgent", () => {
@@ -82,11 +81,6 @@ describe("claude-code-session-state", () => {
}) })
describe("mainSessionID", () => { describe("mainSessionID", () => {
beforeEach(() => {
// #given - ensure clean state for mainSessionID tests
setMainSession(undefined)
})
test("should store and retrieve main session ID", () => { test("should store and retrieve main session ID", () => {
// #given // #given
const mainID = "main-session-123" const mainID = "main-session-123"
@@ -1,13 +1,19 @@
export const subagentSessions = new Set<string>() export const subagentSessions = new Set<string>()
export let mainSessionID: string | undefined let _mainSessionID: string | undefined
export function setMainSession(id: string | undefined) { export function setMainSession(id: string | undefined) {
mainSessionID = id _mainSessionID = id
} }
export function getMainSessionID(): string | undefined { export function getMainSessionID(): string | undefined {
return mainSessionID return _mainSessionID
}
/** @internal For testing only */
export function _resetForTesting(): void {
_mainSessionID = undefined
subagentSessions.clear()
} }
const sessionAgentMap = new Map<string, string>() const sessionAgentMap = new Map<string, string>()