From 0d1d1f5c4ee163af465f58e7fe3c8251fc5a3d0f Mon Sep 17 00:00:00 2001
From: Thibault REMY <thibremy.ep@gmail.com>
Date: Mon, 7 Mar 2016 23:55:07 +0100
Subject: [PATCH] fix: authenticate

---
 index.js                | 3 ++-
 test/unit/index_test.js | 5 +++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/index.js b/index.js
index fda674f..8fb8b08 100644
--- a/index.js
+++ b/index.js
@@ -41,10 +41,11 @@ KoaOAuthServer.prototype.authenticate = function() {
 
   return function *(next) {
     var request = new Request(this.request);
+    var response = new Response(this.response);
 
     try {
       this.state.oauth = {
-        token: yield server.authenticate(request)
+        token: yield server.authenticate(request, response)
       };
     } catch (e) {
       return handleError.call(this, e);
diff --git a/test/unit/index_test.js b/test/unit/index_test.js
index 9505562..c27af73 100644
--- a/test/unit/index_test.js
+++ b/test/unit/index_test.js
@@ -32,10 +32,11 @@ describe('KoaOAuthServer', function() {
       yield request(app.listen())
         .get('/')
         .end();
-
+  
       oauth.server.authenticate.callCount.should.equal(1);
-      oauth.server.authenticate.firstCall.args.should.have.length(1);
+      oauth.server.authenticate.firstCall.args.should.have.length(2);
       oauth.server.authenticate.firstCall.args[0].should.be.an.instanceOf(Request);
+      oauth.server.authenticate.firstCall.args[1].should.be.an.instanceOf(Response);
       oauth.server.authenticate.restore();
     });
   });