Update graph.logic tests

JoshuaCWebDeveloper · Jun 11, 2024 · 9069880 · 9069880
1 parent be22edb
commit 9069880
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/packages/flow-client/src/app/redux/modules/flow/graph.logic.spec.ts b/packages/flow-client/src/app/redux/modules/flow/graph.logic.spec.ts
@@ -85,7 +85,14 @@ describe('graph.logic', () => {
     beforeEach(() => {
         // Reset mocks before each test
         vi.clearAllMocks();
-        mockNodeLogic = {} as NodeLogic;
+        mockNodeLogic = {
+            selectSubflowEntitiesAsPaletteNodes: vi.fn(
+                () => ({})
+            ) as unknown as NodeLogic['selectSubflowEntitiesAsPaletteNodes'],
+            selectInOutPaletteNodeEntities: vi.fn(
+                () => ({})
+            ) as unknown as NodeLogic['selectInOutPaletteNodeEntities'],
+        } as NodeLogic;
         graphLogic = new GraphLogic(mockNodeLogic);
     });
 
@@ -410,9 +417,6 @@ describe('graph.logic', () => {
             mockedSelectPaletteNodeEntities.mockImplementation(() => ({
                 node2: mockNodeEntity,
             }));
-            mockNodeLogic.selectSubflowEntitiesAsPaletteNodes = vi.fn(
-                () => ({})
-            ) as unknown as NodeLogic['selectSubflowEntitiesAsPaletteNodes'];
             mockedSelectFlowEntityById.mockImplementation(() => mockFlow);
             mockedSelectFlowNodesByFlowId.mockImplementation(() => mockNodes);