Function startMockUpstream

test/test-e2e.ts:204–300

()

}

async function startMockUpstream(): Promise<MockUpstream> {
  const requests: MockRequest[] = [];

  const server = createServer(async (req: IncomingMessage, res: ServerResponse) => {
    const path = new URL(req.url ?? "/", "http://127.0.0.1").pathname;

    if (req.method === "GET" && path === "/mock-assets/tiny.png") {
      const data = Buffer.from(TINY_PNG_DATA_URI.split(",")[1]!, "base64");
      res.writeHead(200, { "Content-Type": "image/png", "Content-Length": data.length });
      res.end(data);
      return;
    }

    if (req.method === "GET" && path === "/mock-assets/tiny.mp3") {
      res.writeHead(200, { "Content-Type": "audio/mpeg", "Content-Length": TINY_MP3_BYTES.length });
      res.end(TINY_MP3_BYTES);
      return;
    }

    const bodyText = await readRequestText(req);
    let parsed: unknown;
    try {
      parsed = bodyText ? JSON.parse(bodyText) : undefined;
    } catch {
      parsed = undefined;
    }
    requests.push({ method: req.method ?? "GET", path, bodyText, json: parsed });

    if (req.method === "POST" && path === "/v1/images/generations") {
      writeJson(res, 200, {
        created: Math.floor(Date.now() / 1000),
        data: [{ url: TINY_PNG_DATA_URI, revised_prompt: "e2e mock image" }],
      });
      return;
    }

    if (req.method === "POST" && path === "/v1/audio/generations") {
      const port = getListeningPort(server);
      writeJson(res, 200, {
        created: Math.floor(Date.now() / 1000),
        model: "minimax/music-2.5+",
        data: [
          {
            url: `http://127.0.0.1:${port}/mock-assets/tiny.mp3`,
            duration_seconds: 1,
          },
        ],
      });
      return;
    }

    if (req.method !== "POST" || path !== "/v1/chat/completions") {
      writeJson(res, 404, { error: { message: `mock route not found: ${req.method} ${path}` } });
      return;
    }

    if (!parsed) {
      writeJson(res, 400, { error: { message: "Invalid JSON request body" } });