Files
writing-agent/tests/parse-json-doc.test.ts

53 lines
1.7 KiB
TypeScript

import { describe, expect, it } from "vitest";
import {
canonicalizeInsertedContent,
extractLastProductJson,
selectJsonPayload,
tryParseJsonDoc,
} from "../src/parse/json-doc.js";
describe("canonicalizeInsertedContent", () => {
it("extracts a broken fragment string into canonical JSON", () => {
const raw = `{
"schema":"context-fragment-v1",
"技能":"主角设定",
"brief":"点题",
"mount":"world-simulator"
},
"正文": { "称呼": "@玩家" }
}`;
const stored = canonicalizeInsertedContent(raw);
expect(JSON.parse(stored)).toMatchObject({
schema: "context-fragment.v1",
技能: "主角设定",
正文: { 称呼: "@玩家" },
});
expect(canonicalizeInsertedContent("开场是一段散文。")).toBe("开场是一段散文。");
});
});
describe("selectJsonPayload", () => {
it("keeps content when it is already the product", () => {
const json = '{"schema":"context-fragment.v1","brief":"a","正文":{}}';
expect(selectJsonPayload(json, "thinking...")).toBe(json);
});
it("falls back to reasoning when content is not JSON", () => {
const json = '{"schema":"context-fragment.v1","brief":"a","正文":{}}';
expect(selectJsonPayload("Done.", json)).toBe(json);
});
});
describe("extractLastProductJson", () => {
it("skips an earlier example object and takes the fragment", () => {
const raw = `example { "foo": 1 } then ${JSON.stringify({
schema: "context-fragment.v1",
brief: "点题",
正文: { x: 1 },
})}`;
const parsed = extractLastProductJson(raw) as { brief?: string };
expect(parsed?.brief).toBe("点题");
expect(tryParseJsonDoc(raw)).toMatchObject({ brief: "点题" });
});
});