unexpected-express/lib/classifyRequestBodyAndUpdateHttpRequest.js

Extend the unexpected assertion library with support for testing Express middleware

module.exports = function classifyRequestBodyAndUpdateHttpRequest(
  requestBody,
  httpRequest
) {
  let hasFormData = false;
  let hasStream = false;

  if (typeof requestBody === 'undefined') {
    return null;
  }

  httpRequest.headers.set('Transfer-Encoding', 'chunked');
  if (requestBody.pipe) {
    hasStream = true;
    if (
      requestBody.constructor &&
      requestBody.constructor.name === 'FormData'
    ) {
      hasFormData = true;
      if (!httpRequest.headers.has('Content-Type')) {
        httpRequest.headers.set(
          'Content-Type',
          `multipart/form-data; boundary=${requestBody.getBoundary()}`
        );
      }
    }
  } else {
    if (typeof requestBody === 'object' && !Buffer.isBuffer(requestBody)) {
      if (!httpRequest.headers.has('Content-Type')) {
        httpRequest.headers.set('Content-Type', 'application/json');
      }
    }

    if (
      !httpRequest.headers.has('Content-Length') &&
      !httpRequest.headers.has('Transfer-Encoding')
    ) {
      httpRequest.headers.set('Content-Length', String(requestBody.length));
    }
  }

  return {
    hasFormData,
    hasStream,
    requestBody: !hasStream ? httpRequest.unchunkedBody : requestBody,
  };
};