moar coverage

3 files changed, 29 insertions, 0 deletions

diff --git a/lib/internal/setImmediate.js b/lib/internal/setImmediate.js
index e14e52e..96d26d0 100644
--- a/lib/internal/setImmediate.js
+++ b/lib/internal/setImmediate.js
@@ -1,4 +1,5 @@
 'use strict';
+/* istanbul ignore file */
 
 export var hasSetImmediate = typeof setImmediate === 'function' && setImmediate;
 export var hasNextTick = typeof process === 'object' && typeof process.nextTick === 'function';
diff --git a/test/es2017/awaitableFunctions.js b/test/es2017/awaitableFunctions.js
index a5c9e7b..de54e8f 100644
--- a/test/es2017/awaitableFunctions.js
+++ b/test/es2017/awaitableFunctions.js
@@ -562,6 +562,17 @@ module.exports = function () {
         expect(calls).to.eql([0, 1, 2])
     });
 
+    it('should return a Promise: retry (multiple cb args)', async () => {
+        expect (async.retry.name).to.contain('retry')
+        let counter = 0
+        const results = await async.retry((cb) => {
+            counter++
+            if (counter < 3) return cb(new Error())
+            cb(null, 0, 1, 2)
+        })
+        expect(results).to.eql([0, 1, 2])
+    });
+
     it('should return a Promise: tryEach', async () => {
         expect (async.tryEach.name).to.contain('tryEach')
         const calls = []
diff --git a/test/retryable.js b/test/retryable.js
index 0107e53..bba4291 100644
--- a/test/retryable.js
+++ b/test/retryable.js
@@ -18,6 +18,23 @@ describe('retryable', () => {
         });
     });
 
+    it('success', (done) => {
+        var calls = 0;
+        var retryableTask = async.retryable(3, (arg, cb) => {
+            calls++;
+            expect(arg).to.equal(42);
+            if (calls > 1) return cb (null, 1, 2)
+            cb('fail');
+        });
+
+        retryableTask(42, (err, a, b) => {
+            expect(err).to.eql(null);
+            expect(calls).to.equal(2);
+            expect([a, b]).to.eql([1, 2])
+            done();
+        });
+    });
+
     it('basics with error test function', (done) => {
         var calls = 0;
         var special = 'special';