michael@0: // This file was written by Andy Wingo and originally michael@0: // contributed to V8 as generators-objects.js, available here: michael@0: // michael@0: // http://code.google.com/p/v8/source/browse/branches/bleeding_edge/test/mjsunit/harmony/generators-objects.js michael@0: michael@0: // Test aspects of the generator runtime. michael@0: michael@0: michael@0: var GeneratorFunction = (function*(){yield 1;}).constructor; michael@0: michael@0: michael@0: function TestGeneratorResultPrototype() { michael@0: function* g() { yield 1; } michael@0: var iter = g(); michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorDone(iter, undefined); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: TestGeneratorResultPrototype(); michael@0: michael@0: function TestGenerator(g, expected_values_for_next, michael@0: send_val, expected_values_for_send) { michael@0: function testNext(thunk) { michael@0: var iter = thunk(); michael@0: for (var i = 0; i < expected_values_for_next.length; i++) { michael@0: assertIteratorResult(iter.next(), expected_values_for_next[i], michael@0: i == expected_values_for_next.length - 1); michael@0: } michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: function testSend(thunk) { michael@0: var iter = thunk(); michael@0: for (var i = 0; i < expected_values_for_send.length; i++) { michael@0: assertIteratorResult(i ? iter.next(send_val) : iter.next(), michael@0: expected_values_for_send[i], michael@0: i == expected_values_for_send.length - 1); michael@0: } michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: function testThrow(thunk) { michael@0: for (var i = 0; i < expected_values_for_next.length; i++) { michael@0: var iter = thunk(); michael@0: for (var j = 0; j < i; j++) { michael@0: assertIteratorResult(iter.next(), michael@0: expected_values_for_next[j], michael@0: j == expected_values_for_next.length - 1); michael@0: } michael@0: var Sentinel = function () {} michael@0: assertThrowsInstanceOf(function () { iter.throw(new Sentinel); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: } michael@0: michael@0: testNext(g); michael@0: testSend(g); michael@0: testThrow(g); michael@0: michael@0: testNext(function*() { return yield* g(); }); michael@0: testSend(function*() { return yield* g(); }); michael@0: testThrow(function*() { return yield* g(); }); michael@0: michael@0: if (g instanceof GeneratorFunction) { michael@0: testNext(function() { return new g(); }); michael@0: testSend(function() { return new g(); }); michael@0: testThrow(function() { return new g(); }); michael@0: } michael@0: } michael@0: michael@0: TestGenerator(function* g1() { }, michael@0: [undefined], michael@0: "foo", michael@0: [undefined]); michael@0: michael@0: TestGenerator(function* g2() { yield 1; }, michael@0: [1, undefined], michael@0: "foo", michael@0: [1, undefined]); michael@0: michael@0: TestGenerator(function* g3() { yield 1; yield 2; }, michael@0: [1, 2, undefined], michael@0: "foo", michael@0: [1, 2, undefined]); michael@0: michael@0: TestGenerator(function* g4() { yield 1; yield 2; return 3; }, michael@0: [1, 2, 3], michael@0: "foo", michael@0: [1, 2, 3]); michael@0: michael@0: TestGenerator(function* g5() { return 1; }, michael@0: [1], michael@0: "foo", michael@0: [1]); michael@0: michael@0: TestGenerator(function* g6() { var x = yield 1; return x; }, michael@0: [1, undefined], michael@0: "foo", michael@0: [1, "foo"]); michael@0: michael@0: TestGenerator(function* g7() { var x = yield 1; yield 2; return x; }, michael@0: [1, 2, undefined], michael@0: "foo", michael@0: [1, 2, "foo"]); michael@0: michael@0: TestGenerator(function* g8() { for (var x = 0; x < 4; x++) { yield x; } }, michael@0: [0, 1, 2, 3, undefined], michael@0: "foo", michael@0: [0, 1, 2, 3, undefined]); michael@0: michael@0: // Generator with arguments. michael@0: TestGenerator( michael@0: function g9() { michael@0: return (function*(a, b, c, d) { michael@0: yield a; yield b; yield c; yield d; michael@0: })("fee", "fi", "fo", "fum"); michael@0: }, michael@0: ["fee", "fi", "fo", "fum", undefined], michael@0: "foo", michael@0: ["fee", "fi", "fo", "fum", undefined]); michael@0: michael@0: // Too few arguments. michael@0: TestGenerator( michael@0: function g10() { michael@0: return (function*(a, b, c, d) { michael@0: yield a; yield b; yield c; yield d; michael@0: })("fee", "fi"); michael@0: }, michael@0: ["fee", "fi", undefined, undefined, undefined], michael@0: "foo", michael@0: ["fee", "fi", undefined, undefined, undefined]); michael@0: michael@0: // Too many arguments. michael@0: TestGenerator( michael@0: function g11() { michael@0: return (function*(a, b, c, d) { michael@0: yield a; yield b; yield c; yield d; michael@0: })("fee", "fi", "fo", "fum", "I smell the blood of an Englishman"); michael@0: }, michael@0: ["fee", "fi", "fo", "fum", undefined], michael@0: "foo", michael@0: ["fee", "fi", "fo", "fum", undefined]); michael@0: michael@0: // The arguments object. michael@0: TestGenerator( michael@0: function g12() { michael@0: return (function*(a, b, c, d) { michael@0: for (var i = 0; i < arguments.length; i++) { michael@0: yield arguments[i]; michael@0: } michael@0: })("fee", "fi", "fo", "fum", "I smell the blood of an Englishman"); michael@0: }, michael@0: ["fee", "fi", "fo", "fum", "I smell the blood of an Englishman", michael@0: undefined], michael@0: "foo", michael@0: ["fee", "fi", "fo", "fum", "I smell the blood of an Englishman", michael@0: undefined]); michael@0: michael@0: // Access to captured free variables. michael@0: TestGenerator( michael@0: function g13() { michael@0: return (function(a, b, c, d) { michael@0: return (function*() { michael@0: yield a; yield b; yield c; yield d; michael@0: })(); michael@0: })("fee", "fi", "fo", "fum"); michael@0: }, michael@0: ["fee", "fi", "fo", "fum", undefined], michael@0: "foo", michael@0: ["fee", "fi", "fo", "fum", undefined]); michael@0: michael@0: // Abusing the arguments object. michael@0: TestGenerator( michael@0: function g14() { michael@0: return (function*(a, b, c, d) { michael@0: arguments[0] = "Be he live"; michael@0: arguments[1] = "or be he dead"; michael@0: arguments[2] = "I'll grind his bones"; michael@0: arguments[3] = "to make my bread"; michael@0: yield a; yield b; yield c; yield d; michael@0: })("fee", "fi", "fo", "fum"); michael@0: }, michael@0: ["Be he live", "or be he dead", "I'll grind his bones", "to make my bread", michael@0: undefined], michael@0: "foo", michael@0: ["Be he live", "or be he dead", "I'll grind his bones", "to make my bread", michael@0: undefined]); michael@0: michael@0: // Abusing the arguments object: strict mode. michael@0: TestGenerator( michael@0: function g15() { michael@0: return (function*(a, b, c, d) { michael@0: "use strict"; michael@0: arguments[0] = "Be he live"; michael@0: arguments[1] = "or be he dead"; michael@0: arguments[2] = "I'll grind his bones"; michael@0: arguments[3] = "to make my bread"; michael@0: yield a; yield b; yield c; yield d; michael@0: })("fee", "fi", "fo", "fum"); michael@0: }, michael@0: ["fee", "fi", "fo", "fum", undefined], michael@0: "foo", michael@0: ["fee", "fi", "fo", "fum", undefined]); michael@0: michael@0: // GC. michael@0: if (typeof gc == 'function') { michael@0: TestGenerator(function* g16() { yield "baz"; gc(); yield "qux"; }, michael@0: ["baz", "qux", undefined], michael@0: "foo", michael@0: ["baz", "qux", undefined]); michael@0: } michael@0: michael@0: // Receivers. michael@0: TestGenerator( michael@0: function g17() { michael@0: function* g() { yield this.x; yield this.y; } michael@0: var o = { start: g, x: 1, y: 2 }; michael@0: return o.start(); michael@0: }, michael@0: [1, 2, undefined], michael@0: "foo", michael@0: [1, 2, undefined]); michael@0: michael@0: // FIXME: Capture the generator object as "this" in new g(). Bug 907742. michael@0: // TestGenerator( michael@0: // function g18() { michael@0: // function* g() { yield this.x; yield this.y; } michael@0: // var iter = new g; michael@0: // iter.x = 1; michael@0: // iter.y = 2; michael@0: // return iter; michael@0: // }, michael@0: // [1, 2, undefined], michael@0: // "foo", michael@0: // [1, 2, undefined]); michael@0: michael@0: TestGenerator( michael@0: function* g19() { michael@0: var x = 1; michael@0: yield x; michael@0: with({x:2}) { yield x; } michael@0: yield x; michael@0: }, michael@0: [1, 2, 1, undefined], michael@0: "foo", michael@0: [1, 2, 1, undefined]); michael@0: michael@0: TestGenerator( michael@0: function* g20() { yield (1 + (yield 2) + 3); }, michael@0: [2, NaN, undefined], michael@0: "foo", michael@0: [2, "1foo3", undefined]); michael@0: michael@0: TestGenerator( michael@0: function* g21() { return (1 + (yield 2) + 3); }, michael@0: [2, NaN], michael@0: "foo", michael@0: [2, "1foo3"]); michael@0: michael@0: TestGenerator( michael@0: function* g22() { yield (1 + (yield 2) + 3); yield (4 + (yield 5) + 6); }, michael@0: [2, NaN, 5, NaN, undefined], michael@0: "foo", michael@0: [2, "1foo3", 5, "4foo6", undefined]); michael@0: michael@0: TestGenerator( michael@0: function* g23() { michael@0: return (yield (1 + (yield 2) + 3)) + (yield (4 + (yield 5) + 6)); michael@0: }, michael@0: [2, NaN, 5, NaN, NaN], michael@0: "foo", michael@0: [2, "1foo3", 5, "4foo6", "foofoo"]); michael@0: michael@0: // Rewind a try context with and without operands on the stack. michael@0: TestGenerator( michael@0: function* g24() { michael@0: try { michael@0: return (yield (1 + (yield 2) + 3)) + (yield (4 + (yield 5) + 6)); michael@0: } catch (e) { michael@0: throw e; michael@0: } michael@0: }, michael@0: [2, NaN, 5, NaN, NaN], michael@0: "foo", michael@0: [2, "1foo3", 5, "4foo6", "foofoo"]); michael@0: michael@0: // Yielding in a catch context, with and without operands on the stack. michael@0: TestGenerator( michael@0: function* g25() { michael@0: try { michael@0: throw (yield (1 + (yield 2) + 3)) michael@0: } catch (e) { michael@0: if (typeof e == 'object') throw e; michael@0: return e + (yield (4 + (yield 5) + 6)); michael@0: } michael@0: }, michael@0: [2, NaN, 5, NaN, NaN], michael@0: "foo", michael@0: [2, "1foo3", 5, "4foo6", "foofoo"]); michael@0: michael@0: // Generator function instances. michael@0: TestGenerator(GeneratorFunction(), michael@0: [undefined], michael@0: "foo", michael@0: [undefined]); michael@0: michael@0: TestGenerator(new GeneratorFunction(), michael@0: [undefined], michael@0: "foo", michael@0: [undefined]); michael@0: michael@0: TestGenerator(GeneratorFunction('yield 1;'), michael@0: [1, undefined], michael@0: "foo", michael@0: [1, undefined]); michael@0: michael@0: TestGenerator( michael@0: function() { return GeneratorFunction('x', 'y', 'yield x + y;')(1, 2) }, michael@0: [3, undefined], michael@0: "foo", michael@0: [3, undefined]); michael@0: michael@0: // Access to this with formal arguments. michael@0: TestGenerator( michael@0: function () { michael@0: return ({ x: 42, g: function* (a) { yield this.x } }).g(0); michael@0: }, michael@0: [42, undefined], michael@0: "foo", michael@0: [42, undefined]); michael@0: michael@0: function TestTryCatch(instantiate) { michael@0: function* g() { yield 1; try { yield 2; } catch (e) { yield e; } yield 3; } michael@0: function Sentinel() {} michael@0: michael@0: function Test1(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: assertIteratorNext(iter, 3); michael@0: assertIteratorDone(iter, undefined); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test1(instantiate(g)); michael@0: michael@0: function Test2(iter) { michael@0: assertThrowsInstanceOf(function() { iter.throw(new Sentinel); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test2(instantiate(g)); michael@0: michael@0: function Test3(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertThrowsInstanceOf(function() { iter.throw(new Sentinel); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test3(instantiate(g)); michael@0: michael@0: function Test4(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: var exn = new Sentinel; michael@0: assertIteratorResult(iter.throw(exn), exn, false); michael@0: assertIteratorNext(iter, 3); michael@0: assertIteratorDone(iter, undefined); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test4(instantiate(g)); michael@0: michael@0: function Test5(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: var exn = new Sentinel; michael@0: assertIteratorResult(iter.throw(exn), exn, false); michael@0: assertIteratorNext(iter, 3); michael@0: assertThrowsInstanceOf(function() { iter.throw(new Sentinel); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: michael@0: } michael@0: Test5(instantiate(g)); michael@0: michael@0: function Test6(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: var exn = new Sentinel; michael@0: assertIteratorResult(iter.throw(exn), exn, false); michael@0: assertThrowsInstanceOf(function() { iter.throw(new Sentinel); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test6(instantiate(g)); michael@0: } michael@0: TestTryCatch(function (g) { return g(); }); michael@0: TestTryCatch(function* (g) { return yield* g(); }); michael@0: michael@0: function TestTryFinally(instantiate) { michael@0: function* g() { yield 1; try { yield 2; } finally { yield 3; } yield 4; } michael@0: function Sentinel() {} michael@0: function Sentinel2() {} michael@0: michael@0: function Test1(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: assertIteratorNext(iter, 3); michael@0: assertIteratorNext(iter, 4); michael@0: assertIteratorDone(iter, undefined); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test1(instantiate(g)); michael@0: michael@0: function Test2(iter) { michael@0: assertThrowsInstanceOf(function() { iter.throw(new Sentinel); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test2(instantiate(g)); michael@0: michael@0: function Test3(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertThrowsInstanceOf(function() { iter.throw(new Sentinel); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test3(instantiate(g)); michael@0: michael@0: function Test4(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: assertIteratorResult(iter.throw(new Sentinel), 3, false); michael@0: assertThrowsInstanceOf(function() { iter.next(); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: michael@0: } michael@0: Test4(instantiate(g)); michael@0: michael@0: function Test5(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: assertIteratorResult(iter.throw(new Sentinel), 3, false); michael@0: assertThrowsInstanceOf(function() { iter.throw(new Sentinel2); }, Sentinel2); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test5(instantiate(g)); michael@0: michael@0: function Test6(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: assertIteratorNext(iter, 3); michael@0: assertThrowsInstanceOf(function() { iter.throw(new Sentinel); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test6(instantiate(g)); michael@0: michael@0: function Test7(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: assertIteratorNext(iter, 3); michael@0: assertIteratorNext(iter, 4); michael@0: assertThrowsInstanceOf(function() { iter.throw(new Sentinel); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test7(instantiate(g)); michael@0: } michael@0: TestTryFinally(function (g) { return g(); }); michael@0: TestTryFinally(function* (g) { return yield* g(); }); michael@0: michael@0: function TestNestedTry(instantiate) { michael@0: function* g() { michael@0: try { michael@0: yield 1; michael@0: try { yield 2; } catch (e) { yield e; } michael@0: yield 3; michael@0: } finally { michael@0: yield 4; michael@0: } michael@0: yield 5; michael@0: } michael@0: function Sentinel() {} michael@0: function Sentinel2() {} michael@0: michael@0: function Test1(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: assertIteratorNext(iter, 3); michael@0: assertIteratorNext(iter, 4); michael@0: assertIteratorNext(iter, 5); michael@0: assertIteratorDone(iter, undefined); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test1(instantiate(g)); michael@0: michael@0: function Test2(iter) { michael@0: assertThrowsInstanceOf(function() { iter.throw(new Sentinel); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test2(instantiate(g)); michael@0: michael@0: function Test3(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorResult(iter.throw(new Sentinel), 4, false); michael@0: assertThrowsInstanceOf(function() { iter.next(); }, Sentinel); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test3(instantiate(g)); michael@0: michael@0: function Test4(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorResult(iter.throw(new Sentinel), 4, false); michael@0: assertThrowsInstanceOf(function() { iter.throw(new Sentinel2); }, Sentinel2); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test4(instantiate(g)); michael@0: michael@0: function Test5(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: var exn = new Sentinel; michael@0: assertIteratorResult(iter.throw(exn), exn, false); michael@0: assertIteratorNext(iter, 3); michael@0: assertIteratorNext(iter, 4); michael@0: assertIteratorNext(iter, 5); michael@0: assertIteratorDone(iter, undefined); michael@0: assertIteratorDone(iter, undefined); michael@0: michael@0: } michael@0: Test5(instantiate(g)); michael@0: michael@0: function Test6(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: var exn = new Sentinel; michael@0: assertIteratorResult(iter.throw(exn), exn, false); michael@0: assertIteratorResult(iter.throw(new Sentinel2), 4, false); michael@0: assertThrowsInstanceOf(function() { iter.next(); }, Sentinel2); michael@0: assertIteratorDone(iter, undefined); michael@0: } michael@0: Test6(instantiate(g)); michael@0: michael@0: function Test7(iter) { michael@0: assertIteratorNext(iter, 1); michael@0: assertIteratorNext(iter, 2); michael@0: var exn = new Sentinel; michael@0: assertIteratorResult(iter.throw(exn), exn, false); michael@0: assertIteratorNext(iter, 3); michael@0: assertIteratorResult(iter.throw(new Sentinel2), 4, false); michael@0: assertThrowsInstanceOf(function() { iter.next(); }, Sentinel2); michael@0: assertIteratorDone(iter, undefined); michael@0: michael@0: } michael@0: Test7(instantiate(g)); michael@0: michael@0: // That's probably enough. michael@0: } michael@0: TestNestedTry(function (g) { return g(); }); michael@0: TestNestedTry(function* (g) { return yield* g(); }); michael@0: michael@0: function TestRecursion() { michael@0: function TestNextRecursion() { michael@0: function* g() { yield iter.next(); } michael@0: var iter = g(); michael@0: return iter.next(); michael@0: } michael@0: function TestSendRecursion() { michael@0: function* g() { yield iter.next(42); } michael@0: var iter = g(); michael@0: return iter.next(); michael@0: } michael@0: function TestThrowRecursion() { michael@0: function* g() { yield iter.throw(1); } michael@0: var iter = g(); michael@0: return iter.next(); michael@0: } michael@0: assertThrowsInstanceOf(TestNextRecursion, TypeError); michael@0: assertThrowsInstanceOf(TestSendRecursion, TypeError); michael@0: assertThrowsInstanceOf(TestThrowRecursion, TypeError); michael@0: } michael@0: TestRecursion(); michael@0: michael@0: if (typeof reportCompare == "function") michael@0: reportCompare(true, true);