michael@0: load(libdir + "asserts.js");
michael@0: load(libdir + "eqArrayHelper.js");
michael@0: load(libdir + "iteration.js");
michael@0: 
michael@0: let makeCall    = farg => Function("f", "arg", "return f(" + farg + ");");
michael@0: let makeFunCall = farg => Function("f", "arg", "return f.call(null, " + farg + ");");
michael@0: let makeNew     = farg => Function("f", "arg", "return new f(" + farg + ").value;");
michael@0: 
michael@0: function checkCommon(f, makeFn) {
michael@0:   assertEqArray(makeFn("...[1, 2, 3]")(f), [1, 2, 3]);
michael@0:   assertEqArray(makeFn("1, ...[2], 3")(f), [1, 2, 3]);
michael@0:   assertEqArray(makeFn("1, ...[2], ...[3]")(f), [1, 2, 3]);
michael@0:   assertEqArray(makeFn("1, ...[2, 3]")(f), [1, 2, 3]);
michael@0:   assertEqArray(makeFn("1, ...[], 2, 3")(f), [1, 2, 3]);
michael@0: 
michael@0:   // other iterable objects
michael@0:   assertEqArray(makeFn("...arg")(f, new Int32Array([1, 2, 3])), [1, 2, 3]);
michael@0:   assertEqArray(makeFn("...arg")(f, "abc"), ["a", "b", "c"]);
michael@0:   assertEqArray(makeFn("...arg")(f, [1, 2, 3][std_iterator]()), [1, 2, 3]);
michael@0:   assertEqArray(makeFn("...arg")(f, Set([1, 2, 3])), [1, 2, 3]);
michael@0:   assertEqArray(makeFn("...arg")(f, Map([["a", "A"], ["b", "B"], ["c", "C"]])).map(([k, v]) => k + v), ["aA", "bB", "cC"]);
michael@0:   let itr = {};
michael@0:   itr[std_iterator] = function() {
michael@0:       return {
michael@0:           i: 1,
michael@0:           next: function() {
michael@0:               if (this.i < 4)
michael@0:                   return { value: this.i++, done: false };
michael@0:               else
michael@0:                   return { value: undefined, done: true };
michael@0:           }
michael@0:       };
michael@0:   };
michael@0:   assertEqArray(makeFn("...arg")(f, itr), [1, 2, 3]);
michael@0:   function gen() {
michael@0:       for (let i = 1; i < 4; i ++)
michael@0:           yield i;
michael@0:   }
michael@0:   assertEqArray(makeFn("...arg")(f, gen()), [1, 2, 3]);
michael@0: 
michael@0:   assertEqArray(makeFn("...arg=[1, 2, 3]")(f), [1, 2, 3]);
michael@0: 
michael@0:   // According to the draft spec, null and undefined are to be treated as empty
michael@0:   // arrays. However, they are not iterable. If the spec is not changed to be in
michael@0:   // terms of iterables, these tests should be fixed.
michael@0:   //assertEqArray(makeFn(1, ...null, 2, 3)(f), [1, 2, 3]);
michael@0:   //assertEqArray(makeFn(1, ...undefined, 2, 3)(f), [1, 2, 3]);
michael@0:   assertThrowsInstanceOf(makeFn("1, ...null, 2, 3"), TypeError);
michael@0:   assertThrowsInstanceOf(makeFn("1, ...undefined, 2, 3"), TypeError);
michael@0: }
michael@0: 
michael@0: function checkNormal(f, makeFn) {
michael@0:   checkCommon(f, makeFn);
michael@0: 
michael@0:   assertEqArray(makeFn("...[]")(f), [undefined, undefined, undefined]);
michael@0:   assertEqArray(makeFn("...[1]")(f), [1, undefined, undefined]);
michael@0:   assertEqArray(makeFn("...[1, 2]")(f), [1, 2, undefined]);
michael@0:   assertEqArray(makeFn("...[1, 2, 3, 4]")(f), [1, 2, 3]);
michael@0: 
michael@0:   assertEqArray(makeFn("...[undefined]")(f), [undefined, undefined, undefined]);
michael@0: }
michael@0: 
michael@0: checkNormal(function(a, b, c) [a, b, c], makeCall);
michael@0: checkNormal(function(a, b, c) [a, b, c], makeFunCall);
michael@0: checkNormal((a, b, c) => [a, b, c], makeCall);
michael@0: checkNormal((a, b, c) => [a, b, c], makeFunCall);
michael@0: function normalClass(a, b, c) {
michael@0:   this.value = [a, b, c];
michael@0:   assertEq(Object.getPrototypeOf(this), normalClass.prototype);
michael@0: }
michael@0: checkNormal(normalClass, makeNew);
michael@0: 
michael@0: function checkDefault(f, makeFn) {
michael@0:   checkCommon(f, makeFn);
michael@0: 
michael@0:   assertEqArray(makeFn("...[]")(f), [-1, -2, -3]);
michael@0:   assertEqArray(makeFn("...[1]")(f), [1, -2, -3]);
michael@0:   assertEqArray(makeFn("...[1, 2]")(f), [1, 2, -3]);
michael@0:   assertEqArray(makeFn("...[1, 2, 3, 4]")(f), [1, 2, 3]);
michael@0: 
michael@0:   assertEqArray(makeFn("...[undefined]")(f), [-1, -2, -3]);
michael@0: }
michael@0: 
michael@0: checkDefault(function(a = -1, b = -2, c = -3) [a, b, c], makeCall);
michael@0: checkDefault(function(a = -1, b = -2, c = -3) [a, b, c], makeFunCall);
michael@0: checkDefault((a = -1, b = -2, c = -3) => [a, b, c], makeCall);
michael@0: checkDefault((a = -1, b = -2, c = -3) => [a, b, c], makeFunCall);
michael@0: function defaultClass(a = -1, b = -2, c = -3) {
michael@0:   this.value = [a, b, c];
michael@0:   assertEq(Object.getPrototypeOf(this), defaultClass.prototype);
michael@0: }
michael@0: checkDefault(defaultClass, makeNew);
michael@0: 
michael@0: function checkRest(f, makeFn) {
michael@0:   checkCommon(f, makeFn);
michael@0: 
michael@0:   assertEqArray(makeFn("...[]")(f), []);
michael@0:   assertEqArray(makeFn("1, ...[2, 3, 4], 5")(f), [1, 2, 3, 4, 5]);
michael@0:   assertEqArray(makeFn("1, ...[], 2")(f), [1, 2]);
michael@0:   assertEqArray(makeFn("1, ...[2, 3], 4, ...[5, 6]")(f), [1, 2, 3, 4, 5, 6]);
michael@0: 
michael@0:   assertEqArray(makeFn("...[undefined]")(f), [undefined]);
michael@0: }
michael@0: 
michael@0: checkRest(function(...x) x, makeCall);
michael@0: checkRest(function(...x) x, makeFunCall);
michael@0: checkRest((...x) => x, makeCall);
michael@0: checkRest((...x) => x, makeFunCall);
michael@0: function restClass(...x) {
michael@0:   this.value = x;
michael@0:   assertEq(Object.getPrototypeOf(this), restClass.prototype);
michael@0: }
michael@0: checkRest(restClass, makeNew);