callstack__agent-device/test/integration/provider-scenarios/cloud-webdriver-test-server.ts

import type { IncomingHttpHeaders } from 'node:http';

export type CloudWebDriverHttpCall = {
  method: string;
  path: string;
  headers: IncomingHttpHeaders;
  body?: unknown;
};

export type CloudWebDriverTestResponse = {
  body: unknown;
  status?: number;
};

/**
 * In-memory Fetch transport for Cloud WebDriver integration scenarios.
 *
 * The production client still uses Fetch; this fixture observes the exact
 * request URL, headers and body, then supplies a real Response. Avoiding a
 * loopback listener keeps the suite hermetic in sandboxes that prohibit
 * binding ports without reducing request/response transport coverage.
 */
export abstract class CloudWebDriverTestServer {
  readonly calls: CloudWebDriverHttpCall[] = [];
  // fallow-ignore-next-line unused-class-member
  readonly url = 'http://cloud-webdriver.test';

  protected abstract respond(call: CloudWebDriverHttpCall): CloudWebDriverTestResponse;

  // fallow-ignore-next-line unused-class-member
  readonly fetch: typeof globalThis.fetch = async (input, init) => {
    const request = new Request(input, init);
    const call: CloudWebDriverHttpCall = {
      method: request.method,
      path: new URL(request.url).pathname,
      headers: Object.fromEntries(request.headers.entries()),
      ...(await requestBody(request)),
    };
    this.calls.push(call);
    const response = this.respond(call);
    return new Response(JSON.stringify(response.body), {
      status: response.status ?? 200,
      headers: { 'Content-Type': 'application/json' },
    });
  };
}

export type StartedCloudWebDriverTestServer<T extends CloudWebDriverTestServer> = T & {
  close(): Promise<void>;
};

export async function startCloudWebDriverTestServer<T extends CloudWebDriverTestServer>(
  testServer: T,
): Promise<StartedCloudWebDriverTestServer<T>> {
  const previousFetch = globalThis.fetch;
  globalThis.fetch = testServer.fetch;
  return Object.assign(testServer, {
    close: async () => {
      if (globalThis.fetch === testServer.fetch) globalThis.fetch = previousFetch;
    },
  });
}

export function cloudWebDriverTestJson(body: unknown, status = 200): CloudWebDriverTestResponse {
  return { body, status };
}

async function requestBody(request: Request): Promise<{ body?: unknown }> {
  if (!request.body) return {};
  const buffer = Buffer.from(await request.arrayBuffer());
  if (request.headers.get('content-type')?.startsWith('multipart/form-data')) {
    return { body: { multipartBytes: buffer.length } };
  }
  const text = buffer.toString('utf8');
  return text ? { body: JSON.parse(text) as unknown } : {};
}