openidea/src/features/search/__tests__/fileStructure.test.tsx

import { afterEach, describe, expect, it, vi } from "vitest";
import { act, fireEvent, render, screen } from "@testing-library/react";
import type { OutlineItem } from "@/platform/ipc";

vi.mock("../structure/store", async (orig) => ({
  ...(await orig<typeof import("../structure/store")>()),
  useLiveOutline: () => undefined,
}));

const { FileStructureOverlay, useFileStructurePopup } = await import("../structure/StructureView");
const { useOutline, useStructureOptions } = await import("../structure/store");

const item = (name: string, kind: OutlineItem["kind"], line: number, children: OutlineItem[] = []): OutlineItem => ({
  name,
  detail: null,
  kind,
  range: { start: { line, col: 0 }, end: { line: line + 1, col: 1 } },
  selection_range: { start: { line, col: 4 }, end: { line, col: 4 + name.length } },
  children,
});

function open() {
  useOutline.setState({
    path: "/p/Order.kt",
    doc: 1,
    loading: false,
    error: null,
    items: [item("OrderController", "class", 0, [item("list", "method", 2), item("create", "method", 8)]), item("helper", "function", 22)],
  });
  useStructureOptions.setState({ narrowDown: true });
  render(<FileStructureOverlay />);
  act(() => useFileStructurePopup.setState({ open: true }));
  return screen.getByLabelText("Filter symbols");
}

const rows = () => screen.getAllByRole("treeitem").map((r) => r.textContent);

afterEach(() => act(() => useFileStructurePopup.setState({ open: false })));

describe("File Structure popup", () => {
  it("shows the outline as an expanded tree; ←/→ collapse and expand", () => {
    const input = open();
    expect(rows()).toEqual(["OrderController", "list", "create", "helper"]);
    // The first symbol is preselected; ← collapses it, → expands it again.
    fireEvent.keyDown(input, { key: "ArrowLeft" });
    expect(rows()).toEqual(["OrderController", "helper"]);
    fireEvent.keyDown(input, { key: "ArrowRight" });
    expect(rows()).toEqual(["OrderController", "list", "create", "helper"]);
  });

  it("narrows down on typing keeping ancestors, or only highlights when that is off", () => {
    const input = open();
    fireEvent.change(input, { target: { value: "creat" } });
    expect(rows()).toEqual(["OrderController", "create"]);
    fireEvent.click(screen.getByLabelText("Narrow down on typing"));
    expect(useStructureOptions.getState().narrowDown).toBe(false);
    expect(rows()).toEqual(["OrderController", "list", "create", "helper"]);
  });

  it("Esc clears the filter first, then closes", () => {
    const input = open();
    fireEvent.change(input, { target: { value: "help" } });
    fireEvent.keyDown(input, { key: "Escape" });
    expect((input as HTMLInputElement).value).toBe("");
    expect(useFileStructurePopup.getState().open).toBe(true);
    fireEvent.keyDown(input, { key: "Escape" });
    expect(useFileStructurePopup.getState().open).toBe(false);
  });

  it("offers Show inherited members only when the server knows type hierarchies", () => {
    open();
    expect(screen.queryByLabelText("Show inherited members")).toBeNull();
    const inherited = [{ item: item("toString", "method", 40), path: "/lib/Object.kt", owner: "Any" }];
    act(() => useOutline.setState({ inheritedSupported: true, inherited: new Map([["0:4", inherited]]) }));
    expect(rows()).toEqual(["OrderController", "list", "create", "helper"]);
    fireEvent.click(screen.getByLabelText("Show inherited members"));
    expect(useStructureOptions.getState().showInherited).toBe(true);
    expect(rows()).toEqual(["OrderController", "list", "create", "toString↑Any", "helper"]);
    act(() => {
      useStructureOptions.setState({ showInherited: false });
      useOutline.setState({ inheritedSupported: false, inherited: null });
    });
  });
});