Handle async middleware rejections (#46)

Author: jkyberneees

lib/next.js

@@ -48,11 +48,17 @@ function next (middlewares, req, res, index, routers, defaultRoute, errorHandler
       }
 
       // Call router's lookup method
-      return middleware.lookup(req, res, step)
+      const result = middleware.lookup(req, res, step)
+      return result && typeof result.then === 'function'
+        ? result.catch(err => errorHandler(err, req, res))
+        : result
     }
 
     // Regular middleware function
-    return middleware(req, res, step)
+    const result = middleware(req, res, step)
+    return result && typeof result.then === 'function'
+      ? result.catch(err => errorHandler(err, req, res))
+      : result
   } catch (err) {
     return errorHandler(err, req, res)
   }

tests/router-coverage.test.js

@@ -366,6 +366,21 @@ describe('0http - Router Coverage', () => {
       expect(res.error).to.equal('Middleware exception')
     })
 
+    it('should handle async middleware rejections', async () => {
+      const req = {}
+      const res = {}
+      const middleware = async (req, res, next) => {
+        throw new Error('Async failure')
+      }
+      const defaultRoute = () => {}
+      const errorHandler = (err, req, res) => {
+        res.error = err.message
+      }
+
+      await next([middleware], req, res, 0, {}, defaultRoute, errorHandler)
+      expect(res.error).to.equal('Async failure')
+    })
+
     it('should handle nested router without pattern', () => {
       const req = { url: '/test', path: '/test' }
       const res = {}