38a4bbc75f
In-memory Map-based store that records tool restriction objects (e.g., question: false) by sessionID when prompts are sent. This enables retrieving the original session's tool parameters when background tasks complete and need to notify the parent session.
73 lines
1.8 KiB
TypeScript
73 lines
1.8 KiB
TypeScript
import { describe, test, expect, beforeEach } from "bun:test"
|
|
import { setSessionTools, getSessionTools, clearSessionTools } from "./session-tools-store"
|
|
|
|
describe("session-tools-store", () => {
|
|
beforeEach(() => {
|
|
clearSessionTools()
|
|
})
|
|
|
|
test("returns undefined for unknown session", () => {
|
|
//#given
|
|
const sessionID = "ses_unknown"
|
|
|
|
//#when
|
|
const result = getSessionTools(sessionID)
|
|
|
|
//#then
|
|
expect(result).toBeUndefined()
|
|
})
|
|
|
|
test("stores and retrieves tools for a session", () => {
|
|
//#given
|
|
const sessionID = "ses_abc123"
|
|
const tools = { question: false, task: true, call_omo_agent: true }
|
|
|
|
//#when
|
|
setSessionTools(sessionID, tools)
|
|
const result = getSessionTools(sessionID)
|
|
|
|
//#then
|
|
expect(result).toEqual({ question: false, task: true, call_omo_agent: true })
|
|
})
|
|
|
|
test("overwrites existing tools for same session", () => {
|
|
//#given
|
|
const sessionID = "ses_abc123"
|
|
setSessionTools(sessionID, { question: false })
|
|
|
|
//#when
|
|
setSessionTools(sessionID, { question: true, task: false })
|
|
const result = getSessionTools(sessionID)
|
|
|
|
//#then
|
|
expect(result).toEqual({ question: true, task: false })
|
|
})
|
|
|
|
test("clearSessionTools removes all entries", () => {
|
|
//#given
|
|
setSessionTools("ses_1", { question: false })
|
|
setSessionTools("ses_2", { task: true })
|
|
|
|
//#when
|
|
clearSessionTools()
|
|
|
|
//#then
|
|
expect(getSessionTools("ses_1")).toBeUndefined()
|
|
expect(getSessionTools("ses_2")).toBeUndefined()
|
|
})
|
|
|
|
test("returns a copy, not a reference", () => {
|
|
//#given
|
|
const sessionID = "ses_abc123"
|
|
const tools = { question: false }
|
|
setSessionTools(sessionID, tools)
|
|
|
|
//#when
|
|
const result = getSessionTools(sessionID)!
|
|
result.question = true
|
|
|
|
//#then
|
|
expect(getSessionTools(sessionID)).toEqual({ question: false })
|
|
})
|
|
})
|