YeonGyu-Kim
cc97a023cc
Tests that mock an AgentFactory were using an `as AgentFactory` cast and a separate mutation of `mockFactory.mode` to satisfy the type. Replace with Object.assign so the factory type is constructed correctly without casts. Also type the empty discoveredSkills fixture so its element type is inferred from the function signature instead of collapsing to never[].
65 lines
1.8 KiB
TypeScript
65 lines
1.8 KiB
TypeScript
import { describe, expect, test } from "bun:test"
|
|
import { buildAgent } from "./agent-builder"
|
|
import type { AgentFactory } from "./types"
|
|
|
|
describe("#given an agent factory with mode", () => {
|
|
const mockFactory: AgentFactory = Object.assign((model: string) => ({
|
|
name: "test-agent",
|
|
description: "Test",
|
|
instructions: "test",
|
|
model,
|
|
temperature: 0.1,
|
|
}), { mode: "subagent" as const })
|
|
|
|
test("#when building agent from factory", () => {
|
|
const agent = buildAgent(mockFactory, "test-model")
|
|
expect(agent.mode).toBe("subagent")
|
|
})
|
|
})
|
|
|
|
describe("#given an agent factory with mode=primary", () => {
|
|
const mockFactory: AgentFactory = Object.assign((model: string) => ({
|
|
name: "primary-agent",
|
|
description: "Primary Test",
|
|
instructions: "test",
|
|
model,
|
|
temperature: 0.1,
|
|
}), { mode: "primary" as const })
|
|
|
|
test("#when building agent from factory", () => {
|
|
const agent = buildAgent(mockFactory, "test-model")
|
|
expect(agent.mode).toBe("primary")
|
|
})
|
|
})
|
|
|
|
describe("#given an agent config object without mode", () => {
|
|
const mockConfig = {
|
|
name: "config-agent",
|
|
description: "Config Test",
|
|
instructions: "test",
|
|
model: "test-model",
|
|
temperature: 0.1,
|
|
}
|
|
|
|
test("#when building agent from config object", () => {
|
|
const agent = buildAgent(mockConfig, "test-model")
|
|
expect(agent.mode).toBeUndefined()
|
|
})
|
|
})
|
|
|
|
describe("#given an agent factory with mode but config already has mode", () => {
|
|
const mockFactory: AgentFactory = Object.assign((model: string) => ({
|
|
name: "override-agent",
|
|
description: "Override Test",
|
|
instructions: "test",
|
|
model,
|
|
temperature: 0.1,
|
|
mode: "all" as const,
|
|
}), { mode: "subagent" as const })
|
|
|
|
test("#when building agent from factory", () => {
|
|
const agent = buildAgent(mockFactory, "test-model")
|
|
expect(agent.mode).toBe("all")
|
|
})
|
|
})
|