OmniRoute/tests/unit/chipotle-executor.test.ts

import { describe, it } from "node:test";
import assert from "node:assert";
import { ChipotleExecutor } from "../../open-sse/executors/chipotle.ts";

const executor = new ChipotleExecutor();

describe("ChipotleExecutor", () => {
  it("buildHeaders returns static headers", () => {
    const headers = (executor as any).buildHeaders({});
    assert.strictEqual(headers["Content-Type"], "application/json");
  });

  it("buildUrl returns Amelia endpoint", () => {
    const url = executor.buildUrl("pepper-1", false);
    assert.ok(url.includes("amelia.chipotle.com"));
  });

  it("transformRequest passes model through", () => {
    const result = (executor as any).transformRequest(
      "pepper-1",
      { model: "pepper-1", messages: [{ role: "user", content: "hi" }] },
      false,
    );
    assert.strictEqual(result.model, "pepper-1");
  });

  it("returns 499 on pre-aborted signal", async () => {
    const controller = new AbortController();
    controller.abort(new Error("cancelled"));

    const result = await executor.execute({
      model: "pepper-1",
      body: { messages: [{ role: "user", content: "hi" }], stream: false },
      stream: false,
      signal: controller.signal,
      credentials: {},
      log: { debug: () => {}, info: () => {}, warn: () => {}, error: () => {} },
    });

    assert.strictEqual((result as any).response.status, 499);
  });

  it("is registered in executor index", async () => {
    const { getExecutor } = await import("../../open-sse/executors/index.ts");
    const exec = getExecutor("chipotle");
    assert.ok(exec, "chipotle executor should be registered");
    assert.ok(exec instanceof ChipotleExecutor);
  });

  it("pepper alias works", async () => {
    const { getExecutor } = await import("../../open-sse/executors/index.ts");
    const exec = getExecutor("pepper");
    assert.ok(exec, "pepper alias should be registered");
    assert.ok(exec instanceof ChipotleExecutor);
  });
});