ff762b8f17
- Enhanced cursor behavior for new tools: semantic select, mask lasso, and mask rectangle. - Updated mask edit state to include mask asset ID and kind. - Implemented inpaint region commands for adding, applying, and removing inpaint regions. - Introduced new operations for lasso and semantic selection tools. - Created UI components for candidate review and inpaint region management. - Added tests for inpaint region commands to ensure functionality. - Updated various components to support new inpaint features and improve user experience.
73 lines
3.4 KiB
TypeScript
73 lines
3.4 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
|
import type { ImageDocument } from "@core/document";
|
|
import { initialToolState } from "@editor/tools";
|
|
import { checkGenerationPreconditions } from "./preconditions";
|
|
|
|
describe("generation preconditions", () => {
|
|
test("allows text-to-image with a prompt and artboard but no layer selection", () => {
|
|
expect(checkGenerationPreconditions(document(), { artboardId: "artboard", layerIds: [] }, settings("text-to-image"))).toEqual({ ready: true });
|
|
});
|
|
|
|
test("requires exactly one source layer for image-to-image and outpaint", () => {
|
|
const selection = { artboardId: "artboard", layerIds: [] };
|
|
expect(checkGenerationPreconditions(document(), selection, settings("image-to-image"))).toEqual({ ready: false, message: "Select exactly one image or raster layer for image-to-image generation." });
|
|
expect(checkGenerationPreconditions(document(), selection, settings("outpaint"))).toEqual({ ready: false, message: "Select exactly one image or raster layer to outpaint." });
|
|
});
|
|
|
|
test("requires an enabled mask for inpaint", () => {
|
|
expect(checkGenerationPreconditions(document(), selected(), settings("inpaint"))).toEqual({ ready: false, message: "Paint an AI edit region over the area you want to replace.", repair: "add-mask" });
|
|
});
|
|
|
|
test("allows inpaint when the selected source has an aligned enabled mask", () => {
|
|
expect(checkGenerationPreconditions(document(true), selected(), settings("inpaint"))).toEqual({ ready: true });
|
|
});
|
|
|
|
test("explains prompt and outpaint-padding requirements", () => {
|
|
expect(checkGenerationPreconditions(document(), selected(), { ...settings("image-to-image"), prompt: " " })).toEqual({ ready: false, message: "Enter a prompt to generate an image." });
|
|
expect(checkGenerationPreconditions(document(), selected(), { ...settings("outpaint"), outpaint: { left: 0, top: 0, right: 0, bottom: 0, feathering: 0 } })).toEqual({ ready: false, message: "Add outpaint padding on at least one side.", repair: "set-outpaint-padding" });
|
|
});
|
|
});
|
|
|
|
function settings(mode: "text-to-image" | "image-to-image" | "inpaint" | "outpaint") {
|
|
return { ...initialToolState.generate, mode, prompt: "A lighthouse" };
|
|
}
|
|
|
|
function selected() {
|
|
return { artboardId: "artboard" as const, layerIds: ["source"] };
|
|
}
|
|
|
|
function document(masked = false): ImageDocument {
|
|
const transform = { position: { x: 0, y: 0 }, scale: { x: 1, y: 1 }, rotation: 0 };
|
|
return {
|
|
id: "document",
|
|
name: "Test",
|
|
version: 1,
|
|
assets: [
|
|
{ id: "source-asset", name: "Source", mimeType: "image/png", source: "source", intrinsicSize: { w: 100, h: 100 } },
|
|
{ id: "mask-asset", name: "Mask", mimeType: "image/png", source: "mask", intrinsicSize: { w: 100, h: 100 } },
|
|
],
|
|
inpaintRegions: masked ? [{ id: "region", name: "AI edit", targetLayerId: "source", maskAssetId: "mask-asset", enabled: true }] : [],
|
|
artboards: [{
|
|
id: "artboard",
|
|
name: "Artboard",
|
|
bounds: { x: 0, y: 0, w: 100, h: 100 },
|
|
backgroundColor: "transparent",
|
|
visible: true,
|
|
locked: false,
|
|
layers: [
|
|
{ id: "mask", type: "raster", name: "Mask", visible: true, locked: false, opacity: 1, assetId: "mask-asset", transform },
|
|
{
|
|
id: "source",
|
|
type: "raster",
|
|
name: "Source",
|
|
visible: true,
|
|
locked: false,
|
|
opacity: 1,
|
|
assetId: "source-asset",
|
|
transform,
|
|
},
|
|
],
|
|
}],
|
|
};
|
|
}
|