Setting retry timeout on entire retry operation

lib/index.js

@@ -2,15 +2,16 @@
 var retrier = require('retry');
 
 function retry(fn, opts) {
+  var options = opts || {};
+
+  // Default `randomize` to true
+  if (!('randomize' in options)) {
+    options.randomize = true;
+  }
+
   function run(resolve, reject) {
-    var options = opts || {};
     var op;
 
-    // Default `randomize` to true
-    if (!('randomize' in options)) {
-      options.randomize = true;
-    }
-
     op = retrier.operation(options);
 
     // We allow the user to abort retrying
@@ -55,7 +56,18 @@ function retry(fn, opts) {
     op.attempt(runAttempt);
   }
 
-  return new Promise(run);
+  // Setting up overall timeout for a retry
+  const { retryTimeout } = options;
+  return retryTimeout
+    ? Promise.race([
+        new Promise(run),
+        new Promise((_, reject) => {
+          setTimeout(() => {
+            reject(new Error(`Retry timed out in ${retryTimeout}ms`));
+          }, retryTimeout);
+        }),
+      ])
+    : new Promise(run);
 }
 
 module.exports = retry;

test/index.js

@@ -129,3 +129,19 @@ test('with number of retries', async t => {
     t.deepEqual(retries, 2);
   }
 });
+
+test('with retry timeout', async t => {
+  const retryTimeout = 2000;
+  try {
+    await retry(
+      async () => {
+        await sleep(3000);
+      },
+      {
+        retryTimeout,
+      }
+    );
+  } catch (err) {
+    t.deepEqual(err.message, `Retry timed out in ${retryTimeout}ms`);
+  }
+});