kandev/apps/web/components/task/file-browser-toolbar.test.tsx

import { useState } from "react";
import { afterEach, describe, expect, it, vi } from "vitest";
import { cleanup, fireEvent, render, screen, waitFor } from "@testing-library/react";
import { TooltipProvider } from "@kandev/ui/tooltip";
import { FileBrowserToolbar } from "./file-browser-toolbar";
import { InlineFileInput } from "./inline-file-input";

let isMobile = false;

vi.mock("@/hooks/use-responsive-breakpoint", () => ({
  useResponsiveBreakpoint: () => ({ isMobile }),
}));

afterEach(() => {
  cleanup();
  isMobile = false;
});

describe("FileBrowserToolbar workspace actions", () => {
  it("keeps Open workspace folder available while Add sources explains why it is unavailable", async () => {
    const onAddSources = vi.fn();
    const onOpenFolder = vi.fn();

    render(
      <TooltipProvider>
        <FileBrowserToolbar
          displayPath="workspace"
          fullPath="/workspace"
          copied={false}
          expandedPathsSize={0}
          onCopyPath={vi.fn()}
          onOpenFolder={onOpenFolder}
          onStartSearch={vi.fn()}
          onCollapseAll={vi.fn()}
          showCreateButton={false}
          onAddSources={onAddSources}
          addSourcesDisabledReason="Wait for the active agent turn to finish"
        />
      </TooltipProvider>,
    );

    const trigger = screen.getByRole("button", { name: "Workspace actions" });
    fireEvent.pointerDown(trigger, { button: 0, ctrlKey: false });
    fireEvent.click(trigger);

    const addSources = screen.getByRole("menuitem", {
      name: /Add Repositories to workspace/i,
    });
    expect(addSources.hasAttribute("data-disabled")).toBe(true);
    expect(addSources.className).toContain("min-h-11");
    expect(screen.getByText("Wait for the active agent turn to finish")).toBeTruthy();
    fireEvent.click(screen.getByRole("menuitem", { name: "Open workspace folder" }));
    await waitFor(() => expect(onOpenFolder).toHaveBeenCalledOnce());
    expect(onOpenFolder).toHaveBeenCalledWith(trigger);
    expect(onAddSources).not.toHaveBeenCalled();
  });

  it("opens Add sources only after the menu closes and uses the combined trigger as its opener", async () => {
    const onAddSources = vi.fn();
    const openerRef = { current: null as HTMLButtonElement | null };

    render(
      <TooltipProvider>
        <FileBrowserToolbar
          displayPath="workspace"
          fullPath="/workspace"
          copied={false}
          expandedPathsSize={0}
          onCopyPath={vi.fn()}
          onOpenFolder={vi.fn()}
          onStartSearch={vi.fn()}
          onCollapseAll={vi.fn()}
          showCreateButton={false}
          onAddSources={onAddSources}
          addSourcesButtonRef={openerRef}
        />
      </TooltipProvider>,
    );

    const trigger = screen.getByRole("button", { name: "Workspace actions" });
    fireEvent.pointerDown(trigger, { button: 0, ctrlKey: false });
    fireEvent.click(trigger);
    fireEvent.click(screen.getByRole("menuitem", { name: "Add Repositories to workspace" }));

    await waitFor(() => expect(onAddSources).toHaveBeenCalledWith(trigger));
    expect(openerRef.current).toBe(trigger);
  });

  it("restores the mobile trigger after the source Drawer finishes closing", async () => {
    isMobile = true;
    const onAddSources = vi.fn();
    render(
      <TooltipProvider>
        <FileBrowserToolbar
          displayPath="workspace"
          fullPath="/workspace"
          copied={false}
          expandedPathsSize={0}
          onCopyPath={vi.fn()}
          onOpenFolder={vi.fn()}
          onStartSearch={vi.fn()}
          onCollapseAll={vi.fn()}
          showCreateButton={false}
          onAddSources={onAddSources}
        />
      </TooltipProvider>,
    );

    const trigger = screen.getByRole("button", { name: "Workspace actions" });
    fireEvent.pointerDown(trigger, { button: 0, ctrlKey: false });
    fireEvent.click(trigger);
    fireEvent.click(screen.getByRole("menuitem", { name: "Add Repositories to workspace" }));
    await waitFor(() => expect(onAddSources).toHaveBeenCalledOnce());

    const drawer = document.createElement("div");
    drawer.dataset.testid = "add-workspace-sources-drawer";
    drawer.dataset.state = "closed";
    document.body.append(drawer);
    fireEvent.animationEnd(drawer);

    await waitFor(() => expect(document.activeElement).toBe(trigger));
    drawer.remove();
  });
});

const CREATE_MENU_TESTID = "files-create-menu";
const createMenuBaseProps = {
  displayPath: "workspace",
  fullPath: "/workspace",
  copied: false,
  expandedPathsSize: 0,
  onCopyPath: vi.fn(),
  onOpenFolder: vi.fn(),
  onStartSearch: vi.fn(),
  onCollapseAll: vi.fn(),
};

async function openCreateMenu(trigger: HTMLElement) {
  fireEvent.pointerDown(trigger, { button: 0, ctrlKey: false });
  fireEvent.click(trigger);
  return screen.findByRole("menuitem", { name: "New file" });
}

function CreateMenuInlineInputHarness() {
  const [creating, setCreating] = useState(false);

  return (
    <TooltipProvider>
      <FileBrowserToolbar
        {...createMenuBaseProps}
        showCreateButton
        onStartCreate={() => setCreating(true)}
        onUploadFiles={vi.fn()}
      />
      {creating && (
        <InlineFileInput depth={0} onSubmit={vi.fn()} onCancel={() => setCreating(false)} />
      )}
    </TooltipProvider>
  );
}

describe("FileBrowserToolbar create menu", () => {
  it("keeps one-click New File when upload is unavailable", () => {
    const onStartCreate = vi.fn();
    render(
      <TooltipProvider>
        <FileBrowserToolbar
          {...createMenuBaseProps}
          showCreateButton
          onStartCreate={onStartCreate}
        />
      </TooltipProvider>,
    );

    fireEvent.click(screen.getByRole("button", { name: "New file" }));
    expect(onStartCreate).toHaveBeenCalledTimes(1);
    expect(screen.queryByTestId(CREATE_MENU_TESTID)).toBeNull();
  });

  it("offers New File, Upload files, and Upload folder when upload is available", async () => {
    const onStartCreate = vi.fn();
    const onUploadFiles = vi.fn();
    render(
      <TooltipProvider>
        <FileBrowserToolbar
          {...createMenuBaseProps}
          showCreateButton
          onStartCreate={onStartCreate}
          onUploadFiles={onUploadFiles}
        />
      </TooltipProvider>,
    );

    const menuTrigger = screen.getByTestId(CREATE_MENU_TESTID);
    await openCreateMenu(menuTrigger);
    expect(screen.getByRole("menuitem", { name: "Upload files" })).toBeTruthy();
    expect(screen.getByRole("menuitem", { name: "Upload folder" })).toBeTruthy();
  });

  it("keeps inline creation focused after the create menu closes", async () => {
    // @covers AC-UI-WORKSPACE-FILE-TRANSFER-001.1
    render(<CreateMenuInlineInputHarness />);

    const menuTrigger = screen.getByTestId(CREATE_MENU_TESTID);
    fireEvent.click(await openCreateMenu(menuTrigger));
    await waitFor(() => expect(screen.queryByRole("menuitem", { name: "New file" })).toBeNull());

    const input = await screen.findByPlaceholderText("filename...");
    await waitFor(() => expect(document.activeElement).toBe(input));

    fireEvent.keyDown(input, { key: "Escape" });
    await waitFor(() => expect(screen.queryByPlaceholderText("filename...")).toBeNull());
    await openCreateMenu(menuTrigger);
    fireEvent.keyDown(document.activeElement ?? document.body, { key: "Escape" });
    await waitFor(() => expect(screen.queryByRole("menuitem", { name: "New file" })).toBeNull());
    expect(screen.queryByPlaceholderText("filename...")).toBeNull();
  });

  it("requests the right picker for each upload item", async () => {
    const onUploadFiles = vi.fn();
    const onStartCreate = vi.fn();
    render(
      <TooltipProvider>
        <FileBrowserToolbar
          {...createMenuBaseProps}
          showCreateButton
          onStartCreate={onStartCreate}
          onUploadFiles={onUploadFiles}
        />
      </TooltipProvider>,
    );

    const menuTrigger = screen.getByTestId(CREATE_MENU_TESTID);
    await openCreateMenu(menuTrigger);
    fireEvent.click(await screen.findByRole("menuitem", { name: "Upload folder" }));
    await waitFor(() => expect(onUploadFiles).toHaveBeenCalledWith("folder"));
    await waitFor(() => expect(document.activeElement).toBe(menuTrigger));
    expect(onStartCreate).not.toHaveBeenCalled();

    onUploadFiles.mockClear();
    await openCreateMenu(menuTrigger);
    fireEvent.click(await screen.findByRole("menuitem", { name: "Upload files" }));
    await waitFor(() => expect(onUploadFiles).toHaveBeenCalledWith("files"));
    await waitFor(() => expect(document.activeElement).toBe(menuTrigger));
    expect(onStartCreate).not.toHaveBeenCalled();
  });

  it("restores the create-menu trigger when the menu is dismissed", async () => {
    const onStartCreate = vi.fn();
    render(
      <TooltipProvider>
        <FileBrowserToolbar
          {...createMenuBaseProps}
          showCreateButton
          onStartCreate={onStartCreate}
          onUploadFiles={vi.fn()}
        />
      </TooltipProvider>,
    );

    const menuTrigger = screen.getByTestId(CREATE_MENU_TESTID);
    await openCreateMenu(menuTrigger);
    fireEvent.keyDown(document.activeElement ?? document.body, { key: "Escape" });

    await waitFor(() => expect(screen.queryByRole("menuitem", { name: "New file" })).toBeNull());
    await waitFor(() => expect(document.activeElement).toBe(menuTrigger));
    expect(onStartCreate).not.toHaveBeenCalled();
  });
});