38 lines
1.5 KiB
TypeScript
38 lines
1.5 KiB
TypeScript
import { describe, it, expect, vi } from "vitest";
|
|
import { backgroundRemoverPlugin } from "./index";
|
|
import { runBackgroundRemover } from "./run";
|
|
import type { BackgroundRemoverOptions } from "./remove";
|
|
import type { ToolContext } from "../../core/plugins/plugin-types";
|
|
|
|
const mockContext: ToolContext = {
|
|
signal: new AbortController().signal,
|
|
reportProgress: vi.fn(),
|
|
logger: { info: vi.fn(), warn: vi.fn(), error: vi.fn() },
|
|
};
|
|
|
|
const options: BackgroundRemoverOptions = { quality: "small", useGpu: true };
|
|
|
|
describe("Background Remover Plugin", () => {
|
|
it("should have correct manifest", () => {
|
|
expect(backgroundRemoverPlugin.manifest.id).toBe("background-remover");
|
|
expect(backgroundRemoverPlugin.manifest.category).toBe("image");
|
|
expect(backgroundRemoverPlugin.manifest.input.type).toBe("files");
|
|
expect(backgroundRemoverPlugin.manifest.output.type).toBe("files");
|
|
});
|
|
|
|
it("should run in a worker with wasm capability", () => {
|
|
expect(backgroundRemoverPlugin.capabilities?.worker).toBe(true);
|
|
expect(backgroundRemoverPlugin.capabilities?.persistentWorker).toBe(true);
|
|
expect(backgroundRemoverPlugin.capabilities?.wasm).toBe(true);
|
|
expect(backgroundRemoverPlugin.capabilities?.customUi).toBe(true);
|
|
expect(backgroundRemoverPlugin.worker).toBeDefined();
|
|
expect(backgroundRemoverPlugin.customUi).toBeDefined();
|
|
});
|
|
|
|
it("should throw if no image is provided", async () => {
|
|
await expect(
|
|
runBackgroundRemover({ files: [] }, options, mockContext)
|
|
).rejects.toThrow("No image provided.");
|
|
});
|
|
});
|