pass initial next function through

koajs · Sep 25, 2015 · 1c9c3d1 · 1c9c3d1
1 parent 8b84aca
commit 1c9c3d1
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -38,7 +38,7 @@ function tryCatch(fn, ctx, next) {
 
 function middlewareReducer(next, mw) {
   return function(ctx, nextFn) {
-    ctx.next = next && function () { return next(ctx) }
+    ctx.next = next && function () { return next(ctx, nextFn) }
       || nextFn
       || noop
     return tryCatch(mw, ctx, ctx.next)

diff --git a/test/test.js b/test/test.js
@@ -179,6 +179,28 @@ describe('Koa Compose', function(){
     })
   })
 
+  it('should return a valid middleware', function () {
+    var val = 0
+    compose([
+      compose([
+        (ctx, next) => {
+          val++
+          return next()
+        },
+        (ctx, next) => {
+          val++
+          return next()
+        }
+      ]),
+      (ctx, next) => {
+        val++
+        return next()
+      }
+    ])({}).then(function () {
+      val.should.equal(3)
+    })
+  })
+
   it('should expose next on the context', function () {
     var stack = [];