scripts/testing/team-computers-fixture.test.ts

Rama main · réplica de lectura del repositorio original.

import { once } from "node:events";
import { existsSync, readFileSync } from "node:fs";
import { createServer } from "node:http";
import { join } from "node:path";
import { expect, it } from "vitest";
import { launchVerificationServer, type VerificationServer } from "../control-omb.ts";

it.each([
  "https://127.0.0.1:12345", "http://localhost:12345", "http://192.0.2.1:12345",
  "http://127.0.0.1:12345/api", "http://127.0.0.1:12345?token=secret", "http://127.0.0.1:12345#fragment",
  "http://user:password@127.0.0.1:12345", "http://127.0.0.1:0", "http://127.0.0.1:65536",
])("rejects an unsafe Boat fixture endpoint before launching: %s", async (endpoint) => {
  await expect(launchVerificationServer({}, undefined, undefined, undefined, undefined, undefined, [], endpoint))
    .rejects.toThrow(/Boat verification requires/);
});

it.each([false, true])("adds a Boat account only for an opted-in Boat fixture, and never a Boat-run engine: %s", async (withBoat) => {
  const calls: string[] = [];
  const provider = createServer((request, response) => {
    calls.push(`${request.method} ${request.url}`);
    response.writeHead(200, { "content-type": "application/json" });
    response.end(JSON.stringify({ ok: true, sandboxes: [] }));
  });
  let fixture: VerificationServer | undefined;
  try {
    provider.listen(0, "127.0.0.1");
    await once(provider, "listening");
    const address = provider.address();
    if (!address || typeof address === "string") throw new Error("Boat fixture has no loopback port");
    const endpoint = `http://127.0.0.1:${address.port}`;
    fixture = await launchVerificationServer({}, undefined, undefined, undefined, undefined, undefined, [], withBoat ? endpoint : undefined);
    const config = JSON.parse(readFileSync(join(fixture.info.dataDir, "config.json"), "utf8"));
    const response = await fetch(`${fixture.info.url}/api/instances`);
    expect(response.ok).toBe(true);
    const { instances } = await response.json() as { instances: Array<{ instanceId: string; driverKind: string }> };
    // Bots use a cloud computer as a tool on their own engine; no engine
    // runs its turns on Boat.
    expect(instances.filter(instance => instance.driverKind === "boxAgent" || instance.instanceId === "computer")).toEqual([]);
    expect(config.instances).not.toHaveProperty("computer");
    if (withBoat) {
      expect(config.box).toEqual({ token: "box_verification_fixture" });
    } else {
      expect(config).not.toHaveProperty("box");
      expect(calls).toEqual([]);
    }
    // Merely advertising the opted-in backend must never provision a boat.
    expect(calls.every(call => call.startsWith("GET "))).toBe(true);
  } finally {
    await fixture?.close();
    provider.closeAllConnections();
    if (provider.listening) await new Promise<void>((resolve, reject) => provider.close(error => error ? reject(error) : resolve()));
  }
  expect(existsSync(fixture!.info.dataDir)).toBe(false);
}, 30_000);