Commit abbb86a0 authored by Jérome Perrin's avatar Jérome Perrin

restricted: allow itertools.next

parent ea32ff15
......@@ -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
......@@ -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),
)
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment