新增样例文章加载器

2026-07-01 14:58:13 +08:00
parent 34e6da23b0
commit 1781088640
3 changed files with 319 additions and 0 deletions
@@ -0,0 +1,105 @@
+import { mkdtempSync, rmSync, writeFileSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+
+import { loadArticleSamples } from "../e2e/sample-flow/sample-loader";
+
+describe("loadArticleSamples", () => {
+  let tempDir: string;
+
+  beforeEach(() => {
+    tempDir = mkdtempSync(join(tmpdir(), "geo-samples-"));
+  });
+
+  afterEach(() => {
+    rmSync(tempDir, { recursive: true, force: true });
+  });
+
+  it("loads valid samples in filename order and normalizes optional fields", () => {
+    writeFileSync(
+      join(tempDir, "b-sample.json"),
+      JSON.stringify({
+        name: "B sample",
+        input: {
+          body: "第二篇文章正文",
+          platform: "media_article",
+        },
+      }),
+    );
+    writeFileSync(
+      join(tempDir, "a-sample.json"),
+      JSON.stringify({
+        name: "A sample",
+        input: {
+          title: "标题",
+          body: "第一篇文章正文",
+          image_lines: "图一",
+          platform: "official_site",
+          user_instructions: "保持事实准确",
+        },
+        expectedHardFailures: ["body_quality"],
+        expectedWarnings: ["context_sensitive_terms"],
+      }),
+    );
+
+    const result = loadArticleSamples(tempDir);
+
+    expect(result.valid.map((sample) => sample.fileName)).toEqual([
+      "a-sample.json",
+      "b-sample.json",
+    ]);
+    expect(result.invalid).toEqual([]);
+    expect(result.valid[1]).toEqual(
+      expect.objectContaining({
+        name: "B sample",
+        input: expect.objectContaining({
+          title: "",
+          image_lines: "",
+          user_instructions: "",
+        }),
+        expectedHardFailures: [],
+        expectedWarnings: [],
+      }),
+    );
+  });
+
+  it("returns invalid entries for malformed samples without throwing", () => {
+    writeFileSync(
+      join(tempDir, "empty-body.json"),
+      JSON.stringify({
+        name: "Empty body",
+        input: {
+          body: "   ",
+          platform: "official_site",
+        },
+      }),
+    );
+    writeFileSync(
+      join(tempDir, "bad-platform.json"),
+      JSON.stringify({
+        name: "Bad platform",
+        input: {
+          body: "正文",
+          platform: "unknown",
+        },
+      }),
+    );
+
+    const result = loadArticleSamples(tempDir);
+
+    expect(result.valid).toEqual([]);
+    expect(result.invalid).toEqual([
+      expect.objectContaining({
+        fileName: "bad-platform.json",
+        reason:
+          "input.platform must be one of official_site, media_article, comparison_review, recommendation_list",
+      }),
+      expect.objectContaining({
+        fileName: "empty-body.json",
+        reason: "input.body must be a non-empty string",
+      }),
+    ]);
+  });
+});