test: OCR组件冒烟回归(本地离线推理): 通用/键值/金额/验证码/卡证/票据/离线文本定位表格 10项真实识别通过

src/main/capabilities/ocr-smoke.test.ts

@@ -0,0 +1,87 @@
+import { describe, expect, it } from "vitest";
+import { getHandler } from "../engine/registry";
+import type { CatalogComponent, RunContext } from "../engine/types";
+import "./ocr-caps";
+import "./remaining-core";
+
+const ctx = {
+  signal: new AbortController().signal,
+  log: () => {},
+  vars: {},
+  evidenceDir: "D:/office-test2/ocr-test"
+} as unknown as RunContext;
+
+const node = { id: "smoke", componentId: "smoke" };
+
+async function runOcr(id: string, args: Record<string, unknown>) {
+  const handler = getHandler(id);
+  expect(handler, `${id} 应已注册`).toBeTruthy();
+  return handler!(ctx, node, args, {} as CatalogComponent);
+}
+
+describe("OCR 组件冒烟（本地离线识别，真实模型推理）", () => {
+  it("OCR_GeneralTextV2 通用文本", async () => {
+    const out = await runOcr("OCR_GeneralTextV2", { path: "D:/office-test2/ocr-test/general.png" });
+    console.log("[GeneralTextV2]", JSON.stringify(out));
+    expect(String(out)).toContain("张三");
+  }, 300000);
+
+  it("OCR_KeyValueIdentification 键值对", async () => {
+    const out = await runOcr("OCR_KeyValueIdentification", { image_path: "D:/office-test2/ocr-test/general.png" });
+    console.log("[KeyValue]", JSON.stringify(out));
+    expect(String(out).length).toBeGreaterThan(0);
+  }, 300000);
+
+  it("OCR_AmountIdentification 金额", async () => {
+    const out = await runOcr("OCR_AmountIdentification", { path: "D:/office-test2/ocr-test/general.png" });
+    console.log("[Amount]", JSON.stringify(out));
+    expect(String(out).length).toBeGreaterThan(0);
+  }, 300000);
+
+  it("OCR_CalculationCode 算术验证码", async () => {
+    const out = await runOcr("OCR_CalculationCode", { path: "D:/office-test2/ocr-test/code.png" });
+    console.log("[CalcCode]", JSON.stringify(out));
+    expect(String(out).length).toBeGreaterThan(0);
+  }, 300000);
+
+  it("OCR_CommonEnglishNumbersCode 普通验证码", async () => {
+    const out = await runOcr("OCR_CommonEnglishNumbersCode", { path: "D:/office-test2/ocr-test/code.png" });
+    console.log("[CommonCode]", JSON.stringify(out));
+    expect(String(out).length).toBeGreaterThan(0);
+  }, 300000);
+
+  it("OCR_IdCardV2 卡证族代表", async () => {
+    const out = await runOcr("OCR_IdCardV2", { path: "D:/office-test2/ocr-test/general.png" });
+    console.log("[IdCardV2]", JSON.stringify(out));
+    expect(String(out).length).toBeGreaterThan(0);
+  }, 300000);
+
+  it("OCR_VatInvoiceV2 票据族代表", async () => {
+    const out = await runOcr("OCR_VatInvoiceV2", { path: "D:/office-test2/ocr-test/table.png" });
+    console.log("[VatInvoiceV2]", JSON.stringify(out));
+    expect(String(out).length).toBeGreaterThan(0);
+  }, 300000);
+
+  it("OcrOffline_GetOcrTextNew 离线文本", async () => {
+    const out = await runOcr("OcrOffline_GetOcrTextNew", { uiSelector: "D:/office-test2/ocr-test/general.png" });
+    console.log("[OcrTextNew]", JSON.stringify(out));
+    expect(String(out)).toContain("订单号");
+  }, 300000);
+
+  it("OcrOffline_GetOcrTextPosition 离线定位", async () => {
+    const out = await runOcr("OcrOffline_GetOcrTextPosition", { image_path: "D:/office-test2/ocr-test/general.png" });
+    const words = out as Array<{ text: string; x: number; y: number }>;
+    console.log("[OcrPosition]", JSON.stringify(words?.slice(0, 3)));
+    expect(Array.isArray(words)).toBe(true);
+    expect(words.length).toBeGreaterThan(0);
+    expect(words[0]).toHaveProperty("x");
+  }, 300000);
+
+  it("OcrOffline_GetOcrTable 离线表格", async () => {
+    const out = await runOcr("OcrOffline_GetOcrTable", { image_path: "D:/office-test2/ocr-test/table.png", showTable: "1", isExcel: "0" });
+    const rows = out as string[][];
+    console.log("[OcrTable]", JSON.stringify(rows));
+    expect(Array.isArray(rows)).toBe(true);
+    expect(rows.length).toBeGreaterThan(0);
+  }, 300000);
+});