Spaces:
Sleeping
Sleeping
File size: 2,630 Bytes
05c5ed5 | 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 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 | import { describe, it, expect } from "vitest";
import { Edge } from "@xyflow/react";
import { extractWorkflowDiff } from "./extract-workflow-diff";
import { UINode, NodeKind } from "./workflow.interface";
describe("extractWorkflowDiff", () => {
const createTestNode = (
id: string,
name: string,
position = { x: 0, y: 0 },
): UINode => ({
id,
type: "default",
position,
data: {
id,
name,
kind: NodeKind.Input,
outputSchema: { type: "object", properties: {} },
runtime: {},
},
});
const createTestEdge = (
id: string,
source: string,
target: string,
): Edge => ({
id,
source,
target,
});
it("should detect added nodes and edges", () => {
const oldData = {
nodes: [createTestNode("node1", "Node 1")],
edges: [createTestEdge("edge1", "node1", "node2")],
};
const newData = {
nodes: [
createTestNode("node1", "Node 1"),
createTestNode("node2", "Node 2", { x: 100, y: 100 }),
],
edges: [
createTestEdge("edge1", "node1", "node2"),
createTestEdge("edge2", "node2", "node3"),
],
};
const result = extractWorkflowDiff(oldData, newData);
expect(result.deleteNodes).toHaveLength(0);
expect(result.deleteEdges).toHaveLength(0);
expect(result.updateNodes).toHaveLength(1);
expect(result.updateEdges).toHaveLength(1);
expect(result.updateNodes[0].id).toBe("node2");
expect(result.updateEdges[0].id).toBe("edge2");
});
it("should detect deleted and updated nodes and edges", () => {
const oldData = {
nodes: [
createTestNode("node1", "Node 1"),
createTestNode("node2", "Node 2"),
createTestNode("node3", "Node 3", { x: 50, y: 50 }),
],
edges: [
createTestEdge("edge1", "node1", "node2"),
createTestEdge("edge2", "node2", "node3"),
],
};
const newData = {
nodes: [
createTestNode("node1", "Node 1 Updated", { x: 10, y: 10 }),
createTestNode("node3", "Node 3", { x: 50, y: 50 }),
],
edges: [createTestEdge("edge1", "node1", "node3")],
};
const result = extractWorkflowDiff(oldData, newData);
expect(result.deleteNodes).toHaveLength(1);
expect(result.deleteNodes[0].id).toBe("node2");
expect(result.deleteEdges).toHaveLength(1);
expect(result.deleteEdges[0].id).toBe("edge2");
expect(result.updateNodes).toHaveLength(1);
expect(result.updateNodes[0].id).toBe("node1");
expect(result.updateEdges).toHaveLength(1);
expect(result.updateEdges[0].id).toBe("edge1");
});
});
|