From 61a3b935ac5cdb2e5507678627f0ac39d3a49d4a Mon Sep 17 00:00:00 2001
From: Andesr Gjetting <nye@topdanmark.dk>
Date: Wed, 27 Dec 2017 14:55:14 +0100
Subject: [PATCH] added extra argument to stub method as some AWS sdk methods
 take more than 2 arguments e.g waitFor(state, params = {}, callback)

---
 .gitignore | 1 +
 index.js   | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

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;
   });
 }