PaperClipAI/server/src/__tests__/llms-routes.test.ts

import express from "express";
import request from "supertest";
import { beforeEach, describe, expect, it, vi } from "vitest";

const mockAgentService = vi.hoisted(() => ({
  getById: vi.fn(),
}));

const mockListServerAdapters = vi.hoisted(() => vi.fn());

vi.mock("../services/agents.js", () => ({
  agentService: () => mockAgentService,
}));

vi.mock("../adapters/index.js", () => ({
  listServerAdapters: mockListServerAdapters,
}));

function registerModuleMocks() {
  vi.doMock("../services/agents.js", () => ({
    agentService: () => mockAgentService,
  }));

  vi.doMock("../adapters/index.js", () => ({
    listServerAdapters: mockListServerAdapters,
  }));
}

async function createApp(actor: Record<string, unknown>) {
  const [{ llmRoutes }, { errorHandler }] = await Promise.all([
    vi.importActual<typeof import("../routes/llms.js")>("../routes/llms.js"),
    vi.importActual<typeof import("../middleware/index.js")>("../middleware/index.js"),
  ]);
  const app = express();
  app.use(express.json());
  app.use((req, _res, next) => {
    (req as any).actor = actor;
    next();
  });
  app.use("/api", llmRoutes({} as never));
  app.use(errorHandler);
  return app;
}

describe("llm routes", () => {
  beforeEach(() => {
    vi.resetModules();
    vi.doUnmock("../routes/llms.js");
    vi.doUnmock("../middleware/index.js");
    registerModuleMocks();
    vi.clearAllMocks();
    mockListServerAdapters.mockReturnValue([
      { type: "codex_local", agentConfigurationDoc: "# codex_local agent configuration" },
    ]);
  });

  it("documents timer heartbeats as opt-in for new hires", async () => {
    const app = await createApp({
      type: "board",
      userId: "board-user",
      companyIds: ["company-1"],
      source: "local_implicit",
      isInstanceAdmin: true,
    });

    const res = await request(app).get("/api/llms/agent-configuration.txt");

    expect(res.status).toBe(200);
    expect(res.text).toContain("Use the paperclip-create-agent skill for end-to-end hiring");
    expect(res.text).toContain("desiredSkills");
    expect(res.text).toContain("sourceIssueId/sourceIssueIds");
    expect(res.text).toContain("Timer heartbeats are opt-in for new hires.");
    expect(res.text).toContain("Leave runtimeConfig.heartbeat.enabled false");
  });

  it("serves static Hermes Gateway configuration docs before the plugin is installed", async () => {
    const app = await createApp({
      type: "board",
      userId: "board-user",
      companyIds: ["company-1"],
      source: "local_implicit",
      isInstanceAdmin: true,
    });

    const indexRes = await request(app).get("/api/llms/agent-configuration.txt");
    expect(indexRes.status).toBe(200);
    expect(indexRes.text).toContain(
      "- hermes_gateway: /llms/agent-configuration/hermes_gateway.txt",
    );

    const res = await request(app).get("/api/llms/agent-configuration/hermes_gateway.txt");

    expect(res.status).toBe(200);
    expect(res.text).toContain("Adapter: hermes_gateway");
    expect(res.text).toContain('adapterType": "hermes_gateway"');
    expect(res.text).toContain("API_SERVER_ENABLED=true");
    expect(res.text).toContain("API_SERVER_KEY");
    expect(res.text).toContain("hermes gateway run --replace --accept-hooks");
    expect(res.text).toContain("Default Hermes API server port: 8642");
    expect(res.text).toContain("agentDefaultsPayload.apiBaseUrl");
    expect(res.text).toContain("agentDefaultsPayload.paperclipApiUrl");
    expect(res.text).toContain("hermes_local");
    expect(res.text).toContain("Hermes-originated Paperclip API usage");
    expect(res.text).toContain("http://127.0.0.1:8642");
    expect(res.text).toContain("http://192.168.1.25:8642");
    expect(res.text).toContain("tailnet-name.ts.net:8642");
    expect(res.text).toContain("http://host.docker.internal:8642");
    expect(res.text).toContain("https://hermes-gateway.example");
  });
});