test(each): add generator test

suguru03 · May 13, 2019 · 6a20cee · 6a20cee
1 parent a111632
commit 6a20cee
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/test/collections/test.each.js b/test/collections/test.each.js
@@ -1363,4 +1363,27 @@ parallel('#eachLimit', function() {
     });
   });
 
+  it('should work with generator', function(done) {
+    var limit = 2;
+    var collection = ['abc', 'def', 'ghi', 'jkl'];
+    var order = [];
+    var gen = util.makeGenerator(collection);
+
+    async.eachLimit(
+      gen,
+      limit,
+      function(v, i, cb) {
+        order.push(i);
+        setTimeout(cb, Math.random() * delay, null, v);
+      },
+      function(err) {
+        if (err) {
+          return done();
+        }
+        assert.deepStrictEqual(order, [0, 1, 2, 3]);
+        done();
+      }
+    );
+  });
+
 });