add more tests

Author: OmgImAlexis

package.json

@@ -1,7 +1,6 @@
 {
   "name": "koa-404-handler",
-  "description":
-    "404 handler for Lad and Koa (best used with koa-better-error-handler)",
+  "description": "404 handler for Lad and Koa (best used with koa-better-error-handler)",
   "version": "0.0.1",
   "author": "Nick Baugh <[email protected]> (http://niftylettuce.com/)",
   "bugs": {
@@ -14,17 +13,21 @@
   "dependencies": {},
   "devDependencies": {
     "ava": "^0.22.0",
+    "boom": "^7.1.1",
     "codecov": "^2.3.0",
     "cross-env": "^5.0.5",
     "eslint": "^4.5.0",
     "eslint-config-prettier": "^2.3.0",
     "eslint-plugin-prettier": "^2.2.0",
     "husky": "^0.14.3",
+    "koa": "^2.4.1",
+    "koa-router": "^7.3.0",
     "lint-staged": "^4.0.4",
     "nyc": "^11.1.0",
     "prettier": "^1.6.1",
     "remark-cli": "^4.0.0",
     "remark-preset-github": "^0.0.6",
+    "supertest": "^3.0.0",
     "xo": "^0.19.0"
   },
   "engines": {
@@ -48,11 +51,16 @@
       "prettier --write --single-quote --trailing-comma none",
       "git add"
     ],
-    "*.md": ["remark . -qfo", "git add"]
+    "*.md": [
+      "remark . -qfo",
+      "git add"
+    ]
   },
   "main": "index.js",
   "remarkConfig": {
-    "plugins": ["preset-github"]
+    "plugins": [
+      "preset-github"
+    ]
   },
   "repository": {
     "type": "git",
@@ -67,7 +75,9 @@
   },
   "xo": {
     "extends": "prettier",
-    "plugins": ["prettier"],
+    "plugins": [
+      "prettier"
+    ],
     "parserOptions": {
       "sourceType": "script"
     },

test/test.js

@@ -1,7 +1,66 @@
+const Koa = require('koa');
+const Router = require('koa-router');
+const Boom = require('boom');
 const test = require('ava');
+const request = require('supertest');
+const koa404Handler = require('..');
 
-const koa404Handler = require('../');
+const ok = ctx => {
+  ctx.status = 200;
+  ctx.body = { ok: 'ok' };
+};
+
+const error = () => {
+  throw new Error('Big Bad Error!');
+};
 
 test('returns a function', t => {
   t.true(typeof koa404Handler === 'function');
 });
+
+test.failing('middleware can be added to the app at any stage', async t => {
+  const app = new Koa();
+  const router = new Router();
+
+  router.get('/', ok);
+
+  // Mount the app's defined and nested routes
+  app.use(router.routes());
+
+  // Options method
+  app.use(
+    router.allowedMethods({
+      throw: true,
+      notImplemented: () => new Boom.notImplemented(), // eslint-disable-line new-cap,max-len
+      methodNotAllowed: () => new Boom.methodNotAllowed() // eslint-disable-line new-cap,max-len
+    })
+  );
+
+  // 404 handler
+  app.use(koa404Handler);
+
+  const res = await request(app.listen()).options('/');
+
+  t.is(200, res.status);
+  t.is(res.body.ok, 'ok');
+});
+
+test('emits error on app instance', async t => {
+  const app = new Koa();
+  const router = new Router();
+
+  router.get('/', error);
+
+  // Mount the app's defined and nested routes
+  app.use(router.routes());
+
+  // 404 handler
+  app.use(koa404Handler);
+
+  app.on('error', error => {
+    t.truthy(error);
+    t.is(error.message, 'Big Bad Error!');
+  });
+
+  await request(app.listen()).get('/');
+});