boss/tests/master-agent-evolution-routes.test.ts

import test from "node:test";
import assert from "node:assert/strict";
import os from "node:os";
import path from "node:path";
import { mkdir, mkdtemp, rm } from "node:fs/promises";
import { NextRequest } from "next/server";

let runtimeRoot = "";
let createAuthSession: (typeof import("../src/lib/boss-data"))["createAuthSession"];
let recordMasterAgentEvolutionSignal: (typeof import("../src/lib/master-agent-evolution"))["recordMasterAgentEvolutionSignal"];
let GETEvolutionRoute: (typeof import("../src/app/api/v1/master-agent/evolution/route"))["GET"];
let POSTEvolutionConfigRoute: (typeof import("../src/app/api/v1/master-agent/evolution/config/route"))["POST"];
let POSTApproveRoute: (typeof import("../src/app/api/v1/master-agent/evolution/proposals/[proposalId]/approve/route"))["POST"];
let POSTRejectRoute: (typeof import("../src/app/api/v1/master-agent/evolution/proposals/[proposalId]/reject/route"))["POST"];
let AUTH_SESSION_COOKIE = "";

async function setup() {
  if (runtimeRoot) {
    return;
  }
  runtimeRoot = await mkdtemp(path.join(os.tmpdir(), "boss-master-agent-evolution-routes-"));
  process.env.BOSS_RUNTIME_ROOT = runtimeRoot;
  process.env.BOSS_STATE_FILE = path.join(runtimeRoot, "boss-state.json");
  const [data, auth, evolution, evolutionRoute, configRoute, approveRoute, rejectRoute] = await Promise.all([
    import("../src/lib/boss-data.ts"),
    import("../src/lib/boss-auth.ts"),
    import("../src/lib/master-agent-evolution.ts"),
    import("../src/app/api/v1/master-agent/evolution/route.ts"),
    import("../src/app/api/v1/master-agent/evolution/config/route.ts"),
    import("../src/app/api/v1/master-agent/evolution/proposals/[proposalId]/approve/route.ts"),
    import("../src/app/api/v1/master-agent/evolution/proposals/[proposalId]/reject/route.ts"),
  ]);
  createAuthSession = data.createAuthSession;
  recordMasterAgentEvolutionSignal = evolution.recordMasterAgentEvolutionSignal;
  GETEvolutionRoute = evolutionRoute.GET;
  POSTEvolutionConfigRoute = configRoute.POST;
  POSTApproveRoute = approveRoute.POST;
  POSTRejectRoute = rejectRoute.POST;
  AUTH_SESSION_COOKIE = auth.AUTH_SESSION_COOKIE;
}

async function createAdminRequest(url: string, body?: unknown) {
  const session = await createAuthSession({
    account: "17600003315",
    role: "highest_admin",
    displayName: "Boss 超级管理员",
    loginMethod: "password",
  });
  return new NextRequest(url, {
    method: body === undefined ? "GET" : "POST",
    headers: {
      cookie: `${AUTH_SESSION_COOKIE}=${session.sessionToken}`,
      ...(body === undefined ? {} : { "content-type": "application/json" }),
    },
    body: body === undefined ? undefined : JSON.stringify(body),
  });
}

test.after(async () => {
  if (runtimeRoot) {
    await rm(runtimeRoot, { recursive: true, force: true });
  }
});

test.beforeEach(async () => {
  await setup();
  await rm(runtimeRoot, { recursive: true, force: true });
  await mkdir(runtimeRoot, { recursive: true });
});

test("GET /api/v1/master-agent/evolution returns config proposals and rules", async () => {
  const response = await GETEvolutionRoute(
    await createAdminRequest("http://127.0.0.1:3000/api/v1/master-agent/evolution"),
  );
  assert.equal(response.status, 200);
  const payload = await response.json() as { ok: boolean; proposals: unknown[]; rules: unknown[] };
  assert.equal(payload.ok, true);
  assert.ok(Array.isArray(payload.proposals));
  assert.ok(Array.isArray(payload.rules));
});

test("POST /api/v1/master-agent/evolution/config switches mode", async () => {
  const response = await POSTEvolutionConfigRoute(
    await createAdminRequest("http://127.0.0.1:3000/api/v1/master-agent/evolution/config", { mode: "autonomous" }),
  );
  assert.equal(response.status, 200);
  const payload = await response.json() as { ok: boolean; config?: { mode?: string } };
  assert.equal(payload.ok, true);
  assert.equal(payload.config?.mode, "autonomous");
});

test("approve and reject evolution proposals update proposal status", async () => {
  const created = await recordMasterAgentEvolutionSignal({
    kind: "repeated_question",
    account: "17600003315",
    requestText: "当前主节点在线吗",
  });
  const proposalId = created.proposal?.proposalId ?? "";
  assert.ok(proposalId);

  const approveResponse = await POSTApproveRoute(
    await createAdminRequest(`http://127.0.0.1:3000/api/v1/master-agent/evolution/proposals/${proposalId}/approve`, {}),
    { params: Promise.resolve({ proposalId }) },
  );
  assert.equal(approveResponse.status, 200);
  const approvePayload = await approveResponse.json() as { ok: boolean; proposal?: { status?: string } };
  assert.equal(approvePayload.ok, true);
  assert.equal(approvePayload.proposal?.status, "approved");

  const created2 = await recordMasterAgentEvolutionSignal({
    kind: "backend_fallback",
    account: "17600003315",
    requestText: "切到稳定后端",
  });
  const proposalId2 = created2.proposal?.proposalId ?? "";
  assert.ok(proposalId2);
  const rejectResponse = await POSTRejectRoute(
    await createAdminRequest(`http://127.0.0.1:3000/api/v1/master-agent/evolution/proposals/${proposalId2}/reject`, {}),
    { params: Promise.resolve({ proposalId: proposalId2 }) },
  );
  assert.equal(rejectResponse.status, 200);
  const rejectPayload = await rejectResponse.json() as { ok: boolean; proposal?: { status?: string } };
  assert.equal(rejectPayload.ok, true);
  assert.equal(rejectPayload.proposal?.status, "rejected");
});