51 lines
1.8 KiB
TypeScript
51 lines
1.8 KiB
TypeScript
|
import { describe, test, expect, beforeEach } from "bun:test"
|
||
|
|
import { setPendingSwitch, 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" })
|
||
|
|
})
|
||
|
|
})
|