igo/src/dev/test/agent.js

Igo is a Node.js Web Framework based on Express

const _       = require('lodash');
const http    = require('http');

const url     = require('url');

const app     = require('../../app');

//
const mockRequest = function(options) {
  const req = {};

  req.hostname    = 'test';
  req.method      = options.method || 'GET';
  req.url         = options.url;
  req.originalUrl = options.url;
  req.path        = url.parse(options.url).pathname;
  req.query       = options.query   || require('querystring').parse(options.url.split('?')[1]);
  req.params      = options.params  || {};
  req.cookies     = options.cookies || {};
  req.session     = options.session || {};
  req.body        = options.body    || {};
  req.headers     = options.headers || {};
  req.files       = options.files   || {};
  req.resume      = function() {};
  req.listeners   = function() { return [] };
  req.unpipe      = function() {};
  req.connection  = {};

  return req;
};

//
const mockResponse = function(callback, req) {

  const res = {
    headers: {},
    _headers: {},
    _headerNames: {},
    locals: {
      flash: {}
    }
  };

  res.getHeader = function(name) {
    return res.headers[name];
  };

  res.setHeader = function(name, value) {
    res.headers[name] = value;
  };

  // http hack
  http.OutgoingMessage.prototype.setHeader = res.setHeader;

  res.redirect = function(statusCode, redirectUrl) {
    if (!_.isInteger(statusCode)) {
      redirectUrl = statusCode;
      statusCode  = 302;
    }
    res.statusCode  = statusCode;
    res.redirectUrl = redirectUrl;
    callback(null, res, req);
  };

  res.send = function(data) {
    res.body = data;
    callback(null, res, req);
  }

  return res;
};

//
module.exports.send = function(url, options, callback) {
  options.url = url;
  const req = mockRequest(options);
  const res = mockResponse(callback, req);

  app.handle(req, res);
};

//
module.exports.get = function(url, options, callback) {
  if (_.isFunction(options)) {
    callback  = options;
    options   = {};
  }
  options.method = 'GET';
  module.exports.send(url, options, callback);
};

//
module.exports.post = function(url, options, callback) {
  if (_.isFunction(options)) {
    callback  = options;
    options   = {};
  }
  options.method = 'POST';
  module.exports.send(url, options, callback);
};