reverse-engineering-anything/src/application/environment/environmentReplayCapacity.test.ts

import { expect, it } from "vitest";
import { environmentServeSourceWithBody } from "../../../tests/fixtures/environmentServe.js";
import {
  replayEnvironmentTool,
  selectReplayExchange,
} from "./environmentReplay.js";

it("matches every repeated query occurrence and a distant path coordinate", () => {
  const source = environmentServeSourceWithBody("{}", "application/json");
  const original = source.manifest.tools.find(
    (tool) => tool.name === "get_users_by_id",
  )!;
  const ordinal = original.record_ordinals[0]!;
  const exchange = source.exchanges.get(ordinal)!;
  const tool = {
    ...original,
    params: [{ ...original.params[0]!, index: 20000 }],
    query_parameter_names: ["q"],
  };
  const retained = {
    ...exchange,
    path: `/${"literal/".repeat(20000)}wanted`,
    query: `?${"q=miss&".repeat(20000)}q=hit`,
  };
  const exchanges = new Map([[ordinal, retained]]);
  const selected = selectReplayExchange(tool, exchanges, {
    id: "wanted",
    q: [...Array.from({ length: 20000 }, () => "miss"), "hit"],
  });
  expect(selected?.exchange).toBe(retained);
  expect(selected?.exact).toBe(true);
  expect(
    selectReplayExchange(tool, exchanges, { id: "wanted", q: "hit" })?.exact,
  ).toBe(false);
});

it("bounds content-type display without changing body interpretation or retained source", () => {
  const contentType = `application/json;${"x".repeat(100000)}`;
  const source = environmentServeSourceWithBody('{"value":1}', contentType);
  const tool = source.manifest.tools.find(
    (tool) => tool.name === "get_users_by_id",
  )!;
  const result = replayEnvironmentTool(source, tool, { id: "2" });
  expect(result.ok).toBe(true);
  if (!result.ok) throw result.error;
  expect(result.value.body).toEqual({ value: 1 });
  expect(
    Buffer.byteLength(result.value.content_type ?? ""),
  ).toBeLessThanOrEqual(1024);
  expect(result.value.replay.content_type_truncated).toBe(true);
  expect(source.exchanges.get(tool.record_ordinals[0]!)?.contentType).toBe(
    contentType,
  );
});

it("keeps observed query order distinct from calls supplying only selected fields", () => {
  const source = environmentServeSourceWithBody("{}", "application/json");
  const original = source.manifest.tools.find(
    (tool) => tool.name === "get_users_by_id",
  )!;
  const ordinal = original.record_ordinals[0]!;
  const tool = { ...original, query_parameter_names: ["ignored", "q"] };
  const retained = {
    ...source.exchanges.get(ordinal)!,
    query: `?${"ignored=%FF+value&".repeat(20000)}%71=hi%74`,
  };
  const exchanges = new Map([[ordinal, retained]]);
  const args = { id: "2", q: "hit" };
  expect(selectReplayExchange(tool, exchanges, args)?.exact).toBe(true);
  expect(selectReplayExchange(tool, exchanges, args, "?q=hit")?.exact).toBe(
    false,
  );
  expect(
    selectReplayExchange(tool, exchanges, args, retained.query)?.exact,
  ).toBe(true);
});