happner/test/lib/helper_mock_req.js

distributed application engine with evented storage and mesh services

module.exports = MockIncomingMessage;

var Transform = require('stream').Transform,
  util = require('util');

function MockIncomingMessage(options) {
  var self = this;
  options = options || {};

  Transform.call(this);
  this._writableState.objectMode = true;
  this._readableState.objectMode = false;

  // Copy unreserved options
  var reservedOptions = [
    'method',
    'url',
    'headers',
    'rawHeaders'
  ];

  Object.keys(options).forEach(function(key) {
    if (reservedOptions.indexOf(key) === -1)
      self[key] = options[key];
  });

  this.method = options.method || 'GET';
  this.url = options.url || '';

  // Set header names
  this.headers = {};
  this.rawHeaders = [];
  if (options.headers)
    Object.keys(options.headers).forEach(function(key) {
      var val = options.headers[key];

      if(val !== undefined) {
        if (typeof val !== 'string') {
          val += '';
        }

        self.headers[key.toLowerCase()] = val;
        self.rawHeaders.push(key);
        self.rawHeaders.push(val);
      }
    });

  // Auto-end when no body
  if (this.method === 'GET' || this.method === 'HEAD' || this.method === 'DELETE')
    this.end();
}

util.inherits(MockIncomingMessage, Transform);

MockIncomingMessage.prototype._transform = function(chunk, encoding, next) {
  if (this._failError)
    return this.emit('error', this._failError);

  if (typeof chunk !== 'string' && !Buffer.isBuffer(chunk))
    chunk = JSON.stringify(chunk);

  this.push(chunk);

  next();
};

// Causes the request to emit an error when the body is read.
MockIncomingMessage.prototype._fail = function(error) {
  this._failError = error;
};