1 files changed, 65 insertions, 0 deletions
diff --git a/mocha_test/retryable.js b/mocha_test/retryable.js
new file mode 100644
index 0000000..8cad403
--- /dev/null
+++ b/mocha_test/retryable.js
@@ -0,0 +1,65 @@
+var async = require('../lib');
+var expect = require('chai').expect;
+var assert = require('assert');
+
+describe('retryable', function () {
+    it('basics', function (done) {
+        var calls = 0;
+        var retryableTask = async.retryable(3, function (arg, cb) {
+            calls++;
+            expect(arg).to.equal(42);
+            cb('fail');
+        });
+
+        retryableTask(42, function (err) {
+            expect(err).to.equal('fail');
+            expect(calls).to.equal(3);
+            done();
+        });
+
+        setTimeout(function () {
+        }, 15);
+    });
+
+    // TODO: re-enable when auto's args are swapped
+    it.skip('should work as an embedded task', function(done) {
+        var retryResult = 'RETRY';
+        var fooResults;
+        var retryResults;
+
+        async.auto({
+            dep: async.constant('dep'),
+            foo: ['dep', function(results, callback){
+                fooResults = results;
+                callback(null, 'FOO');
+            }],
+            retry: ['dep', async.retryable(function(results, callback) {
+                retryResults = results;
+                callback(null, retryResult);
+            })]
+        }, function(err, results){
+            assert.equal(results.retry, retryResult, "Incorrect result was returned from retry function");
+            assert.equal(fooResults, retryResults, "Incorrect results were passed to retry function");
+            done();
+        });
+    });
+
+    it('should work as an embedded task with interval', function(done) {
+        var start = new Date().getTime();
+        var opts = {times: 5, interval: 100};
+
+        async.auto({
+            foo: function(callback){
+                callback(null, 'FOO');
+            },
+            retry: async.retryable(opts, function(callback) {
+                callback('err');
+            })
+        }, function(){
+            var duration = new Date().getTime() - start;
+            var expectedMinimumDuration = (opts.times -1) * opts.interval;
+            assert(duration >= expectedMinimumDuration, "The duration should have been greater than " + expectedMinimumDuration + ", but was " + duration);
+            done();
+        });
+    });
+});