Porting code from chris-troutner to PSF GitHub Repo

test/unit/mocks/ctx-mock.js

@@ -0,0 +1,50 @@
+// Ripped from https://github.com/koajs/koa/blob/master/test/helpers/context.js
+// Solution courtesy of user @fl0w. See: https://github.com/koajs/koa/issues/999#issuecomment-309270599
+// Take from this gist: https://gist.github.com/emmanuelnk/f1254eed8f947a81e8d715476d9cc92c
+
+// if you want more comprehensive Koa Context object to test stuff like Cookies etc
+// then use https://www.npmjs.com/package/@shopify/jest-koa-mocks (requires Jest)
+
+// INSTRUCTIONS:
+// Import in test file as below:
+//
+// const mockContext = require('./mocks/ctx-mock').context
+// const ctx = mockContext()
+// ...
+
+const Stream = require('stream')
+const Koa = require('koa')
+
+const context = (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()]
+
+  const retApp = app.createContext(req, res)
+
+  return retApp
+}
+
+const request = (req, res, app) => context(req, res, app).request
+
+const response = (req, res, app) => context(req, res, app).response
+
+module.exports = {
+  context,
+  request,
+  response
+}