Collections: add Symbol.iterator

Signed-off-by: Rick Waldron <waldron.rick@gmail.com>
rwaldron · May 2, 2017 · be3e8bd · islemaster · May 2, 2017 · rwaldron
1 parent 291a1db
commit be3e8bd
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/lib/mixins/collection.js b/lib/mixins/collection.js
@@ -59,6 +59,10 @@ function Collection(numsOrObjects) {
   }
 }
 
+if (Symbol && Symbol.iterator) {
+  Collection.prototype[Symbol.iterator] = Array.prototype[Symbol.iterator];
+}
+
 Collection.prototype.add = function() {
   var length = this.length;
   var aLen = arguments.length;
@@ -230,6 +234,10 @@ util.inherits(Collection.Emitter, Collection);
 
 Object.assign(Collection.Emitter.prototype, Emitter.prototype);
 
+if (Symbol && Symbol.iterator) {
+  Collection.prototype[Symbol.iterator] = Array.prototype[Symbol.iterator];
+}
+
 Collection.Emitter.prototype.add = function() {
   var inputs = Array.from(arguments);
 

diff --git a/test/collections.js b/test/collections.js
@@ -131,6 +131,15 @@ exports["Collection"] = {
     Collection.purge();
     done();
   },
+
+  "Symbol.iterator": function(test) {
+    if (Symbol && Symbol.iterator) {
+      test.expect(1);
+      test.equal(Collection.prototype[Symbol.iterator], Array.prototype[Symbol.iterator]);
+    }
+    test.done();
+  },
+
   nested: function(test) {
     test.expect(9);
 
@@ -322,6 +331,14 @@ exports["Collection.Emitter"] = {
     done();
   },
 
+  "Symbol.iterator": function(test) {
+    if (Symbol && Symbol.iterator) {
+      test.expect(1);
+      test.equal(Collection.prototype[Symbol.iterator], Array.prototype[Symbol.iterator]);
+    }
+    test.done();
+  },
+
   nested: function(test) {
     test.expect(9);