Add iterator closing for ecma_op_container_create (#3457)

JerryScript-DCO-1.0-Signed-off-by: Adam Szilagyi aszilagy@inf.u-szeged.hu

tests/jerry/es2015/map.js

@@ -123,3 +123,27 @@ m = new Map([{0: "foo", 1: 3}, {0 : "bar", 1 : 2}]);
 assert (m.size === 2);
 assert (m.get("foo") === 3);
 assert (m.get("bar") === 2);
+
+function createIterable(arr, methods = {}) {
+  let iterable = function *() {
+    let idx = 0;
+    while (idx < arr.length) {
+      yield arr[idx];
+      idx++;
+    }
+  }();
+  iterable['return'] = methods['return'];
+  iterable['throw'] = methods['throw'];
+
+  return iterable;
+};
+
+var closed = false;
+var iter = createIterable([1, 2, 3], {
+  'return': function(){ closed = true; return {}; }
+});
+try {
+  new Map(iter);
+} catch(e){}
+
+assert(closed === true);

tests/jerry/es2015/set.js

@@ -91,3 +91,30 @@ assert(set.size === 0);
 set.add(3);
 assert(set.delete(3));
 assert(!set.delete(3));
+
+function createIterable(arr, methods = {}) {
+  let iterable = function *() {
+    let idx = 0;
+    while (idx < arr.length) {
+      yield arr[idx];
+      idx++;
+    }
+  }();
+  iterable['return'] = methods['return'];
+  iterable['throw'] = methods['throw'];
+
+  return iterable;
+};
+
+var closed = false;
+var iter = createIterable([1, 2, 3], {
+  'return': function(){ closed = true; return {}; }
+});
+var add = Set.prototype.add;
+Set.prototype.add = function(){ throw 0 };
+try {
+  new Set(iter);
+} catch(e){}
+Set.prototype.add = add;
+
+assert(closed === true);

tests/jerry/es2015/weakmap.js

@@ -135,6 +135,30 @@ try {
   assert (e instanceof TypeError);
 }
 
+function createIterable(arr, methods = {}) {
+  let iterable = function *() {
+    let idx = 0;
+    while (idx < arr.length) {
+      yield arr[idx];
+      idx++;
+    }
+  }();
+  iterable['return'] = methods['return'];
+  iterable['throw'] = methods['throw'];
+
+  return iterable;
+};
+
+var closed = false;
+var iter = createIterable([1, 2, 3], {
+  'return': function(){ closed = true; return {}; }
+});
+try {
+  new WeakMap(iter);
+} catch(e){}
+
+assert(closed === true);
+
 m1.set([], []);
 
 assert (WeakMap.prototype.toString() === "[object WeakMap]");

tests/jerry/es2015/weakset.js

@@ -119,6 +119,30 @@ try {
   assert (e instanceof TypeError);
 }
 
+function createIterable(arr, methods = {}) {
+  let iterable = function *() {
+    let idx = 0;
+    while (idx < arr.length) {
+      yield arr[idx];
+      idx++;
+    }
+  }();
+  iterable['return'] = methods['return'];
+  iterable['throw'] = methods['throw'];
+
+  return iterable;
+};
+
+var closed = false;
+var iter = createIterable([1, 2, 3], {
+  'return': function(){ closed = true; return {}; }
+});
+try {
+  new WeakMap(iter);
+} catch(e){}
+
+assert(closed === true);
+
 m1.add([]);
 
 assert (WeakSet.prototype.toString() === "[object WeakSet]");