koa-lite/test/helpers/context.js

22 lines
811 B
JavaScript

'use strict';
const Stream = require('stream');
const Koa = require('../..');
module.exports = (req, res, app) => {
const socket = new Stream.Duplex();
req = Object.assign({ headers: {}, socket }, Stream.Readable.prototype, req);
res = Object.assign({ _headers: {}, socket }, Stream.Writable.prototype, res);
req.socket.remoteAddress = req.socket.remoteAddress || '127.0.0.1';
app = app || new Koa();
res.getHeader = k => res._headers[k.toLowerCase()];
res.setHeader = (k, v) => res._headers[k.toLowerCase()] = v;
res.removeHeader = (k, v) => delete res._headers[k.toLowerCase()];
return app.createContext(req, res);
};
module.exports.request = (req, res, app) => module.exports(req, res, app).request;
module.exports.response = (req, res, app) => module.exports(req, res, app).response;