Feat network error once (#141)

* Add shortcut function networkErrorOnce and timoutOnce * Set replyOnce function
ctimmerm · Jun 20, 2018 · 4a0e874 · 4a0e874
1 parent 8abf4a5
commit 4a0e874
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 5 deletions.
diff --git a/src/index.js b/src/index.js
@@ -76,14 +76,16 @@ VERBS.concat('any').forEach(function(method) {
       return _this;
     }
 
+    function replyOnce(code, response, headers) {
+      var handler = [matcher, body, requestHeaders, code, response, headers, true];
+      addHandler(method, _this.handlers, handler);
+      return _this;
+    }
+
     return {
       reply: reply,
 
-      replyOnce: function replyOnce(code, response, headers) {
-        var handler = [matcher, body, requestHeaders, code, response, headers, true];
-        addHandler(method, _this.handlers, handler);
-        return _this;
-      },
+      replyOnce: replyOnce,
 
       passThrough: function passThrough() {
         var handler = [matcher, body];
@@ -99,13 +101,30 @@ VERBS.concat('any').forEach(function(method) {
         });
       },
 
+      networkErrorOnce: function() {
+        replyOnce(function(config) {
+          var error = new Error('Network Error');
+          error.config = config;
+          return Promise.reject(error);
+        });
+      },
+
       timeout: function() {
         reply(function(config) {
           var error = new Error('timeout of ' + config.timeout + 'ms exceeded');
           error.config = config;
           error.code = 'ECONNABORTED';
           return Promise.reject(error);
         });
+      },
+
+      timeoutOnce: function() {
+        replyOnce(function(config) {
+          var error = new Error('timeout of ' + config.timeout + 'ms exceeded');
+          error.config = config;
+          error.code = 'ECONNABORTED';
+          return Promise.reject(error);
+        });
       }
     };
   };

diff --git a/types/index.d.ts b/types/index.d.ts
@@ -7,6 +7,8 @@ type ResponseSpecFunc = (statusOrCallback: number | CallbackResponseSpecFunc, da
 interface RequestHandler {
   reply: ResponseSpecFunc;
   replyOnce: ResponseSpecFunc;
+  timeoutOnce: ResponseSpecFunc;
+  networkErrorOnce: ResponseSpecFunc;
 
   passThrough(): void;
   networkError(): void;