diff --git a/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/CollapsibleMessage.test.tsx b/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/CollapsibleMessage.test.tsx index 1477febd967..3aa8d21c31c 100644 --- a/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/CollapsibleMessage.test.tsx +++ b/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/CollapsibleMessage.test.tsx @@ -37,4 +37,18 @@ describe("CollapsibleMessage", () => { await user.click(screen.getByText("SYSTEM")); expect(screen.getByText("Toggle me")).toBeInTheDocument(); }); + + it("should expand with Enter and collapse with Space from the keyboard", async () => { + const user = userEvent.setup(); + render(); + + expect(screen.getByText("Toggle me")).not.toBeVisible(); + + await user.tab(); + await user.keyboard("{Enter}"); + expect(screen.getByText("Toggle me")).toBeVisible(); + + await user.keyboard(" "); + expect(screen.getByText("Toggle me")).not.toBeVisible(); + }); }); diff --git a/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/HistoryTree.test.tsx b/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/HistoryTree.test.tsx index 10d6bfa0847..0726a9f4b82 100644 --- a/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/HistoryTree.test.tsx +++ b/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/HistoryTree.test.tsx @@ -41,4 +41,22 @@ describe("HistoryTree", () => { expect(screen.getByText("Hello")).toBeInTheDocument(); expect(screen.getByText("Hi there")).toBeInTheDocument(); }); + + it("should expand with Enter and collapse with Space from the keyboard", async () => { + const user = userEvent.setup(); + const messages: ParsedMessage[] = [ + { role: "user", content: "Hello" }, + { role: "assistant", content: "Hi there" }, + ]; + render(); + + expect(screen.getByText("Hello")).not.toBeVisible(); + + await user.tab(); + await user.keyboard("{Enter}"); + expect(screen.getByText("Hello")).toBeVisible(); + + await user.keyboard(" "); + expect(screen.getByText("Hello")).not.toBeVisible(); + }); });