Fix/add test for execution (#665)

* Add changeset

* Add some additional tests and fix import issue

Author: SorsOps

.changeset/large-dogs-melt.md

@@ -0,0 +1,5 @@
+---
+"@tokens-studio/graph-editor": patch
+---
+
+Fixes an issue with the editor losing the reference to the graph frame when moving between frames

packages/graph-engine/package.json

@@ -9,6 +9,10 @@
     ".": {
       "import": "./dist/index.js",
       "types": "./dist/index.d.ts"
+    },
+    "./nodes/*":{
+      "import": "./dist/nodes/*",
+      "types": "./dist/nodes/*"
     }
   },
   "repository": {

packages/graph-engine/tests/suites/graph/execution.test.ts

@@ -1,6 +1,8 @@
 import { Graph } from '../../../src/graph/graph.js';
 import { Node, NumberSchema } from '../../../src/index.js';
 import { describe, expect, test } from 'vitest';
+import Add from '../../../src/nodes/math/add.js';
+import Output from '../../../src/nodes/generic/output.js';
 
 class ExternalNode extends Node {
 	constructor(props) {
@@ -45,4 +47,31 @@ describe('Graph/execution', () => {
 
 		expect(graph.nodes.a.outputs.foo.value).toEqual(5);
 	});
+
+	test('works without an input node', async () => {
+		const graph = new Graph();
+
+		const a = new Add({ id: 'a', graph });
+		const b = new Add({ id: 'b', graph });
+
+		a.inputs.a.setValue(2);
+		a.inputs.b.setValue(3);
+
+		a.outputs.value.connect(b.inputs.a);
+		b.inputs.b.setValue(5);
+
+		const output = new Output({ id: 'c', graph });
+
+		output.addInput('value', { type: NumberSchema });
+
+		b.outputs.value.connect(output.inputs.value);
+
+		graph.addNode(a);
+		graph.addNode(b);
+		graph.addNode(output);
+
+		const result = await graph.execute();
+
+		expect(result.output!.value.value).toEqual(10);
+	});
 });

packages/graph-engine/tests/suites/nodes/logic/and.test.ts

@@ -0,0 +1,53 @@
+import { Graph } from '../../../../src/graph/graph.js';
+import { describe, expect, test } from 'vitest';
+import Node from '../../../../src/nodes/logic/and.js';
+
+describe('logic/and', () => {
+	test('returns true when all inputs are true', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.inputs.inputs.setValue([true, true, true]);
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBe(true);
+	});
+
+	test('returns false when any input is false', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.inputs.inputs.setValue([true, false, true]);
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBe(false);
+	});
+
+	test('returns false when all inputs are false', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.inputs.inputs.setValue([false, false, false]);
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBe(false);
+	});
+
+	test('returns true when inputs array is empty', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.inputs.inputs.setValue([]);
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBe(true);
+	});
+});

packages/graph-engine/tests/suites/nodes/logic/or.test.ts

@@ -0,0 +1,53 @@
+import { Graph } from '../../../../src/graph/graph.js';
+import { describe, expect, test } from 'vitest';
+import Node from '../../../../src/nodes/logic/or.js';
+
+describe('logic/or', () => {
+	test('returns true when any input is true', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.inputs.inputs.setValue([false, true, false]);
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBe(true);
+	});
+
+	test('returns false when all inputs are false', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.inputs.inputs.setValue([false, false, false]);
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBe(false);
+	});
+
+	test('returns true when all inputs are true', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.inputs.inputs.setValue([true, true, true]);
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBe(true);
+	});
+
+	test('returns false when inputs array is empty', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.inputs.inputs.setValue([]);
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBe(false);
+	});
+});

packages/graph-engine/tests/suites/nodes/logic/switch.test.ts

@@ -0,0 +1,96 @@
+import { AnySchema, ToInput } from '../../../../src/index.js';
+import { Graph } from '../../../../src/graph/graph.js';
+import { describe, expect, test } from 'vitest';
+import Node from '../../../../src/nodes/logic/switch.js';
+
+describe('logic/switch', () => {
+	test('outputs the value of the matching input', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.addInput('foo', { type: AnySchema });
+		node.addInput('bar', { type: AnySchema });
+		node.addInput('default', { type: AnySchema });
+
+		const inputs = node.inputs as ToInput<{
+			foo: any;
+			bar: any;
+			default: any;
+			condition: string;
+		}>;
+
+		inputs.foo.setValue('fooValue');
+		inputs.bar.setValue('barValue');
+		inputs.default.setValue('defaultValue');
+		inputs.condition.setValue('foo');
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBe('fooValue');
+	});
+
+	test('outputs the default value when no condition matches', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.addInput('foo', { type: AnySchema });
+		node.addInput('bar', { type: AnySchema });
+		node.addInput('default', { type: AnySchema });
+
+		const inputs = node.inputs as ToInput<{
+			foo: any;
+			bar: any;
+			default: any;
+			condition: string;
+		}>;
+
+		inputs.foo.setValue('fooValue');
+		inputs.bar.setValue('barValue');
+		inputs.default.setValue('defaultValue');
+		inputs.condition.setValue('baz');
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBe('defaultValue');
+	});
+
+	test('outputs undefined when no inputs and no default value', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.inputs.condition.setValue('foo');
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBeUndefined();
+	});
+
+	test('outputs the value of the matching input with dynamic inputs', async () => {
+		const graph = new Graph();
+		const node = new Node({ graph });
+
+		node.addInput('dynamic1', { type: AnySchema });
+		node.addInput('dynamic2', { type: AnySchema });
+		node.addInput('default', { type: AnySchema });
+
+		const inputs = node.inputs as ToInput<{
+			dynamic1: any;
+			dynamic2: any;
+			default: any;
+			condition: string;
+		}>;
+
+		inputs.dynamic1.setValue('dynamic1Value');
+		inputs.dynamic2.setValue('dynamic2Value');
+		inputs.default.setValue('defaultValue');
+		inputs.condition.setValue('dynamic2');
+
+		await node.execute();
+
+		const result = node.outputs.value.value;
+		expect(result).toBe('dynamic2Value');
+	});
+});