58 lines
1.7 KiB
TypeScript
58 lines
1.7 KiB
TypeScript
// SimpleSyrup - workflow-focused ComfyUI extensions for image generation
|
|
// Copyright (C) 2026 Artificial Sweetener and contributors
|
|
// SPDX-License-Identifier: AGPL-3.0-or-later
|
|
|
|
import { describe, expect, it } from "vitest";
|
|
|
|
import { comfyImageUrl } from "../src/comfyImageUrl";
|
|
import { parseSegPreviewDocument } from "../src/segPreviewTypes";
|
|
|
|
describe("SEG preview transport", () => {
|
|
it("validates execution payloads and uses the latest mapped document", () => {
|
|
const document = validDocument();
|
|
|
|
expect(
|
|
parseSegPreviewDocument({ simple_syrup_segs_preview: [document] })
|
|
).toEqual(document);
|
|
expect(parseSegPreviewDocument({ images: [] })).toBeUndefined();
|
|
expect(() =>
|
|
parseSegPreviewDocument({
|
|
simple_syrup_segs_preview: [{ ...document, version: 2 }]
|
|
})
|
|
).toThrow("unsupported preview payload");
|
|
});
|
|
|
|
it("builds an encoded view URL through Comfy's base-path resolver", () => {
|
|
const url = comfyImageUrl(
|
|
{ filename: "one two.png", subfolder: "a/b", type: "temp" },
|
|
(path) => `/comfy${path}`
|
|
);
|
|
|
|
expect(url).toBe(
|
|
"/comfy/view?filename=one+two.png&subfolder=a%2Fb&type=temp"
|
|
);
|
|
});
|
|
});
|
|
|
|
function validDocument(): object {
|
|
const image = { filename: "asset.png", subfolder: "", type: "temp" };
|
|
return {
|
|
version: 1,
|
|
source: { width: 10, height: 8 },
|
|
preview: { width: 10, height: 8, image },
|
|
atlas: { width: 4, height: 4, image },
|
|
regions: [
|
|
{
|
|
id: "seg-0001",
|
|
index: 0,
|
|
label: "subject",
|
|
confidence: 0.9,
|
|
area: 8,
|
|
color: "#f24236",
|
|
crop: { x: 1, y: 1, width: 4, height: 2 },
|
|
atlas: { x: 0, y: 0, width: 4, height: 2 }
|
|
}
|
|
]
|
|
};
|
|
}
|