openclaw/src/plugins/document-extractor-public-artifacts.test.ts

import { beforeEach, describe, expect, it, vi } from "vitest";

const { publicArtifactModule } = vi.hoisted(() => ({
  publicArtifactModule: {} as Record<string, unknown>,
}));

vi.mock("./public-surface-loader.js", () => ({
  loadBundledPluginPublicArtifactModuleSync: vi.fn(() => publicArtifactModule),
  resolveBundledPluginPublicArtifactPath: vi.fn(
    () => "/repo/extensions/demo/document-extractor.ts",
  ),
}));

import { loadBundledDocumentExtractorEntriesFromDir } from "./document-extractor-public-artifacts.js";

describe("loadBundledDocumentExtractorEntriesFromDir", () => {
  beforeEach(() => {
    for (const key of Object.keys(publicArtifactModule)) {
      delete publicArtifactModule[key];
    }
  });

  it("isolates a throwing factory when another extractor factory succeeds", () => {
    const extract = vi.fn();
    publicArtifactModule.createBrokenDocumentExtractor = () => {
      throw new Error("native probe failed");
    };
    publicArtifactModule.createPdfDocumentExtractor = () => ({
      id: "pdf",
      label: "PDF",
      mimeTypes: ["application/pdf"],
      extract,
    });

    expect(
      loadBundledDocumentExtractorEntriesFromDir({
        dirName: "demo",
        pluginId: "demo",
      }),
    ).toStrictEqual([
      {
        id: "pdf",
        label: "PDF",
        mimeTypes: ["application/pdf"],
        extract,
        pluginId: "demo",
      },
    ]);
  });

  it("surfaces initialization failure when every matching factory throws", () => {
    const cause = new Error("native probe failed");
    publicArtifactModule.createPdfDocumentExtractor = () => {
      throw cause;
    };

    expect(() =>
      loadBundledDocumentExtractorEntriesFromDir({
        dirName: "demo",
        pluginId: "demo",
      }),
    ).toThrow("Unable to initialize document extractors for plugin demo");
  });
});