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();
});
});