58 lines
1.8 KiB
TypeScript
58 lines
1.8 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import {
|
|
buildIntakeFollowUpMessage,
|
|
buildIntakeProgress,
|
|
extractIntakeHeuristic,
|
|
intakeFieldsFromInquiry,
|
|
} from "../src/intake/intake.js";
|
|
|
|
describe("intake fields", () => {
|
|
it("marks ready when all required filled", () => {
|
|
const fields = intakeFieldsFromInquiry({
|
|
prompt: "p",
|
|
targetKey: "用户.需求",
|
|
requiredFields: ["情境", "角色", "轮次"],
|
|
optionalFields: ["输出偏好"],
|
|
});
|
|
const partial = buildIntakeProgress(fields, {
|
|
[fields[0].id]: "德州扑克",
|
|
});
|
|
expect(partial.ready).toBe(false);
|
|
expect(partial.requiredFilled).toBe(1);
|
|
|
|
const full = buildIntakeProgress(fields, {
|
|
[fields[0].id]: "德州扑克",
|
|
[fields[1].id]: "玩家A算计，玩家B保守",
|
|
[fields[2].id]: "单轮",
|
|
});
|
|
expect(full.ready).toBe(true);
|
|
});
|
|
|
|
it("heuristic fills scenario and rounds keywords", () => {
|
|
const fields = intakeFieldsFromInquiry({
|
|
prompt: "p",
|
|
targetKey: "t",
|
|
requiredFields: ["实验情境", "轮次模式"],
|
|
optionalFields: [],
|
|
});
|
|
const values = extractIntakeHeuristic("德州扑克，单轮定胜负", fields, {});
|
|
expect(Object.keys(values).length).toBeGreaterThan(0);
|
|
});
|
|
|
|
it("buildIntakeFollowUpMessage lists missing required fields", () => {
|
|
const fields = intakeFieldsFromInquiry({
|
|
prompt: "p",
|
|
targetKey: "t",
|
|
requiredFields: ["情境", "角色", "进程"],
|
|
optionalFields: [],
|
|
});
|
|
const progress = buildIntakeProgress(fields, {
|
|
[fields[0].id]: "德州扑克",
|
|
});
|
|
const msg = buildIntakeFollowUpMessage(progress);
|
|
expect(msg).toContain("还缺以下必要项");
|
|
expect(msg).toContain("角色");
|
|
expect(msg).toContain("进程");
|
|
});
|
|
});
|