ismeth
74 lines
2.3 KiB
TypeScript
74 lines
2.3 KiB
TypeScript
const { describe, test, expect, beforeEach } = require("bun:test")
|
|
import {
|
|
setPendingSwitch,
|
|
getPendingSwitch,
|
|
clearPendingSwitch,
|
|
consumePendingSwitch,
|
|
_resetForTesting,
|
|
} from "./state"
|
|
|
|
describe("agent-switch state", () => {
|
|
beforeEach(() => {
|
|
_resetForTesting()
|
|
})
|
|
|
|
//#given a pending switch is set
|
|
//#when consumePendingSwitch is called
|
|
//#then it returns the switch and removes it
|
|
test("should store and consume a pending switch", () => {
|
|
setPendingSwitch("session-1", "atlas", "Fix these findings")
|
|
|
|
const entry = consumePendingSwitch("session-1")
|
|
|
|
expect(entry).toEqual({ agent: "atlas", context: "Fix these findings" })
|
|
expect(consumePendingSwitch("session-1")).toBeUndefined()
|
|
})
|
|
|
|
//#given no pending switch exists
|
|
//#when consumePendingSwitch is called
|
|
//#then it returns undefined
|
|
test("should return undefined when no switch is pending", () => {
|
|
expect(consumePendingSwitch("session-1")).toBeUndefined()
|
|
})
|
|
|
|
//#given a pending switch is set
|
|
//#when a new switch is set for the same session
|
|
//#then the latest switch wins
|
|
test("should overwrite previous switch for same session", () => {
|
|
setPendingSwitch("session-1", "atlas", "Fix A")
|
|
setPendingSwitch("session-1", "prometheus", "Plan B")
|
|
|
|
const entry = consumePendingSwitch("session-1")
|
|
|
|
expect(entry).toEqual({ agent: "prometheus", context: "Plan B" })
|
|
})
|
|
|
|
//#given switches for different sessions
|
|
//#when consumed separately
|
|
//#then each session gets its own switch
|
|
test("should isolate switches by session", () => {
|
|
setPendingSwitch("session-1", "atlas", "Fix A")
|
|
setPendingSwitch("session-2", "prometheus", "Plan B")
|
|
|
|
expect(consumePendingSwitch("session-1")).toEqual({ agent: "atlas", context: "Fix A" })
|
|
expect(consumePendingSwitch("session-2")).toEqual({ agent: "prometheus", context: "Plan B" })
|
|
})
|
|
|
|
test("should allow reading without consuming", () => {
|
|
setPendingSwitch("session-1", "atlas", "Fix A")
|
|
|
|
expect(getPendingSwitch("session-1")).toEqual({ agent: "atlas", context: "Fix A" })
|
|
expect(getPendingSwitch("session-1")).toEqual({ agent: "atlas", context: "Fix A" })
|
|
})
|
|
|
|
test("should clear pending switch explicitly", () => {
|
|
setPendingSwitch("session-1", "atlas", "Fix A")
|
|
|
|
clearPendingSwitch("session-1")
|
|
|
|
expect(getPendingSwitch("session-1")).toBeUndefined()
|
|
})
|
|
})
|
|
|
|
export {}
|