started defining print tests

zopefoundation · Sep 22, 2016 · 4a86f50 · 4a86f50
1 parent 0b8181b
commit 4a86f50
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/tests/test_print.py b/tests/test_print.py
@@ -0,0 +1,26 @@
+from RestrictedPython import compile_restricted
+from RestrictedPython import compile_restricted_exec
+import pytest
+import sys
+
+
+
+
+
+ALLOWED_PRINT = """\
+print 'Hello World!'
+"""
+
+ALLOWED_PRINT_WITH_NL = """\
+print 'Hello World!',
+"""
+
+DISSALOWED_PRINT_WITH_CHEVRON = """\
+print >> stream, 'Hello World!'
+"""
+
+
+def test_print__simple_print_statement():
+    code, err, warn, use = compile_restricted_exec(ALLOWED_PRINT, '<undefined>')
+    exec(code)
+    assert 'code' == code