fix(test): add _resetForTesting for proper test isolation

This commit is contained in:
justsisyphus 2026-01-17 16:57:31 +09:00
parent ec2cf22449
commit b4fa31a47a
2 changed files with 11 additions and 11 deletions

View File

@ -6,17 +6,16 @@ import {
updateSessionAgent,
setMainSession,
getMainSessionID,
subagentSessions,
_resetForTesting,
} from "./state"
describe("claude-code-session-state", () => {
beforeEach(() => {
// #given - clean state before each test
_resetForTesting()
clearSessionAgent("test-session-1")
clearSessionAgent("test-session-2")
clearSessionAgent("test-prometheus-session")
setMainSession(undefined)
subagentSessions.clear()
})
describe("setSessionAgent", () => {
@ -82,11 +81,6 @@ describe("claude-code-session-state", () => {
})
describe("mainSessionID", () => {
beforeEach(() => {
// #given - ensure clean state for mainSessionID tests
setMainSession(undefined)
})
test("should store and retrieve main session ID", () => {
// #given
const mainID = "main-session-123"

View File

@ -1,13 +1,19 @@
export const subagentSessions = new Set<string>()
export let mainSessionID: string | undefined
let _mainSessionID: string | undefined
export function setMainSession(id: string | undefined) {
mainSessionID = id
_mainSessionID = id
}
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>()