MajoorWaldi
71ef418bdd
- Added functions to manage sibling asset states and determine if PNG siblings should be hidden based on their context (video, audio, model3D). - Refactored asset appending logic to utilize the new sibling hiding functionality. - Updated the backend to prioritize image assets over video assets in the same stack group. - Introduced a new execution asset buffer to manage job-related assets efficiently. - Created tests for sibling hiding logic and execution asset buffer functionality.
41 lines
1.5 KiB
JavaScript
41 lines
1.5 KiB
JavaScript
import { describe, expect, it } from "vitest";
|
|
|
|
import { createExecutionAssetBuffer } from "../features/stacks/executionAssetBuffer.js";
|
|
|
|
describe("createExecutionAssetBuffer", () => {
|
|
it("buffers by job id and deduplicates files by output identity", () => {
|
|
const buffer = createExecutionAssetBuffer();
|
|
|
|
buffer.add("job-1", [
|
|
{ filename: "a.png", subfolder: "x", type: "output", source_node_id: "1" },
|
|
]);
|
|
const current = buffer.add("job-1", [
|
|
{ filename: "a.png", subfolder: "x", type: "output", source_node_type: "SaveImage" },
|
|
{ filename: "b.png", subfolder: "x", type: "output" },
|
|
]);
|
|
|
|
expect(current).toHaveLength(2);
|
|
expect(buffer.take("job-1")).toEqual([
|
|
{
|
|
filename: "a.png",
|
|
subfolder: "x",
|
|
type: "output",
|
|
source_node_id: "1",
|
|
source_node_type: "SaveImage",
|
|
},
|
|
{ filename: "b.png", subfolder: "x", type: "output" },
|
|
]);
|
|
});
|
|
|
|
it("clears a failed job without affecting others", () => {
|
|
const buffer = createExecutionAssetBuffer();
|
|
buffer.add("job-1", [{ filename: "a.png", type: "output" }]);
|
|
buffer.add("job-2", [{ filename: "b.png", type: "output" }]);
|
|
|
|
buffer.clear("job-1");
|
|
|
|
expect(buffer.take("job-1")).toEqual([]);
|
|
expect(buffer.take("job-2")).toEqual([{ filename: "b.png", type: "output" }]);
|
|
});
|
|
});
|