AgentHub/client/src/components/run/__tests__/ChatAddMenu.test.tsx

/**
 * @file ChatAddMenu.test.tsx
 * @description The `+` menu's existing-resource pickers: Project files (a
 * search with folders, attached as `@path`), the workspace's Context files
 * (files only, none missing), this browser's Recent uploads and, for a
 * browser on the dashboard's machine, a file chosen in the OS's own picker.
 * @author Nguyễn Ngọc Trí Vĩ <vinnt@smartgift.vn>
 */

import { fireEvent, render, screen } from "@testing-library/react";
import { MemoryRouter } from "react-router-dom";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import i18n from "i18next";
import { ChatAddMenu } from "../ChatAddMenu";
import { api } from "../../../lib/api";
import { rememberUploads } from "../../../lib/promptAttach";

vi.mock("../../../lib/api", async (importOriginal) => {
  const actual = await importOriginal<Record<string, unknown>>();
  return {
    ...actual,
    api: {
      run: { files: vi.fn().mockResolvedValue({ items: ["src/", "src/a.ts"] }) },
      lanes: {
        pickPath: vi.fn().mockResolvedValue({ path: "/home/u/Videos/demo clip.mp4" }),
        sources: vi.fn().mockResolvedValue({
          sources: [
            { id: 1, kind: "file", path: "/data/ctx/1/brief.md", name: "brief.md", missing: false },
            { id: 2, kind: "file", path: "/data/ctx/1/gone.md", name: "gone.md", missing: true },
            { id: 3, kind: "folder", path: "/srv/docs", name: "docs", missing: false },
          ],
        }),
      },
      ccConfig: {
        skills: vi.fn().mockResolvedValue({ items: [] }),
        plugins: vi.fn().mockResolvedValue({ plugins: [] }),
      },
    },
  };
});

function renderMenu(laneId: number | null = 7) {
  const onAttach = vi.fn();
  render(
    <MemoryRouter>
      <ChatAddMenu
        cwd="/repo"
        laneId={laneId}
        skills={false}
        onUpload={() => {}}
        onInsert={() => {}}
        onAttach={onAttach}
      />
    </MemoryRouter>
  );
  fireEvent.click(screen.getByTestId("chat-add"));
  return onAttach;
}

beforeEach(() => {
  i18n.changeLanguage("en");
  localStorage.clear();
  vi.clearAllMocks();
});

afterEach(() => vi.unstubAllGlobals());

describe("ChatAddMenu — a file on this computer", () => {
  it("attaches the file the OS picker chose by its path, uploading nothing", async () => {
    const onAttach = renderMenu();
    fireEvent.click(screen.getByTestId("chat-add-pick-local"));
    expect(api.lanes.pickPath).toHaveBeenCalledWith("file", "/repo");
    await vi.waitFor(() =>
      expect(onAttach).toHaveBeenCalledWith([
        { path: '@"/home/u/Videos/demo clip.mp4"', name: "demo clip.mp4" },
      ])
    );
  });

  it("is not offered to a browser on another machine", () => {
    // Only a loopback address: a name that merely starts with "localhost" is not one.
    vi.stubGlobal("location", { ...window.location, hostname: "localhost.example.com" });
    renderMenu();
    expect(screen.queryByTestId("chat-add-pick-local")).toBeNull();
  });
});

describe("ChatAddMenu — existing resources", () => {
  it("searches the project, folders included, and attaches a pick as @path", async () => {
    const onAttach = renderMenu();
    fireEvent.mouseEnter(screen.getByTestId("chat-add-project"));
    fireEvent.click(await screen.findByTestId("chat-add-project-src/"));
    expect(api.run.files).toHaveBeenCalledWith("/repo", "", true);
    expect(onAttach).toHaveBeenCalledWith([{ path: "@src/", name: "src/" }]);
  });

  it("offers the workspace's context files that are still there, files only", async () => {
    const onAttach = renderMenu();
    fireEvent.mouseEnter(screen.getByTestId("chat-add-context"));
    fireEvent.click(await screen.findByTestId("chat-add-context-brief.md"));
    expect(api.lanes.sources).toHaveBeenCalledWith(7);
    expect(screen.queryByTestId("chat-add-context-gone.md")).toBeNull();
    expect(onAttach).toHaveBeenCalledWith([{ path: "@/data/ctx/1/brief.md", name: "brief.md" }]);
  });

  it("has no Context files without a workspace", () => {
    renderMenu(null);
    expect(screen.queryByTestId("chat-add-context")).toBeNull();
  });

  it("lists recent uploads newest first and attaches one as it was sent", () => {
    rememberUploads([{ path: "/tmp/d/old.pdf", name: "old.pdf", image: false }]);
    rememberUploads([{ path: "/tmp/d/0123456789abcdef.png", name: "shot.png", image: true }]);
    const onAttach = renderMenu();
    fireEvent.mouseEnter(screen.getByTestId("chat-add-recent"));
    const rows = screen.getAllByTestId(/^chat-add-recent-/);
    expect(rows.map((r) => r.textContent)).toEqual(["shot.png", "old.pdf"]);
    fireEvent.click(rows[0]!);
    expect(onAttach).toHaveBeenCalledWith([
      {
        path: "/tmp/d/0123456789abcdef.png",
        name: "shot.png",
        url: "/api/run/drop/0123456789abcdef.png",
      },
    ]);
  });
});