restricted: allow itertools.next

product/ERP5Type/patches/Restricted.py

@@ -328,3 +328,7 @@ for member_id in dir(decimal):
   if isinstance(member, type) and issubclass(member, decimal.DecimalException):
     ContainerAssertions[member] = 1
 del member_id, member
+
+allow_module('itertools')
+import itertools
+ContainerAssertions[itertools.count] = 1

product/ERP5Type/tests/testRestrictedPythonSecurity.py

@@ -123,3 +123,12 @@ class TestRestrictedPythonSecurity(ERP5TypeTestCase):
         'return urlparse.parse_qsl("q=s")',
         expected=[('q', 's')]
     )
+
+  def test_itertools(self):
+    self.createAndRunScript(
+        'import itertools',
+        'counter = itertools.count()',
+         'return counter.next(), counter.next()',
+        expected=(0, 1),
+    )
+