AgentHub/client/src/pages/__tests__/AiRouterConfig.test.tsx

/**
 * @file Tests for the AI Router settings tab — listing profiles, creating one,
 * and the two rules that keep a key safe: the form never shows one, and an
 * edit that leaves the field blank must not send an empty key that would wipe
 * the stored one.
 * @author Nguyễn Ngọc Trí Vĩ <vinnt@smartgift.vn>
 */

import { describe, it, expect, vi, beforeEach } from "vitest";
import { render, screen, waitFor, within } from "@testing-library/react";
import userEvent from "@testing-library/user-event";
import { AiRouterConfig } from "../AiRouterConfig";
import { api } from "../../lib/api";
import { settleConfirm, subscribeConfirm, type PendingConfirm } from "../../lib/confirm";

vi.mock("../../lib/api", () => ({
  api: {
    aiRouters: {
      list: vi.fn(),
      create: vi.fn(),
      update: vi.fn(),
      remove: vi.fn(),
      test: vi.fn(),
    },
  },
}));

vi.mock("react-i18next", () => ({
  useTranslation: () => ({
    t: (key: string, opts?: Record<string, unknown>) =>
      opts && typeof opts.label === "string" ? `${key}:${opts.label}` : key,
  }),
}));

const native = {
  name: "native",
  label: "Native (Claude / Codex)",
  builtin: true,
  baseUrl: "",
  hasKey: false,
  model: "claude-sonnet-5",
  models: {
    opus: "claude-opus-5",
    sonnet: "claude-sonnet-5",
    haiku: "claude-haiku-4-5-20251001",
    smallFast: "claude-haiku-4-5-20251001",
    subagent: "claude-haiku-4-5-20251001",
  },
};

const nineRouter = {
  ...native,
  name: "9router",
  label: "9Router",
  builtin: false,
  baseUrl: "http://localhost:20128/v1",
  hasKey: true,
  model: "Highest",
  models: {
    opus: "Highest",
    sonnet: "Highest",
    haiku: "Highest",
    smallFast: "Highest",
    subagent: "Highest",
  },
};

beforeEach(() => {
  vi.mocked(api.aiRouters.list)
    .mockReset()
    .mockResolvedValue({ items: [native] });
  vi.mocked(api.aiRouters.create).mockReset().mockResolvedValue(nineRouter);
  vi.mocked(api.aiRouters.update).mockReset().mockResolvedValue(nineRouter);
  vi.mocked(api.aiRouters.remove).mockReset().mockResolvedValue({ removed: "9router" });
  vi.mocked(api.aiRouters.test)
    .mockReset()
    .mockResolvedValue({ ok: true, status: 200, message: "Highest" });
});

describe("AiRouterConfig", () => {
  it("lists the built-in native profile with no edit or delete control", async () => {
    render(<AiRouterConfig />);
    expect(await screen.findByText("Native (Claude / Codex)")).toBeInTheDocument();
    expect(screen.queryByLabelText(/aiRouter\.edit Native/)).toBeNull();
    expect(screen.queryByLabelText(/aiRouter\.remove Native/)).toBeNull();
  });

  it("saves a new profile with the typed endpoint, key and model", async () => {
    render(<AiRouterConfig />);
    const user = userEvent.setup();

    await user.click(await screen.findByRole("button", { name: /aiRouter\.add/ }));
    await user.type(screen.getByLabelText("aiRouter.fieldLabel"), "9Router");
    await user.type(screen.getByLabelText("aiRouter.fieldBaseUrl"), "http://localhost:20128/v1");
    await user.type(screen.getByLabelText("aiRouter.fieldApiKey"), "sk-typed");
    await user.type(screen.getByLabelText("aiRouter.fieldModel"), "Highest");
    await user.click(screen.getByRole("button", { name: "aiRouter.save" }));

    await waitFor(() =>
      expect(api.aiRouters.create).toHaveBeenCalledWith({
        label: "9Router",
        baseUrl: "http://localhost:20128/v1",
        model: "Highest",
        // Every tier equals the headline model, so none is sent: the server
        // fills them, which is what makes one field enough.
        models: {},
        behavesAs: "",
        apiKey: "sk-typed",
      })
    );
  });

  it("sends the typed behavesAs, which maps the router's model names in Claude Code (#56)", async () => {
    render(<AiRouterConfig />);
    const user = userEvent.setup();

    await user.click(await screen.findByRole("button", { name: /aiRouter\.add/ }));
    await user.type(screen.getByLabelText("aiRouter.fieldLabel"), "9Router");
    await user.type(screen.getByLabelText("aiRouter.fieldBaseUrl"), "http://localhost:20128/v1");
    await user.type(screen.getByLabelText("aiRouter.fieldModel"), "agent");
    await user.type(screen.getByLabelText("aiRouter.fieldBehavesAs"), "claude-opus-5-5");
    await user.click(screen.getByRole("button", { name: "aiRouter.save" }));

    await waitFor(() =>
      expect(api.aiRouters.create).toHaveBeenCalledWith(
        expect.objectContaining({ model: "agent", behavesAs: "claude-opus-5-5" })
      )
    );
  });

  it("sends only the tiers that differ from the headline model", async () => {
    render(<AiRouterConfig />);
    const user = userEvent.setup();

    await user.click(await screen.findByRole("button", { name: /aiRouter\.add/ }));
    await user.type(screen.getByLabelText("aiRouter.fieldLabel"), "9Router");
    await user.type(screen.getByLabelText("aiRouter.fieldBaseUrl"), "http://localhost:20128/v1");
    await user.type(screen.getByLabelText("aiRouter.fieldModel"), "Highest");
    await user.type(screen.getByLabelText("aiRouter.tier.opus"), "PremiumCombo");
    await user.click(screen.getByRole("button", { name: "aiRouter.save" }));

    await waitFor(() =>
      expect(api.aiRouters.create).toHaveBeenCalledWith(
        expect.objectContaining({ models: { opus: "PremiumCombo" } })
      )
    );
  });

  it("never puts a stored key in the edit form, and omits it when left blank", async () => {
    // The API cannot return a key, so an edit that sends `apiKey: ""` would
    // silently erase the one on disk.
    vi.mocked(api.aiRouters.list).mockResolvedValue({ items: [native, nineRouter] });
    render(<AiRouterConfig />);
    const user = userEvent.setup();

    await user.click(await screen.findByLabelText("aiRouter.edit 9Router"));
    expect(screen.getByLabelText("aiRouter.fieldApiKey")).toHaveValue("");

    await user.clear(screen.getByLabelText("aiRouter.fieldLabel"));
    await user.type(screen.getByLabelText("aiRouter.fieldLabel"), "9Router nhà");
    await user.click(screen.getByRole("button", { name: "aiRouter.save" }));

    await waitFor(() => expect(api.aiRouters.update).toHaveBeenCalled());
    const call = vi.mocked(api.aiRouters.update).mock.calls[0];
    expect(call?.[1]).not.toHaveProperty("apiKey");
  });

  it("surfaces a server error instead of closing the form", async () => {
    vi.mocked(api.aiRouters.create).mockRejectedValue(
      new Error("baseUrl must be http:// or https://")
    );
    render(<AiRouterConfig />);
    const user = userEvent.setup();

    await user.click(await screen.findByRole("button", { name: /aiRouter\.add/ }));
    await user.type(screen.getByLabelText("aiRouter.fieldLabel"), "Bad");
    await user.type(screen.getByLabelText("aiRouter.fieldBaseUrl"), "file:///etc/passwd");
    await user.type(screen.getByLabelText("aiRouter.fieldModel"), "x");
    await user.click(screen.getByRole("button", { name: "aiRouter.save" }));

    // Inside the dialog, not behind its scrim.
    const dialog = screen.getByRole("dialog");
    expect(await within(dialog).findByText(/must be http/)).toBeInTheDocument();
  });

  it("closes the form on Escape", async () => {
    render(<AiRouterConfig />);
    const user = userEvent.setup();
    await user.click(await screen.findByRole("button", { name: /aiRouter\.add/ }));
    expect(screen.getByRole("dialog")).toBeInTheDocument();
    await user.keyboard("{Escape}");
    await waitFor(() => expect(screen.queryByRole("dialog")).toBeNull());
  });
});

describe("AiRouterConfig — testing an endpoint before saving it", () => {
  it("sends the typed endpoint, key and model, and reports success", async () => {
    render(<AiRouterConfig />);
    const user = userEvent.setup();

    await user.click(await screen.findByRole("button", { name: /aiRouter\.add/ }));
    await user.type(screen.getByLabelText("aiRouter.fieldBaseUrl"), "http://localhost:20128/v1");
    await user.type(screen.getByLabelText("aiRouter.fieldApiKey"), "sk-typed");
    await user.type(screen.getByLabelText("aiRouter.fieldModel"), "Highest");
    await user.click(screen.getByTestId("router-test"));

    await waitFor(() =>
      expect(api.aiRouters.test).toHaveBeenCalledWith({
        baseUrl: "http://localhost:20128/v1",
        model: "Highest",
        apiKey: "sk-typed",
      })
    );
    expect(await screen.findByTestId("router-test-result")).toHaveTextContent("aiRouter.testOk");
  });

  it("names the profile being edited so the server can reuse its stored key", async () => {
    vi.mocked(api.aiRouters.list).mockResolvedValue({ items: [native, nineRouter] });
    render(<AiRouterConfig />);
    const user = userEvent.setup();

    await user.click(await screen.findByLabelText(/aiRouter\.edit 9Router/));
    await user.click(screen.getByTestId("router-test"));

    await waitFor(() =>
      expect(api.aiRouters.test).toHaveBeenCalledWith({
        baseUrl: "http://localhost:20128/v1",
        model: "Highest",
        name: "9router",
      })
    );
  });

  it("shows the router's own refusal rather than claiming it works", async () => {
    vi.mocked(api.aiRouters.test).mockResolvedValue({
      ok: false,
      status: 401,
      message: "invalid api key",
    });
    render(<AiRouterConfig />);
    const user = userEvent.setup();

    await user.click(await screen.findByRole("button", { name: /aiRouter\.add/ }));
    await user.type(screen.getByLabelText("aiRouter.fieldBaseUrl"), "http://localhost:20128/v1");
    await user.type(screen.getByLabelText("aiRouter.fieldModel"), "Highest");
    await user.click(screen.getByTestId("router-test"));

    expect(await screen.findByTestId("router-test-result")).toHaveTextContent(
      "aiRouter.testFailed"
    );
  });

  it("drops a stale verdict as soon as a field changes", async () => {
    render(<AiRouterConfig />);
    const user = userEvent.setup();

    await user.click(await screen.findByRole("button", { name: /aiRouter\.add/ }));
    await user.type(screen.getByLabelText("aiRouter.fieldBaseUrl"), "http://localhost:20128/v1");
    await user.type(screen.getByLabelText("aiRouter.fieldModel"), "Highest");
    await user.click(screen.getByTestId("router-test"));
    expect(await screen.findByTestId("router-test-result")).toBeInTheDocument();

    await user.type(screen.getByLabelText("aiRouter.fieldBaseUrl"), "2");
    await waitFor(() => expect(screen.queryByTestId("router-test-result")).toBeNull());
  });

  it("opens a profile's Edit and Remove from a right-click; the built-in row has no menu", async () => {
    vi.mocked(api.aiRouters.list).mockResolvedValue({ items: [native, nineRouter] });
    let pending: PendingConfirm | null = null;
    const off = subscribeConfirm((p) => (pending = p));
    const user = userEvent.setup();
    render(<AiRouterConfig />);
    const row = (await screen.findByText("9Router")).closest("tr")!;

    await user.pointer({ keys: "[MouseRight]", target: screen.getByText(native.label) });
    expect(screen.queryByRole("menu")).toBeNull();

    await user.pointer({ keys: "[MouseRight]", target: row });
    await user.click(screen.getByRole("menuitem", { name: "aiRouter.remove" }));
    await waitFor(() => expect(pending?.title).toBe("aiRouter.removeConfirm:9Router"));
    settleConfirm(false);
    await waitFor(() => expect(pending).toBeNull());
    expect(api.aiRouters.remove).not.toHaveBeenCalled();

    await user.pointer({ keys: "[MouseRight]", target: row });
    await user.click(screen.getByRole("menuitem", { name: "aiRouter.edit" }));
    expect(screen.getByRole("dialog")).toBeInTheDocument();
    off();
  });
});