|
|
import { describe, expect, it } from "vitest";
|
|
|
import { classifyMail } from "@/services/parse/classify";
|
|
|
import { extractMailInstructions } from "@/services/parse/split-instructions";
|
|
|
|
|
|
/** 金样：docx/邮件/模板/附件下载_邮件识别 */
|
|
|
const GOLD_SUBJECT =
|
|
|
"\u667a\u9e3f2+WHLC027G597465+WHSU8127240+\u6d1b\u6749\u77f6+40HQ+EDT2026.04-25 ETA2026.05-15\u8239\u540d\u822a\u6b21HMM EMERALD 013E+\u63d0\u62c6\u6d3e";
|
|
|
|
|
|
const GOLD_BODY = [
|
|
|
"\u8bf7\u67e5\u6536\u65b0\u589e\u9884\u62a5\uff1a\u667a\u9e3f2+WHLC027G597465+WHSU8127240+\u6d1b\u6749\u77f6+40HQ",
|
|
|
"DO\u4e5f\u540c\u6b65\u4e0a\u4f20\u81f3\u9644\u4ef6\u8bf7\u6ce8\u610f\u67e5\u6536\uff01",
|
|
|
].join("\n");
|
|
|
|
|
|
const GOLD_FILES = [
|
|
|
"\u6570\u636e\u6a21\u7248.xlsx",
|
|
|
"WHSU8127240-DO.pdf",
|
|
|
"\u667a\u9e3f2+WHLC027G597465+WHSU8127240+\u6d1b\u6749\u77f6+40HQ+EDT2026.04-25 ETA2026.05-15\u8239\u540d\u822a\u6b21HMM EMERALD 013E+\u63d0\u62c6\u6d3e.xlsx",
|
|
|
];
|
|
|
|
|
|
describe("recognize gold — \u9644\u4ef6\u4e0b\u8f7d_\u90ae\u4ef6\u8bc6\u522b", () => {
|
|
|
it("classify: DO primary + NEW_CONTAINER signals", () => {
|
|
|
const r = classifyMail({
|
|
|
subject: GOLD_SUBJECT,
|
|
|
body: GOLD_BODY,
|
|
|
filenames: GOLD_FILES,
|
|
|
});
|
|
|
expect(r.mail_type).toBe("DO_UPLOAD");
|
|
|
expect(r.scores.DO_UPLOAD).toBeGreaterThanOrEqual(40);
|
|
|
expect(r.scores.NEW_CONTAINER).toBeGreaterThanOrEqual(40);
|
|
|
expect(r.signals.some((s) => s.signal === "\u65b0\u589e\u9884\u62a5")).toBe(
|
|
|
true,
|
|
|
);
|
|
|
expect(r.signals.some((s) => /DO/.test(s.signal))).toBe(true);
|
|
|
});
|
|
|
|
|
|
it("body-only \u65b0\u589e\u9884\u62a5 still scores forecast", () => {
|
|
|
const r = classifyMail({
|
|
|
subject: "WHSU8127240",
|
|
|
body: "\u8bf7\u67e5\u6536\u65b0\u589e\u9884\u62a5",
|
|
|
filenames: ["\u6570\u636e\u6a21\u7248.xlsx"],
|
|
|
});
|
|
|
expect(r.scores.NEW_CONTAINER).toBeGreaterThanOrEqual(40);
|
|
|
expect(r.mail_type).toBe("NEW_CONTAINER");
|
|
|
});
|
|
|
|
|
|
it("\u6570\u636e\u6a21\u7248.xlsx alone can lift forecast", () => {
|
|
|
const r = classifyMail({
|
|
|
subject: "WHSU8127240+40HQ+ETA",
|
|
|
body: "\u8bf7\u67e5\u6536",
|
|
|
filenames: ["\u6570\u636e\u6a21\u7248.xlsx"],
|
|
|
});
|
|
|
expect(r.scores.NEW_CONTAINER).toBeGreaterThanOrEqual(40);
|
|
|
expect(r.mail_type).toBe("NEW_CONTAINER");
|
|
|
});
|
|
|
|
|
|
it("split: forecast + do_upload both current", () => {
|
|
|
const units = extractMailInstructions({
|
|
|
subject: GOLD_SUBJECT,
|
|
|
body: GOLD_BODY,
|
|
|
filenames: GOLD_FILES,
|
|
|
});
|
|
|
const kinds = units.map((u) => u.uiKind);
|
|
|
expect(kinds).toContain("forecast");
|
|
|
expect(kinds).toContain("do_upload");
|
|
|
expect(units.filter((u) => u.uiKind === "forecast")[0]?.isCurrent).toBe(
|
|
|
true,
|
|
|
);
|
|
|
expect(units.filter((u) => u.uiKind === "do_upload")[0]?.isCurrent).toBe(
|
|
|
true,
|
|
|
);
|
|
|
});
|
|
|
});
|
