strapi/test/helpers/context.js

'use strict';

const Stream = require('stream');

const Koa = require('../..').server;

module.exports = function (req, res) {
  const socket = new Stream.Duplex();

  req = req || {
    headers: {},
    socket: socket,
    __proto__: Stream.Readable.prototype
  };

  res = res || {
    _headers: {},
    socket: socket,
    __proto__: Stream.Writable.prototype
  };

  res.getHeader = function (k) {
    return res._headers[k.toLowerCase()];
  };

  res.setHeader = function (k, v) {
    res._headers[k.toLowerCase()] = v;
  };

  res.removeHeader = function (k) {
    delete res._headers[k.toLowerCase()];
  };

  return (new Koa()).createContext(req, res);
};

module.exports.request = function (req, res) {
  return module.exports(req, res).request;
};

module.exports.response = function (req, res) {
  return module.exports(req, res).response;
};