added changed size while iterating tests

Author: tomasstupka

graalpython/com.oracle.graal.python.test/src/tests/test_dict.py

@@ -853,4 +853,49 @@ def __eq__(self, o):
         raised = True
     assert raised
 
-    assert 'kw' not in d
+    assert 'kw' not in d
+    
+def test_iter_changed_size():
+    def just_iterate(it):
+        for i in it:
+            pass
+
+    def iterate_and_update(it):
+        for i in it:
+            d.update({3:3})
+
+    # dict
+    d = {1:1, 2:2} 
+    it = iter(d)
+    del d[1]
+    assert_raises(RuntimeError, just_iterate, it)
+    
+    d = {1:1, 2:2}
+    assert_raises(RuntimeError, iterate_and_update, d)
+
+    # keys
+    d = {1:1, 2:2}
+    it = iter(d.keys())
+    del d[1]
+    assert_raises(RuntimeError, just_iterate, it)
+
+    d = {1:1}
+    assert_raises(RuntimeError, iterate_and_update, d.keys())
+
+    # values
+    d = {1:1, 2:2} 
+    it = iter(d.values())
+    del d[1]
+    assert_raises(RuntimeError, just_iterate, it)
+
+    d = {1:1}
+    assert_raises(RuntimeError, iterate_and_update, d.values())
+
+    # items
+    d = {1:1, 2:2}
+    it = iter(d.items())
+    del d[1]
+    assert_raises(RuntimeError, just_iterate, it)
+
+    d = {1:1}
+    assert_raises(RuntimeError, iterate_and_update, d.items())

graalpython/com.oracle.graal.python.test/src/tests/test_mappingproxy.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
+# Copyright (c) 2018, 2020, Oracle and/or its affiliates. All rights reserved.
 # DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 #
 # The Universal Permissive License (UPL), Version 1.0
@@ -129,3 +129,16 @@ def test_iter():
     mp_keys = set([k for k in mp])
     assert d.keys() == mp_keys
 
+def test_iter_changed_size():
+    class A:
+        pass
+
+    def foo():
+        pass
+
+    try:
+        for i in A.__dict__:
+            setattr(A, 'foo', foo)
+    except RuntimeError:
+        raised = True
+    assert raised

graalpython/com.oracle.graal.python.test/src/tests/test_set.py

@@ -460,4 +460,21 @@ def __len__(self):
     assert {10, 11}.isdisjoint(NonSetWithIter())
     assert {10, 11, 12, 13, 14}.isdisjoint(NonSetWithIter())
     assert not {1, 2}.isdisjoint(NonSetWithIter())
-    assert not {1, 2, 3, 4, 5}.isdisjoint(NonSetWithIter())
+    assert not {1, 2, 3, 4, 5}.isdisjoint(NonSetWithIter())
+
+def test_iter_changed_size():
+    def just_iterate(it):
+        for i in it:
+            pass
+
+    def iterate_and_update(it):
+        for i in it:
+            s.add(3)
+
+    s = {1, 2}
+    it = iter(s)
+    s.pop()
+    assert_raises(RuntimeError, just_iterate, it)
+
+    s = {1, 2}
+    assert_raises(RuntimeError, iterate_and_update, s)