can1357
105 lines
4.0 KiB
TypeScript
105 lines
4.0 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from "bun:test";
|
|
import { Settings } from "../../src/config/settings";
|
|
import { TaskTool } from "../../src/task";
|
|
import * as discoveryModule from "../../src/task/discovery";
|
|
import type { TaskParams } from "../../src/task/types";
|
|
import type { ToolSession } from "../../src/tools";
|
|
|
|
const TEST_AGENTS = [
|
|
{
|
|
name: "task",
|
|
description: "General-purpose task agent",
|
|
systemPrompt: "You are a task agent.",
|
|
source: "bundled" as const,
|
|
},
|
|
];
|
|
|
|
function createSession(overrides: Partial<Record<string, unknown>> = {}): ToolSession {
|
|
return {
|
|
cwd: "/tmp",
|
|
hasUI: false,
|
|
settings: Settings.isolated(overrides),
|
|
getSessionFile: () => null,
|
|
getSessionSpawns: () => "*",
|
|
} as unknown as ToolSession;
|
|
}
|
|
|
|
function getSchemaProperties(tool: TaskTool): Record<string, unknown> {
|
|
return ((tool.parameters as { properties?: Record<string, unknown> }).properties ?? {}) as Record<string, unknown>;
|
|
}
|
|
|
|
function getAssignmentDescription(tool: TaskTool): string {
|
|
const properties = getSchemaProperties(tool);
|
|
const tasks = properties.tasks as { items?: { properties?: Record<string, { description?: string }> } } | undefined;
|
|
return tasks?.items?.properties?.assignment?.description ?? "";
|
|
}
|
|
|
|
function getFirstText(result: { content: Array<{ type: string; text?: string }> }): string {
|
|
const content = result.content.find(part => part.type === "text");
|
|
return content?.type === "text" ? (content.text ?? "") : "";
|
|
}
|
|
|
|
describe("task.simple", () => {
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
it("removes only the custom schema input in schema-free mode", async () => {
|
|
vi.spyOn(discoveryModule, "discoverAgents").mockResolvedValue({
|
|
agents: TEST_AGENTS,
|
|
projectAgentsDir: null,
|
|
});
|
|
|
|
const tool = await TaskTool.create(createSession({ "task.simple": "schema-free" }));
|
|
const properties = getSchemaProperties(tool);
|
|
|
|
expect(properties.context).toBeDefined();
|
|
expect(properties.schema).toBeUndefined();
|
|
expect(tool.description).toContain("Current input mode: `schema-free`.");
|
|
expect(tool.description).toContain("- `context`:");
|
|
expect(tool.description).not.toContain("- `schema`:");
|
|
expect(getAssignmentDescription(tool)).toContain("shared background belongs in `context`");
|
|
});
|
|
|
|
it("removes both context and schema inputs in independent mode", async () => {
|
|
vi.spyOn(discoveryModule, "discoverAgents").mockResolvedValue({
|
|
agents: TEST_AGENTS,
|
|
projectAgentsDir: null,
|
|
});
|
|
|
|
const tool = await TaskTool.create(createSession({ "task.simple": "independent" }));
|
|
const properties = getSchemaProperties(tool);
|
|
|
|
expect(properties.context).toBeUndefined();
|
|
expect(properties.schema).toBeUndefined();
|
|
expect(tool.description).toContain("Current input mode: `independent`.");
|
|
expect(tool.description).toContain("Every assignment must stand on its own.");
|
|
expect(tool.description).not.toContain("- `context`:");
|
|
expect(tool.description).not.toContain("- `schema`:");
|
|
expect(getAssignmentDescription(tool)).toContain("include any background that would otherwise live in `context`");
|
|
});
|
|
|
|
it("rejects direct schema and context fields when the mode disables them", async () => {
|
|
vi.spyOn(discoveryModule, "discoverAgents").mockResolvedValue({
|
|
agents: TEST_AGENTS,
|
|
projectAgentsDir: null,
|
|
});
|
|
|
|
const schemaFreeTool = await TaskTool.create(createSession({ "task.simple": "schema-free" }));
|
|
const schemaFreeResult = await schemaFreeTool.execute("tool-1", {
|
|
agent: "task",
|
|
schema: '{"properties":{"ok":{"type":"boolean"}}}',
|
|
tasks: [{ id: "One", description: "label", assignment: "Do the thing." }],
|
|
} as TaskParams);
|
|
expect(getFirstText(schemaFreeResult)).toContain("does not accept `schema`");
|
|
|
|
const independentTool = await TaskTool.create(createSession({ "task.simple": "independent" }));
|
|
const independentResult = await independentTool.execute("tool-2", {
|
|
agent: "task",
|
|
context: "Shared background",
|
|
tasks: [{ id: "Two", description: "label", assignment: "Do the independent thing." }],
|
|
} as TaskParams);
|
|
expect(getFirstText(independentResult)).toContain("does not accept `context`");
|
|
});
|
|
});
|