From 50662e3cb300ff1c4eefa168d87a9900bd0ba979 Mon Sep 17 00:00:00 2001 From: shrimbly Date: Sat, 28 Feb 2026 23:01:21 +1300 Subject: [PATCH] fix: update tests for new menu options and dimmedNodeIds store field ConnectionDropMenu tests expected stale last items and item counts after Router, Switch, and Conditional Switch were added. WorkflowCanvas tests were missing dimmedNodeIds from the mock store state. Co-Authored-By: Claude Opus 4.6 --- .../__tests__/ConnectionDropMenu.test.tsx | 16 ++++++---------- src/components/__tests__/WorkflowCanvas.test.tsx | 1 + 2 files changed, 7 insertions(+), 10 deletions(-) diff --git a/src/components/__tests__/ConnectionDropMenu.test.tsx b/src/components/__tests__/ConnectionDropMenu.test.tsx index 9f79484e..4cf61cf2 100644 --- a/src/components/__tests__/ConnectionDropMenu.test.tsx +++ b/src/components/__tests__/ConnectionDropMenu.test.tsx @@ -218,7 +218,7 @@ describe("ConnectionDropMenu", () => { fireEvent.keyDown(document, { key: "ArrowUp" }); // Last item should now be highlighted - const lastButton = screen.getByText("Image Compare").closest("button"); + const lastButton = screen.getByText("Switch").closest("button"); expect(lastButton).toHaveClass("bg-neutral-700"); }); @@ -245,15 +245,11 @@ describe("ConnectionDropMenu", () => { it("should wrap around when navigating past last item", () => { render(); - // Text target labels: Prompt, Prompt Constructor, Array, Generate Image, Generate Video, Generate Audio, LLM Generate (7 items) - // Navigate down 7 times to wrap to first - fireEvent.keyDown(document, { key: "ArrowDown" }); - fireEvent.keyDown(document, { key: "ArrowDown" }); - fireEvent.keyDown(document, { key: "ArrowDown" }); - fireEvent.keyDown(document, { key: "ArrowDown" }); - fireEvent.keyDown(document, { key: "ArrowDown" }); - fireEvent.keyDown(document, { key: "ArrowDown" }); - fireEvent.keyDown(document, { key: "ArrowDown" }); + // Text target labels: Prompt, Prompt Constructor, Array, Generate Image, Generate Video, Generate Audio, LLM Generate, Router, Switch, Conditional Switch (10 items) + // Navigate down 10 times to wrap to first + for (let i = 0; i < 10; i++) { + fireEvent.keyDown(document, { key: "ArrowDown" }); + } // Should be back on first item (Prompt) const firstButton = screen.getByText("Prompt").closest("button"); diff --git a/src/components/__tests__/WorkflowCanvas.test.tsx b/src/components/__tests__/WorkflowCanvas.test.tsx index f62d1ff9..3570a63a 100644 --- a/src/components/__tests__/WorkflowCanvas.test.tsx +++ b/src/components/__tests__/WorkflowCanvas.test.tsx @@ -139,6 +139,7 @@ const createDefaultState = (overrides = {}) => ({ getNodesWithComments: vi.fn(() => []), markCommentViewed: vi.fn(), canvasNavigationSettings: { panMode: "space", zoomMode: "altScroll", selectionMode: "click" }, + dimmedNodeIds: new Set(), captureSnapshot: vi.fn(), applyEditOperations: vi.fn(() => ({ applied: 0, skipped: [] })), setWorkflowMetadata: vi.fn(),