diff --git a/webview-ui/src/components/chat/ModeSelector.tsx b/webview-ui/src/components/chat/ModeSelector.tsx index 7a0cb04083..529d989541 100644 --- a/webview-ui/src/components/chat/ModeSelector.tsx +++ b/webview-ui/src/components/chat/ModeSelector.tsx @@ -1,5 +1,5 @@ import React from "react" -import { ChevronUp, Check, X, Upload } from "lucide-react" +import { ChevronUp, Check, X, Upload, Download } from "lucide-react" import { cn } from "@/lib/utils" import { useRooPortal } from "@/components/ui/hooks/useRooPortal" import { Popover, PopoverContent, PopoverTrigger, StandardTooltip, Button } from "@/components/ui" @@ -368,14 +368,15 @@ export const ModeSelector = ({ setOpen(false) }} /> - {/* Import button - using IconButton for consistency */} - setShowImportDialog(true)} - disabled={isImporting} - isLoading={isImporting} - /> + disabled={isImporting}> + + {/* Info icon and title on the right - only show info icon when search bar is visible */} diff --git a/webview-ui/src/components/chat/__tests__/ModeSelector.export-import.spec.tsx b/webview-ui/src/components/chat/__tests__/ModeSelector.export-import.spec.tsx index a8f350f547..f5d54d1d76 100644 --- a/webview-ui/src/components/chat/__tests__/ModeSelector.export-import.spec.tsx +++ b/webview-ui/src/components/chat/__tests__/ModeSelector.export-import.spec.tsx @@ -213,8 +213,8 @@ describe("ModeSelector Export/Import", () => { // Import button should be visible const importButton = screen.getByRole("button", { name: "Import Mode" }) expect(importButton).toBeInTheDocument() - // Check for the icon inside the button - const icon = importButton.querySelector(".codicon-cloud-download") + // Check for the Download icon (SVG) inside the button + const icon = importButton.querySelector("svg") expect(icon).toBeInTheDocument() }) @@ -424,16 +424,17 @@ describe("ModeSelector Export/Import", () => { expect(exportButton).toHaveAttribute("aria-label", "Export Mode") }) - test("import button uses IconButton component", async () => { + test("import button has proper structure", async () => { render() // Open the popover fireEvent.click(screen.getByTestId("mode-selector-trigger")) - // Import button should have proper IconButton structure + // Import button should have proper structure const importButton = screen.getByRole("button", { name: "Import Mode" }) - expect(importButton).toHaveAttribute("aria-label", "Import Mode") - expect(importButton.querySelector(".codicon-cloud-download")).toBeInTheDocument() + expect(importButton).toHaveAttribute("title", "Import Mode") + // Check for the Download icon (SVG) inside the button + expect(importButton.querySelector("svg")).toBeInTheDocument() }) }) })