src/pages/VisProgPage/visualProgrammingUI/VisProgStores.tsx

@@ -196,17 +196,20 @@ const useFlowStore = create<FlowState>(UndoRedo((set, get) => ({
 
     // If there's no function, OR, our function tells us we can delete it, let's do so...
     if (ourFunction == undefined || ourFunction()) {
-      deleteElements({
-        nodes: get().nodes.filter((n) => n.id === nodeId),
-        edges: get().edges.filter((e) => e.source !== nodeId && e.target === nodeId)}
-      ).then(() => {
-        get().unregisterNodeRules(nodeId);
-        get().unregisterWarningsForId(nodeId);
-      });
-      // set({
-      // nodes: get().nodes.filter((n) => n.id !== nodeId),
-      // edges: get().edges.filter((e) => e.source !== nodeId && e.target !== nodeId),
-      // })
+      if (deleteElements){
+        deleteElements({
+          nodes: get().nodes.filter((n) => n.id === nodeId),
+          edges: get().edges.filter((e) => e.source !== nodeId && e.target === nodeId)}
+        ).then(() => {
+          get().unregisterNodeRules(nodeId);
+          get().unregisterWarningsForId(nodeId);
+        });
+      } else {
+        set({
+        nodes: get().nodes.filter((n) => n.id !== nodeId),
+        edges: get().edges.filter((e) => e.source !== nodeId && e.target !== nodeId),
+        })
+      }
     }
   },
 

src/pages/VisProgPage/visualProgrammingUI/VisProgTypes.tsx

@@ -69,7 +69,7 @@ export type FlowState = {
    * Deletes a node and any connected edges.
    * @param nodeId - the ID of the node to delete
    */
-  deleteNode: (nodeId: string,  deleteElements: (params: DeleteElementsOptions) => Promise<{
+  deleteNode: (nodeId: string,  deleteElements?: (params: DeleteElementsOptions) => Promise<{
     deletedNodes: Node[]
     deletedEdges: Edge[]
   }>) => void;

test/pages/visProgPage/visualProgrammingUI/VisProgStores.test.tsx

@@ -3,7 +3,6 @@ import {
   type Connection,
   type Edge,
   type Node,
-  useReactFlow
 } from "@xyflow/react";
 import type {HandleRule, RuleResult} from "../../../../src/pages/VisProgPage/visualProgrammingUI/HandleRuleLogic.ts";
 import { NodeDisconnections } from "../../../../src/pages/VisProgPage/visualProgrammingUI/NodeRegistry.ts";
@@ -402,9 +401,10 @@ describe('FlowStore Functionality', () => {
             target: 'B'
           }]
       });
-      const {deleteElements} = useReactFlow();
+
+
       act(()=> {
-        deleteNode(nodeId, deleteElements);
+        deleteNode(nodeId);
       });
 
       const updatedState = useFlowStore.getState();