def setUp(self): """Set up an interpreter directing output to a BytesIO stream.""" self.interpreter = SafeInterpreter(writer=io.BytesIO())
def setUp(self): """Set up an interpreter directing output to a BytesIO stream.""" self.interpreter = SafeInterpreter(writer=io.BytesIO())
class TestSafeInterpreter(unittest.TestCase):
# AWS tests attribute required for tagging via automatic test discovery via
# nosetests
engineAWSClusterTest = 1
def setUp(self):
"""Set up an interpreter directing output to a BytesIO stream."""
self.interpreter = SafeInterpreter(writer=io.BytesIO())
def testPrimitives(self):
"""Verify basic primitives"""
self.assertTrue(self.interpreter("True"))
self.assertFalse(self.interpreter("False"))
self.assertTrue(self.interpreter("None") is None)
def testConditionals(self):
"""Verify basic if statements"""
self.assertTrue(self.interpreter("True if True else False"))
self.assertTrue(self.interpreter("""
foo = False
if not foo:
foo = True
foo
"""))
def testBlacklist(self):
"""Verify that src with blacklisted nodes fail"""
self.interpreter("for x in []: pass")
self.assertIn("NotImplementedError",
(error.get_error()[0] for error in self.interpreter.error))
self.interpreter("while True: pass")
self.assertIn("NotImplementedError",
(error.get_error()[0] for error in self.interpreter.error))
def testParse(self):
"""Verify that parse() returns an AST instance"""
tree = self.interpreter.parse("True")
self.assertTrue(isinstance(tree, ast.AST))
def testCompile(self):
"""Verify that parse() returns a compile()-able AST"""
tree = self.interpreter.parse("True")
codeObj = compile(tree, "<string>", mode="exec")
self.assertTrue(isinstance(codeObj, types.CodeType))
def testSum(self):
"""Verify that sum() works and is correct"""
result = self.interpreter("sum([x*p for x,p in {1:2}.items()])")
self.assertEqual(result, 2)
def testRecursive(self):
"""Verify that a recursive function raises a runtime error"""
self.interpreter("""
def foo():
foo()
foo()
""")
self.assertIn("RuntimeError",
(error.get_error()[0] for error in self.interpreter.error))
def testOpen(self):
"""Verify that an attempt to open a file raises a runtime error"""
self.interpreter("open('foo')")
self.assertIn("RuntimeError",
(error.get_error()[0] for error in self.interpreter.error))
class TestSafeInterpreter(unittest.TestCase):
# AWS tests attribute required for tagging via automatic test discovery via
# nosetests
engineAWSClusterTest = 1
def setUp(self):
"""Set up an interpreter directing output to a BytesIO stream."""
self.interpreter = SafeInterpreter(writer=io.BytesIO())
def testPrimitives(self):
"""Verify basic primitives"""
self.assertTrue(self.interpreter("True"))
self.assertFalse(self.interpreter("False"))
self.assertTrue(self.interpreter("None") is None)
def testConditionals(self):
"""Verify basic if statements"""
self.assertTrue(self.interpreter("True if True else False"))
self.assertTrue(self.interpreter("""
foo = False
if not foo:
foo = True
foo
"""))
def testBlacklist(self):
"""Verify that src with blacklisted nodes fail"""
self.interpreter("for x in []: pass")
self.assertIn("NotImplementedError",
(error.get_error()[0] for error in self.interpreter.error))
self.interpreter("while True: pass")
self.assertIn("NotImplementedError",
(error.get_error()[0] for error in self.interpreter.error))
def testParse(self):
"""Verify that parse() returns an AST instance"""
tree = self.interpreter.parse("True")
self.assertTrue(isinstance(tree, ast.AST))
def testCompile(self):
"""Verify that parse() returns a compile()-able AST"""
tree = self.interpreter.parse("True")
codeObj = compile(tree, "<string>", mode="exec")
self.assertTrue(isinstance(codeObj, types.CodeType))
def testSum(self):
"""Verify that sum() works and is correct"""
result = self.interpreter("sum([x*p for x,p in {1:2}.items()])")
self.assertEqual(result, 2)
def testRecursive(self):
"""Verify that a recursive function raises a runtime error"""
self.interpreter("""
def foo():
foo()
foo()
""")
self.assertIn("RuntimeError",
(error.get_error()[0] for error in self.interpreter.error))
def testOpen(self):
"""Verify that an attempt to open a file raises a runtime error"""
self.interpreter("open('foo')")
self.assertIn("RuntimeError",
(error.get_error()[0] for error in self.interpreter.error))
