gridline/src/components/db-viewer/TableOverflowMenu.test.tsx

import { describe, it, expect, vi, beforeEach } from "vitest";
import { render, screen, fireEvent, waitFor } from "@testing-library/react";
import userEvent from "@testing-library/user-event";
import { TableOverflowMenu } from "./TableOverflowMenu";
import { useDbViewerStore } from "../../stores/dbViewerStore";
import { useUiStore } from "../../stores/uiStore";
import * as commands from "../../lib/commands";
import * as exportData from "../../lib/exportData";

describe("TableOverflowMenu", () => {
  beforeEach(() => {
    Object.defineProperty(navigator, "clipboard", {
      value: { writeText: vi.fn() },
      configurable: true,
      writable: true,
    });
    useDbViewerStore.getState().reset();
    useUiStore.setState({ activeConnectionId: "c1" });
    vi.resetAllMocks();
  });

  it("offers Create Index… and Create Constraint…", () => {
    render(<TableOverflowMenu schema="public" table="users" onOpenTab={() => "tab-1"} connectionId="c1" />);
    fireEvent.click(screen.getByLabelText(/table options/i));
    expect(screen.getByText("Create Index…")).toBeInTheDocument();
    expect(screen.getByText("Create Constraint…")).toBeInTheDocument();
  });

  it("opens objectForm create tabs for Create Index… and Create Constraint…", async () => {
    render(<TableOverflowMenu schema="public" table="users" onOpenTab={() => "tab-1"} connectionId="c1" />);
    fireEvent.click(screen.getByLabelText(/table options/i));
    fireEvent.click(screen.getByText("Create Index…"));
    const st = useDbViewerStore.getState();
    expect(st.tabs).toHaveLength(1);
    expect(st.tabs[0].tabType).toBe("objectForm");
    expect(st.tabs[0].form?.kind).toBe("index");
    expect(st.tabs[0].form?.mode).toBe("create");

    useDbViewerStore.getState().reset();
    render(<TableOverflowMenu schema="public" table="users" onOpenTab={() => "tab-1"} connectionId="c1" />);
    fireEvent.click(screen.getAllByLabelText(/table options/i)[1]);
    fireEvent.click(screen.getByText("Create Constraint…"));
    expect(useDbViewerStore.getState().tabs[0].tabType).toBe("objectForm");
    expect(useDbViewerStore.getState().tabs[0].form?.kind).toBe("constraint");
  });

  it("renders menu trigger button", () => {
    render(<TableOverflowMenu schema="public" table="users" onOpenTab={() => "tab-1"} />);
    expect(screen.getByLabelText(/table options/i)).toBeInTheDocument();
  });

  it("shows menu options on click", async () => {
    const user = userEvent.setup();
    render(<TableOverflowMenu schema="public" table="users" onOpenTab={() => "tab-1"} />);
    await user.click(screen.getByLabelText(/table options/i));
    expect(screen.getByText("Open in new tab")).toBeInTheDocument();
    expect(screen.getByText("Copy table schema")).toBeInTheDocument();
    expect(screen.getByText("Export data (CSV)")).toBeInTheDocument();
    expect(screen.getByText("Export data (Excel)")).toBeInTheDocument();
  });

  it("fires onOpenTab when menu item clicked", async () => {
    const user = userEvent.setup();
    const onOpenTab = vi.fn().mockReturnValue("tab-1");
    render(<TableOverflowMenu schema="public" table="users" onOpenTab={onOpenTab} />);
    await user.click(screen.getByLabelText(/table options/i));
    await user.click(screen.getByText("Open in new tab"));
    expect(onOpenTab).toHaveBeenCalledWith("public", "users", true);
  });

  it("Copy table schema calls getTableDdl and writes clipboard", async () => {
    vi.spyOn(commands, "getTableDdl").mockResolvedValue("CREATE TABLE t (id int)");
    const writeText = (navigator.clipboard as any).writeText as ReturnType<typeof vi.fn>;
    render(<TableOverflowMenu schema="public" table="t" onOpenTab={() => "tab-1"} />);
    fireEvent.click(screen.getByLabelText(/table options/i));
    fireEvent.click(screen.getByText(/copy table schema/i));
    await waitFor(() => expect(writeText).toHaveBeenCalledWith("CREATE TABLE t (id int)"));
  });

  it("Empty Table opens confirm then stages an empty_table change", async () => {
    render(<TableOverflowMenu schema="public" table="t" onOpenTab={() => "tab-1"} />);
    fireEvent.click(screen.getByLabelText(/table options/i));
    fireEvent.click(screen.getByText(/empty table/i));
    fireEvent.click(screen.getByRole("button", { name: /empty table/i }));
    await waitFor(() => {
      const q = useDbViewerStore.getState().changesQueue;
      expect(q[q.length - 1]).toEqual(expect.objectContaining({ type: "empty_table", schema: "public", table: "t" }));
    });
  });

  it("Delete Table opens confirm then stages a drop_table change", async () => {
    vi.spyOn(commands, "getObjectDependencies").mockResolvedValue([]);
    render(<TableOverflowMenu schema="public" table="t" onOpenTab={() => "tab-1"} />);
    fireEvent.click(screen.getByLabelText(/table options/i));
    fireEvent.click(screen.getByText(/delete table/i));
    await waitFor(() => expect(screen.queryByText(/open in new tab/i)).not.toBeInTheDocument());
    fireEvent.click(screen.getByRole("button", { name: /delete table/i }));
    await waitFor(() => {
      const q = useDbViewerStore.getState().changesQueue;
      expect(q[q.length - 1]).toEqual(expect.objectContaining({ type: "drop_table", schema: "public", table: "t" }));
    });
  });

  it("Delete Table fetches dependencies and shows DependencyDialog before confirming", async () => {
    vi.spyOn(commands, "getObjectDependencies").mockResolvedValue([{ deptype: "n", class: "pg_class", name: "v_orders" }]);
    render(<TableOverflowMenu schema="public" table="orders" connectionId="c1" onOpenTab={() => "tab-1"} />);
    fireEvent.click(screen.getByLabelText(/table options/i));
    fireEvent.click(screen.getByText(/delete table/i));
    await waitFor(() => expect(commands.getObjectDependencies).toHaveBeenCalledWith("c1", "public", "table", "orders"));
    await waitFor(() => expect(screen.getByText("v_orders")).toBeInTheDocument());
  });

  it("Export data calls exportData when rows and columns are provided", async () => {
    const spy = vi.spyOn(exportData, "exportData");
    const columns = [{ name: "id", data_type: "integer", is_nullable: false, is_pk: true, is_fk: false, fk_ref: null, default_value: null, editable: true, is_generated: false }];
    render(<TableOverflowMenu schema="public" table="t" onOpenTab={() => "tab-1"} columns={columns} rows={[[1]]} />);
    fireEvent.click(screen.getByLabelText(/table options/i));
    fireEvent.click(screen.getByText(/export data \(csv\)/i));
    await waitFor(() => expect(spy).toHaveBeenCalledWith([[1]], columns, "csv", "public.t"));
  });

  it("Excel export calls exportData and notifies", async () => {
    const spy = vi.spyOn(exportData, "exportData").mockImplementation(() => {});
    const { useNotificationStore } = await import("../../stores/notificationStore");
    useNotificationStore.getState().notifications.length = 0;
    const columns = [{ name: "id", data_type: "integer", is_nullable: false, is_pk: true, is_fk: false, fk_ref: null, default_value: null, editable: true, is_generated: false }];
    render(<TableOverflowMenu schema="public" table="t" onOpenTab={() => "tab-1"} columns={columns} rows={[[1]]} />);
    fireEvent.click(screen.getByLabelText(/table options/i));
    fireEvent.click(screen.getByText(/export data \(excel\)/i));
    await waitFor(() => expect(spy).toHaveBeenCalledWith([[1]], columns, "xlsx", "public.t"));
    const st = useNotificationStore.getState();
    expect(st.notifications.some((n) => n.message.toLowerCase().includes("exported"))).toBe(true);
  });

  it("tree kebab export fetches table data when rows are absent", async () => {
    const spy = vi.spyOn(exportData, "exportData").mockImplementation(() => {});
    const getSpy = vi.spyOn(commands, "getTableData").mockResolvedValue({
      columns: [{ name: "id", data_type: "integer", is_nullable: false, is_pk: true, is_fk: false, fk_ref: null, default_value: null, editable: true, is_generated: false }],
      rows: [[42]],
      total_rows: 1,
      page: 1,
      page_size: 1000,
    });
    const { useNotificationStore } = await import("../../stores/notificationStore");
    useNotificationStore.getState().notifications.length = 0;
    render(<TableOverflowMenu schema="public" table="t" onOpenTab={() => "tab-1"} connectionId="c1" />);
    fireEvent.click(screen.getByLabelText(/table options/i));
    fireEvent.click(screen.getByText(/export data \(excel\)/i));
    await waitFor(() =>
      expect(getSpy).toHaveBeenCalledWith("c1", "public", "t", 1, 1000),
    );
    await waitFor(() => expect(spy).toHaveBeenCalled());
    const st = useNotificationStore.getState();
    expect(st.notifications.some((n) => n.message.includes("Exported"))).toBe(true);
  });

  it("maintenance items are gated by capability and run via confirm", async () => {
    vi.spyOn(commands, "runMaintenance").mockResolvedValue({ duration_ms: 3, message: "VACUUM completed" });
    render(<TableOverflowMenu schema="public" table="users" onOpenTab={() => ""} connectionId="c1" />);
    fireEvent.click(screen.getByLabelText("Table options"));
    fireEvent.click(screen.getByText("VACUUM"));
    fireEvent.click(screen.getByRole("button", { name: /vacuum/i }));
    await screen.findByText(/completed/i);
    expect(commands.runMaintenance).toHaveBeenCalledWith("c1", "public", "users", "vacuum");
  });

  it("Edit Table opens a table form tab", () => {
    const openFormTab = vi.spyOn(useDbViewerStore.getState(), "openFormTab");
    render(<TableOverflowMenu schema="public" table="users" onOpenTab={() => ""} connectionId="c1" columns={[{ name: "id", data_type: "int", is_nullable: false, is_pk: true, is_fk: false, fk_ref: null, default_value: null, editable: true, is_generated: false }]} />);
    fireEvent.click(screen.getByLabelText("Table options"));
    fireEvent.click(screen.getByText("Edit Table…"));
    expect(openFormTab).toHaveBeenCalledWith(expect.objectContaining({ kind: "table", mode: "edit" }));
  });
});