diff --git a/.gitignore b/.gitignore index 5c843fd..07b03e9 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,4 @@ node_modules/ .nyc_output/ coverage/ +.idea/ \ No newline at end of file diff --git a/index.js b/index.js index af98442..996c915 100644 --- a/index.js +++ b/index.js @@ -22,9 +22,9 @@ function stubMethod(service, method, replacement) { if (!isStubbed(service)) stubService(service); if (!replacement) return sinon.stub(getService(service).prototype, method); - return sinon.stub(getService(service).prototype, method).callsFake(function(params, callback) { + return sinon.stub(getService(service).prototype, method).callsFake(function(arg1, arg2, arg3) { var _this = { request: stubRequest(), response: stubResponse() }; - replacement.call(_this, params, callback); + replacement.call(_this, arg1, arg2, arg3); return _this.request; }); }