From 1bfd61726570548749549f9e1f708103d83138fa Mon Sep 17 00:00:00 2001
From: Csongor Czezar <csongor.czezar@gmail.com>
Date: Fri, 2 Jan 2026 14:17:38 -0800
Subject: [PATCH] refactor: moving mock functions out of global

---
 .../tests/store/executionStore.executingNode.test.ts  | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/tests-ui/tests/store/executionStore.executingNode.test.ts b/tests-ui/tests/store/executionStore.executingNode.test.ts
index 0fbec3367..005ecc084 100644
--- a/tests-ui/tests/store/executionStore.executingNode.test.ts
+++ b/tests-ui/tests/store/executionStore.executingNode.test.ts
@@ -17,16 +17,10 @@ const createMockRootGraph = (
   ...options
 })
 
-// Create mock functions
-const mockNodeExecutionIdToNodeLocatorId = vi.fn()
-const mockNodeIdToNodeLocatorId = vi.fn()
-const mockNodeLocatorIdToNodeExecutionId = vi.fn()
-
 // Keep track of the current mock root graph
 const mockAppState = {
   rootGraph: createMockRootGraph()
 }
-
 // Mock the app import and its rootGraph property
 vi.mock('@/scripts/app', () => ({
   app: {
@@ -40,7 +34,6 @@ vi.mock('@/scripts/app', () => ({
     nodePreviewImages: {}
   }
 }))
-
 // Mock the workflowStore
 vi.mock(
   '@/platform/workflow/management/stores/workflowStore',
@@ -49,6 +42,10 @@ vi.mock(
       await importOriginal<
         typeof import('@/platform/workflow/management/stores/workflowStore')
       >()
+    const mockNodeExecutionIdToNodeLocatorId = vi.fn()
+    const mockNodeIdToNodeLocatorId = vi.fn()
+    const mockNodeLocatorIdToNodeExecutionId = vi.fn()
+
     return {
       ComfyWorkflow,
       useWorkflowStore: vi.fn(() => ({