From 68ef595de4bfbfd539aa572563180f4a2f3f9673 Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Thu, 12 Oct 2023 20:15:16 +0100
Subject: [PATCH] Few tweaks

---
 tests-ui/utils/ezgraph.js | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/tests-ui/utils/ezgraph.js b/tests-ui/utils/ezgraph.js
index a91602c60..1c6e27fef 100644
--- a/tests-ui/utils/ezgraph.js
+++ b/tests-ui/utils/ezgraph.js
@@ -9,7 +9,7 @@
  * @typedef { import("../../web/types/litegraph").INodeInputSlot } INodeInputSlot
  * @typedef { import("../../web/types/litegraph").INodeOutputSlot } INodeOutputSlot
  * @typedef { InstanceType<LG["LGraphNode"]> & { widgets?: Array<IWidget> } } LGNode
- * @typedef { (...args: EzOutput[] | [...EzOutput[], Record<string, unknown>]) => Array<EzOutput> & { $: EzNode, node: LG["LGraphNode"]} } EzNodeFactory
+ * @typedef { (...args: EzOutput[] | [...EzOutput[], Record<string, unknown>]) => EzNode } EzNodeFactory
  */
 
 class EzConnection {
@@ -175,6 +175,13 @@ class EzWidget {
 		return this.widget.type === "converted-widget";
 	}
 
+	getConvertedInput() {
+		if (!this.isConvertedToInput)
+			throw new Error(`Widget ${this.widget.name} is not converted to input.`);
+		
+		return this.node.inputs.find(inp => inp.input["widget"]?.name === this.widget.name);
+	}
+
 	convertToWidget() {
 		if (!this.isConvertedToInput)
 			throw new Error(`Widget ${this.widget.name} cannot be converted as it is already a widget.`);
@@ -279,6 +286,10 @@ class EzGraph {
 		this.app.graph.arrange();
 	}
 
+	stringify() {
+		return JSON.stringify(this.app.graph.serialize(), undefined, "\t");
+	}
+
 	/**
 	 * @param { number | LGNode | EzNode } obj
 	 * @returns { EzNode }
@@ -359,7 +370,6 @@ export const Ez = {
 					 */
 					return function (...args) {
 						const ezNode = new EzNode(app, node);
-						const outputs = ezNode.outputs;
 						const inputs = ezNode.inputs;
 
 						let slot = 0;
@@ -374,7 +384,7 @@ export const Ez = {
 							}
 						}
 
-						return outputs;
+						return ezNode;
 					};
 				},
 			}