diff --git a/src/components/FloatingActionBar.tsx b/src/components/FloatingActionBar.tsx index cd25efae..657d966d 100644 --- a/src/components/FloatingActionBar.tsx +++ b/src/components/FloatingActionBar.tsx @@ -14,6 +14,7 @@ const ALL_NODES_CATEGORIES: { label: string; nodes: { type: NodeType; label: str nodes: [ { type: "imageInput", label: "Image Input" }, { type: "audioInput", label: "Audio Input" }, + { type: "videoInput", label: "Video Input" }, { type: "glbViewer", label: "3D Viewer" }, ], }, @@ -416,6 +417,7 @@ export function FloatingActionBar() {
+ diff --git a/src/components/__tests__/FloatingActionBar.test.tsx b/src/components/__tests__/FloatingActionBar.test.tsx index 991264d2..4295500d 100644 --- a/src/components/__tests__/FloatingActionBar.test.tsx +++ b/src/components/__tests__/FloatingActionBar.test.tsx @@ -284,7 +284,7 @@ describe("FloatingActionBar", () => { // Dropdown menu items should appear expect(screen.getByText("Image", { selector: "button.w-full" })).toBeInTheDocument(); - expect(screen.getByText("Video")).toBeInTheDocument(); + expect(screen.getByText("Video", { selector: "button.w-full" })).toBeInTheDocument(); expect(screen.getByText("Text (LLM)")).toBeInTheDocument(); }); @@ -321,7 +321,7 @@ describe("FloatingActionBar", () => { }); fireEvent.click(screen.getByText("Generate")); - fireEvent.click(screen.getByText("Video")); + fireEvent.click(screen.getByText("Video", { selector: "button.w-full" })); expect(mockAddNode).toHaveBeenCalledWith("generateVideo", expect.any(Object)); }); @@ -357,13 +357,13 @@ describe("FloatingActionBar", () => { fireEvent.click(screen.getByText("Generate")); // Verify dropdown is open - expect(screen.getByText("Video")).toBeInTheDocument(); + expect(screen.getByText("Video", { selector: "button.w-full" })).toBeInTheDocument(); // Click an option - fireEvent.click(screen.getByText("Video")); + fireEvent.click(screen.getByText("Video", { selector: "button.w-full" })); // Dropdown should close - expect(screen.queryByText("Video")).not.toBeInTheDocument(); + expect(screen.queryByText("Video", { selector: "button.w-full" })).not.toBeInTheDocument(); }); });