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("进程");
|
||
});
|
||
});
|