SnapOtter/tests/unit/shared/modality.test.ts

import { detectModalityFromMime, MODALITIES, MODALITY_POOL, TOOLS } from "@snapotter/shared";
import { describe, expect, it } from "vitest";

describe("modality metadata", () => {
  it("defines five modalities with UI metadata", () => {
    expect(MODALITIES.map((m) => m.id)).toEqual(["image", "video", "audio", "document", "file"]);
    for (const m of MODALITIES) {
      expect(m.name).toBeTruthy();
      expect(m.icon).toBeTruthy();
    }
  });

  it("maps every modality to a worker pool", () => {
    expect(MODALITY_POOL.image).toBe("image");
    expect(MODALITY_POOL.video).toBe("media");
    expect(MODALITY_POOL.audio).toBe("media");
    expect(MODALITY_POOL.document).toBe("docs");
    expect(MODALITY_POOL.file).toBe("docs");
  });

  it("every tool declares a valid modality, acceptedInputs and executionHint", () => {
    const validModalities = ["image", "video", "audio", "document", "file"];
    expect(TOOLS.length).toBeGreaterThanOrEqual(53);
    for (const tool of TOOLS) {
      expect(validModalities).toContain(tool.modality);
      expect(Array.isArray(tool.acceptedInputs)).toBe(true);
      expect(tool.acceptedInputs.length).toBeGreaterThan(0);
      expect(["fast", "long"]).toContain(tool.executionHint);
    }
  });

  it("pdf-to-image is a document-modality tool", () => {
    const pdfToImage = TOOLS.find((t) => t.id === "pdf-to-image");
    expect(pdfToImage).toBeDefined();
    expect(pdfToImage!.modality).toBe("document");
    expect(pdfToImage!.category).toBe("documents");
    expect(pdfToImage!.acceptedInputs).toEqual([".pdf"]);
  });

  it("AI tools are hinted long (except pure-CV ones)", () => {
    const ai = TOOLS.filter((t) => t.category === "ai");
    expect(ai.length).toBeGreaterThan(0);
    // image-enhancement is categorized "ai" but its core path is pure
    // sharp/CV; only the optional deepEnhance invokes a model.
    const pureCvAiTools = new Set(["image-enhancement"]);
    for (const t of ai) {
      if (pureCvAiTools.has(t.id)) {
        expect(t.executionHint).toBe("fast");
      } else {
        expect(t.executionHint).toBe("long");
      }
    }
  });

  it("detects modality from mime", () => {
    expect(detectModalityFromMime("image/png")).toBe("image");
    expect(detectModalityFromMime("video/mp4")).toBe("video");
    expect(detectModalityFromMime("audio/mpeg")).toBe("audio");
    expect(detectModalityFromMime("application/pdf")).toBe("document");
    expect(detectModalityFromMime("text/csv")).toBe("file");
    expect(detectModalityFromMime("")).toBe("file");
  });
});