59 lines
2.2 KiB
TypeScript
59 lines
2.2 KiB
TypeScript
import { describe, expect, test } from "bun:test"
|
|
import { selectedWebSearchProvider, toolDisplayMetadata, webSearchProviderLabel } from "../../src/util/tool-display"
|
|
|
|
describe("webSearchProviderLabel", () => {
|
|
test("labels known providers", () => {
|
|
expect(webSearchProviderLabel("parallel")).toBe("Parallel Web Search")
|
|
expect(webSearchProviderLabel("exa")).toBe("Exa Web Search")
|
|
})
|
|
|
|
for (const [name, provider] of [
|
|
["undefined", undefined],
|
|
["null", null],
|
|
["an object", {}],
|
|
["an array", []],
|
|
["a number", 1],
|
|
["an unexpected string", "other"],
|
|
] as const) {
|
|
test(`uses the generic label for ${name}`, () => {
|
|
expect(webSearchProviderLabel(provider)).toBe("Web Search")
|
|
})
|
|
}
|
|
})
|
|
|
|
describe("selectedWebSearchProvider", () => {
|
|
test("returns the selected search integration", () => {
|
|
expect(
|
|
selectedWebSearchProvider([
|
|
{ id: "exa", capabilities: [{ type: "search", connection: "optional", selected: false }] },
|
|
{ id: "parallel", capabilities: [{ type: "search", connection: "optional", selected: true }] },
|
|
]),
|
|
).toBe("parallel")
|
|
})
|
|
|
|
test("ignores unselected and unrelated capabilities", () => {
|
|
expect(
|
|
selectedWebSearchProvider([
|
|
{ id: "exa", capabilities: [{ type: "search", connection: "optional", selected: false }] },
|
|
]),
|
|
).toBeUndefined()
|
|
})
|
|
})
|
|
|
|
describe("toolDisplayMetadata", () => {
|
|
test("returns structured metadata for non-pending states", () => {
|
|
const structured = { provider: "parallel", numResults: 3 }
|
|
|
|
expect(toolDisplayMetadata({ status: "running", structured })).toBe(structured)
|
|
expect(toolDisplayMetadata({ status: "completed", structured })).toBe(structured)
|
|
expect(toolDisplayMetadata({ status: "error", structured })).toBe(structured)
|
|
})
|
|
|
|
test("does not expose pending or malformed metadata", () => {
|
|
expect(toolDisplayMetadata({ status: "pending", structured: { provider: "exa" } })).toEqual({})
|
|
expect(toolDisplayMetadata({ status: "completed" })).toEqual({})
|
|
expect(toolDisplayMetadata({ status: "completed", structured: null })).toEqual({})
|
|
expect(toolDisplayMetadata({ status: "completed", structured: [] })).toEqual({})
|
|
expect(toolDisplayMetadata(undefined)).toEqual({})
|
|
})
|
|
})
|