replay/lib/patch_http_request.js

When API testing slows you down: record and replay HTTP responses like a boss

'use strict';

var _assign = require('babel-runtime/core-js/object/assign');

var _assign2 = _interopRequireDefault(_assign);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

// Patch HTTP.request to make all request go through Replay
//
// Patch based on io.js, may not work with Node.js

const HTTP = require('http');
const HTTPS = require('https');
const ProxyRequest = require('./proxy');
const Replay = require('./');
const URL = require('url');

// Route HTTP requests to our little helper.
HTTP.request = function (options, callback) {
  if (typeof options === 'string' || options instanceof String) options = URL.parse(options);

  // WebSocket request: pass through to Node.js library
  if (options.headers && options.headers.Upgrade === 'websocket') return new HTTP.ClientRequest(options, callback);

  const hostname = options.hostname || options.host && options.host.split(':')[0] || 'localhost';
  if (Replay.isLocalhost(hostname) || Replay.isPassThrough(hostname)) return new HTTP.ClientRequest(options, callback);

  // Proxy request
  const request = new ProxyRequest(options, Replay.chain.start);
  if (callback) request.once('response', callback);
  return request;
};

HTTPS.request = function (options, callback) {
  if (typeof options === 'string' || options instanceof String) options = URL.parse(options);

  const httpsOptions = (0, _assign2.default)({
    protocol: 'https:',
    port: 443,
    // Allows agent: false to opt-out of connection pooling.
    _defaultAgent: HTTPS.globalAgent
  }, options);

  return HTTP.request(httpsOptions, callback);
};

// Patch .get method otherwise it calls original HTTP.request
HTTP.get = function (options, cb) {
  const req = HTTP.request(options, cb);
  req.end();
  return req;
};

HTTPS.get = function (options, cb) {
  const req = HTTPS.request(options, cb);
  req.end();
  return req;
};
//# sourceMappingURL=patch_http_request.js.map