@microsoft.azure/autorest.testserver/src/api/request-processor.ts

Autorest test server.

import { Response } from "express";
import { logger } from "../logger";
import { RequestExt } from "../server";
import { coverageService } from "../services";
import { Category } from "./mock-api-router";
import { MockRequest } from "./mock-request";
import { MockResponse } from "./mock-response";
import { ValidationError } from "./validation-error";

export type MockRequestHandler = (req: MockRequest) => MockResponse | Promise<MockResponse>;

export const processRequest = async (
  category: Category,
  name: string | undefined,
  request: RequestExt,
  response: Response,
  func: MockRequestHandler,
): Promise<void> => {
  const mockRequest = new MockRequest(request);
  const mockResponse = await callHandler(mockRequest, response, func);
  if (mockResponse === undefined) {
    return;
  }

  if ((mockResponse.status >= 200 && mockResponse.status < 300) || mockResponse.testSuccessful) {
    if (name) {
      await coverageService.track(category, name);
    }
  }
  processResponse(response, mockResponse);
};

const processResponse = (response: Response, mockResponse: MockResponse) => {
  response.status(mockResponse.status);

  if (mockResponse.headers) {
    response.set(mockResponse.headers);
  }

  if (mockResponse.body) {
    response.contentType(mockResponse.body.contentType).send(mockResponse.body.rawContent);
  }

  response.end();
};

const callHandler = async (
  mockRequest: MockRequest,
  response: Response,
  func: MockRequestHandler,
): Promise<MockResponse | undefined> => {
  try {
    return func(mockRequest);
  } catch (e) {
    if (!(e instanceof ValidationError)) {
      throw e;
    }

    logger.warn(
      [`Request validation failed: ${e.message}:`, ` Expected:\n${e.expected}`, ` Actual: \n${e.actual}`].join("\n"),
    );
    response
      .status(400)
      .contentType("application/json")
      .send(e.toJSON ? e.toJSON() : JSON.stringify(e.message))
      .end();
    return undefined;
  }
};