File size: 1,454 Bytes
c6535db | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | import type {LGraphNode} from "@comfyorg/frontend";
import type {ComfyUITestEnvironment} from "./comfyui_env";
export const PNG_1x1 =
"data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVQIW2P4v5ThPwAG7wKklwQ/bwAAAABJRU5ErkJggg==";
export const PNG_1x2 =
"data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAACCAYAAACZgbYnAAAAEElEQVQIW2NgYGD4D8QM/wEHAwH/OMSHKAAAAABJRU5ErkJggg==";
export const PNG_2x1 =
"data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAIAAAABCAYAAAD0In+KAAAAD0lEQVQIW2NkYGD4D8QMAAUNAQFqjhCLAAAAAElFTkSuQmCC";
export async function pasteImageToLoadImageNode(
env: ComfyUITestEnvironment,
dataUrl?: string,
node?: LGraphNode,
) : Promise<LGraphNode> {
const dataArr = (dataUrl ?? PNG_1x1).split(",");
const mime = dataArr[0]!.match(/:(.*?);/)![1];
const bstr = atob(dataArr[1]!);
let n = bstr.length;
const u8arr = new Uint8Array(n);
while (n--) {
u8arr[n] = bstr.charCodeAt(n);
}
const filename = `test_image_${+new Date()}.png`;
const file = new File([u8arr], filename, {type: mime});
if (!node) {
node = await env.addNode("LoadImage");
}
await (node as any).pasteFiles([file]);
let i = 0;
let good = false;
while (i++ < 10 || good) {
good = node.widgets![0]!.value === filename;
if (good) break;
await env.wait(100);
}
if (!good) {
throw new Error("Expected file not loaded.");
}
return node;
}
|