1 files changed, 178 insertions, 0 deletions
diff --git a/js/src/jit-test/tests/collections/WeakSet-constructor-add.js b/js/src/jit-test/tests/collections/WeakSet-constructor-add.js
new file mode 100644
index 0000000000..b883e074f1
--- /dev/null
+++ b/js/src/jit-test/tests/collections/WeakSet-constructor-add.js
@@ -0,0 +1,178 @@
+load(libdir + "asserts.js");
+load(libdir + "iteration.js");
+
+var k1 = {};
+var k2 = {};
+var k3 = {};
+var k4 = {};
+
+function test_patched() {
+  let orig = WeakSet.prototype.add;
+
+  // If adder is modified, constructor should call it.
+  var called = false;
+
+  WeakSet.prototype.add = function(k, v) {
+    assertEq(k, k1);
+    orig.call(this, k2);
+    called = true;
+  };
+
+  var arr = [k1];
+
+  var m = new WeakSet(arr);
+
+  assertEq(called, true);
+  assertEq(m.has(k1), false);
+  assertEq(m.has(k2), true);
+
+  WeakSet.prototype.add = orig;
+}
+
+function test_proxy1() {
+  let orig = WeakSet.prototype.add;
+
+  // If adder is modified, constructor should call it.
+  var called = false;
+
+  WeakSet.prototype.add = new Proxy(function(k, v) {
+    assertEq(k, k1);
+    orig.call(this, k2);
+    called = true;
+  }, {});
+
+  var arr = [k1];
+
+  var m = new WeakSet(arr);
+
+  assertEq(called, true);
+  assertEq(m.has(k1), false);
+  assertEq(m.has(k2), true);
+
+  WeakSet.prototype.add = orig;
+}
+
+function test_proxy2() {
+  let orig = WeakSet.prototype.add;
+
+  // If adder is modified, constructor should call it.
+  var called = false;
+
+  WeakSet.prototype.add = new Proxy(function() {
+  }, {
+    apply: function(target, that, args) {
+      var [k, v] = args;
+      assertEq(k, k1);
+      orig.call(that, k2);
+      called = true;
+    }
+  });
+
+  var arr = [k1];
+
+  var m = new WeakSet(arr);
+
+  assertEq(called, true);
+  assertEq(m.has(k1), false);
+  assertEq(m.has(k2), true);
+
+  WeakSet.prototype.add = orig;
+}
+
+function test_change1() {
+  let orig = WeakSet.prototype.add;
+
+  // Change to adder in GetIterator(..) call should be ignored.
+  var called = false;
+  var modified = false;
+
+  var arr = [k1];
+
+  var proxy_arr = new Proxy(arr, {
+    get: function(target, name) {
+      if (name == Symbol.iterator) {
+        modified = true;
+        WeakSet.prototype.add = function() {
+          called = true;
+        };
+      }
+      return target[name];
+    }
+  });
+
+  var m = new WeakSet(proxy_arr);
+
+  assertEq(modified, true);
+  assertEq(called, false);
+  assertEq(m.has(k1), true);
+  assertEq(m.has(k2), false);
+
+  WeakSet.prototype.add = orig;
+}
+
+function test_change2() {
+  let orig = WeakSet.prototype.add;
+
+  // Change to adder in adder(...) call should be ignored.
+  var called = false;
+  var count = 0;
+
+  WeakSet.prototype.add = function(k, v) {
+    if (count == 0) {
+      assertEq(k, k1);
+      orig.call(this, k3);
+      WeakSet.prototype.add = function() {
+        called = true;
+      };
+      count = 1;
+    } else {
+      assertEq(k, k2);
+      orig.call(this, k4);
+      count = 2;
+    }
+  };
+
+  var arr = [k1, k2];
+
+  var m = new WeakSet(arr);
+
+  assertEq(called, false);
+  assertEq(count, 2);
+  assertEq(m.has(k1), false);
+  assertEq(m.has(k2), false);
+  assertEq(m.has(k3), true);
+  assertEq(m.has(k4), true);
+
+  WeakSet.prototype.add = orig;
+}
+
+function test_error() {
+  let orig = WeakSet.prototype.add;
+
+  var arr = [k1];
+
+  // WeakSet should throw TypeError if adder is not callable.
+  WeakSet.prototype.add = null;
+  assertThrowsInstanceOf(() => new WeakSet(arr), TypeError);
+  WeakSet.prototype.add = {};
+  assertThrowsInstanceOf(() => new WeakSet(arr), TypeError);
+
+  // WeakSet should propagate error thrown by adder.
+  WeakSet.prototype.add = function() {
+    throw SyntaxError();
+  };
+  assertThrowsInstanceOf(() => new WeakSet(arr), SyntaxError);
+
+  WeakSet.prototype.add = orig;
+}
+
+function test() {
+ test_patched();
+ test_proxy1();
+ test_proxy2();
+ test_change1();
+ test_change2();
+ test_error();
+}
+
+test();