Browse Source

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 <noreply@anthropic.com>
handoff-20260429-1057
shrimbly 5 months ago
parent
commit
50662e3cb3
  1. 16
      src/components/__tests__/ConnectionDropMenu.test.tsx
  2. 1
      src/components/__tests__/WorkflowCanvas.test.tsx

16
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(<ConnectionDropMenu {...defaultProps} handleType="text" connectionType="source" />);
// 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");

1
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<string>(),
captureSnapshot: vi.fn(),
applyEditOperations: vi.fn(() => ({ applied: 0, skipped: [] })),
setWorkflowMetadata: vi.fn(),

Loading…
Cancel
Save