sjfloat
diff --git a/‎__tests__/IterableSeq.ts
Lines changed: 12 additions & 9 deletions b/‎__tests__/IterableSeq.ts
Lines changed: 12 additions & 9 deletions
@@ -1,5 +1,6 @@
 ///<reference path='../resources/jest.d.ts'/>
 ///<reference path='../dist/immutable.d.ts'/>
+declare var Symbol: any;
 jest.autoMockOff();
 
 import Immutable = require('immutable');
@@ -78,21 +79,21 @@ describe('IterableSequence', () => {
 
     it('creates a sequence from a raw iterable', () => {
       var i = new SimpleIterable(10);
-      var s = Immutable.Seq(i['@@iterator']());
+      var s = Immutable.Seq(i[ITERATOR_SYMBOL]());
       expect(s.take(5).toArray()).toEqual([ 0,1,2,3,4 ]);
     })
 
     it('is stable', () => {
       var i = new SimpleIterable(10);
-      var s = Immutable.Seq(i['@@iterator']());
+      var s = Immutable.Seq(i[ITERATOR_SYMBOL]());
       expect(s.take(5).toArray()).toEqual([ 0,1,2,3,4 ]);
       expect(s.take(5).toArray()).toEqual([ 0,1,2,3,4 ]);
       expect(s.take(5).toArray()).toEqual([ 0,1,2,3,4 ]);
     })
 
     it('counts iterations', () => {
       var i = new SimpleIterable(10);
-      var s = Immutable.Seq(i['@@iterator']());
+      var s = Immutable.Seq(i[ITERATOR_SYMBOL]());
       expect(s.forEach(x => x)).toEqual(10);
       expect(s.take(5).forEach(x => x)).toEqual(5);
       expect(s.forEach(x => x < 3)).toEqual(4);
@@ -101,7 +102,7 @@ describe('IterableSequence', () => {
     it('memoizes the iterator', () => {
       var mockFn = jest.genMockFunction();
       var i = new SimpleIterable(10, mockFn);
-      var s = Immutable.Seq(i['@@iterator']());
+      var s = Immutable.Seq(i[ITERATOR_SYMBOL]());
       expect(s.take(3).toArray()).toEqual([ 0,1,2 ]);
       expect(mockFn.mock.calls).toEqual([[0],[1],[2]]);
 
@@ -117,7 +118,7 @@ describe('IterableSequence', () => {
     it('can be iterated', () => {
       var mockFn = jest.genMockFunction();
       var i = new SimpleIterable(3, mockFn);
-      var seq = Immutable.Seq(i['@@iterator']());
+      var seq = Immutable.Seq(i[ITERATOR_SYMBOL]());
       var entries = seq.entries();
       expect(entries.next()).toEqual({ value: [0, 0], done: false });
       // The iteration is lazy
@@ -137,11 +138,11 @@ describe('IterableSequence', () => {
 
     it('can iterate an skipped seq based on an iterator', () => {
       var i = new SimpleIterable(4);
-      var seq = Immutable.Seq(i['@@iterator']());
+      var seq = Immutable.Seq(i[ITERATOR_SYMBOL]());
       expect(seq.size).toBe(undefined);
       var skipped = seq.skip(2);
       expect(skipped.size).toBe(undefined);
-      var iter = skipped['@@iterator']();
+      var iter = skipped[ITERATOR_SYMBOL]();
       // The first two were skipped
       expect(iter.next()).toEqual({ value: 2, done: false });
       expect(iter.next()).toEqual({ value: 3, done: false });
@@ -153,12 +154,14 @@ describe('IterableSequence', () => {
 
 
 // Helper for this test
+var ITERATOR_SYMBOL =
+  typeof Symbol === 'function' && Symbol.iterator || '@@iterator';
 
 function SimpleIterable(max?: number, watcher?: any) {
   this.max = max;
   this.watcher = watcher;
 }
-SimpleIterable.prototype['@@iterator'] = function() {
+SimpleIterable.prototype[ITERATOR_SYMBOL] = function() {
   return new SimpleIterator(this);
 }
 
@@ -173,6 +176,6 @@ SimpleIterator.prototype.next = function() {
   this.iterable.watcher && this.iterable.watcher(this.value);
   return { value: this.value++, done: false };
 }
-SimpleIterator.prototype['@@iterator'] = function() {
+SimpleIterator.prototype[ITERATOR_SYMBOL] = function() {
   return this;
 }