Function startMockOpenAiServer

test/e2e/utils.ts:360–462

(
  response:
    | string
    | ((request: {
        authorization?: string;
        body: Record<string, any> | undefined;
        requestIndex: number;
      }) => {
        status?: number;
        body: Record<string, any>;
        headers?: Record<string, string>;
      })
)

};

export const startMockOpenAiServer = async (
  response:
    | string
    | ((request: {
        authorization?: string;
        body: Record<string, any> | undefined;
        requestIndex: number;
      }) => {
        status?: number;
        body: Record<string, any>;
        headers?: Record<string, string>;
      })
): Promise<{
  authHeaders: string[];
  requestBodies: Array<Record<string, any>>;
  baseUrl: string;
  cleanup: () => Promise<void>;
}> => {
  const authHeaders: string[] = [];
  const requestBodies: Array<Record<string, any>> = [];

  const server = http.createServer((req, res) => {
    const authorization = req.headers.authorization;
    if (authorization) {
      authHeaders.push(
        Array.isArray(authorization) ? authorization[0] : authorization
      );
    }

    const chunks: Buffer[] = [];
    req.on('data', (chunk) => {
      chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
    });
    req.on('end', () => {
      const rawBody = Buffer.concat(chunks).toString('utf8');
      let parsedBody: Record<string, any> | undefined;
      if (rawBody) {
        try {
          parsedBody = JSON.parse(rawBody);
          requestBodies.push(parsedBody);
        } catch {
          requestBodies.push({ rawBody });
        }
      }

      if (req.method === 'POST' && req.url?.includes('/chat/completions')) {
        const payload =
          typeof response === 'string'
            ? {
                status: 200,
                body: {
                  choices: [
                    {
                      message: {
                        content: response
                      }
                    }
                  ]