1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
|
from cStringIO import StringIO
import py
from rpython.tool.udir import udir
from pypy.interpreter.baseobjspace import OperationError
from pypy.interpreter import main
testcode = """\
def main():
aStr = 'hello world'
print len(aStr)
main()
"""
test__file__code = """
assert __file__ is not None
assert __cached__ is None
print len('hello world')
"""
# On module test we want to ensure that the called module __name__ is
# '__main__' and argv is set as expected.
testmodulecode = """
import sys
if __name__ == '__main__':
aStr = sys.argv[1]
print len(aStr)
"""
testresultoutput = '11\n'
def checkoutput(space, expected_output, f, *args):
w_oldout = space.sys.get('stdout')
capturefn = udir.join('capturefile')
w_capturefile = space.call_method(space.builtin, "open", space.wrap(str(capturefn)), space.wrap("w"))
w_sys = space.sys.getmodule('sys')
space.setattr(w_sys, space.wrap("stdout"), w_capturefile)
try:
f(*(args + (space,)))
finally:
space.setattr(w_sys, space.wrap("stdout"), w_oldout)
space.call_method(w_capturefile, "close")
assert capturefn.read(mode='rU') == expected_output
testfn = udir.join('tmp_hello_world.py')
test__file__fn = udir.join('test__file__.py')
testmodule = 'tmp_hello_module'
testpackage = 'tmp_package'
class TestMain:
def setup_class(cls):
testfn.write(testcode, 'w')
test__file__fn.write(test__file__code, 'w')
udir.join(testmodule + '.py').write(testmodulecode, 'w')
udir.ensure(testpackage, '__init__.py')
udir.join(testpackage, testmodule + '.py').write(testmodulecode, 'w')
space = cls.space
cls.w_oldsyspath = space.appexec([space.wrap(str(udir))], """(udir):
import sys
old = sys.path[:]
sys.path.insert(0, udir)
return old
""")
def teardown_class(cls):
cls.space.appexec([cls.w_oldsyspath], """(old):
import sys
sys.path[:] = old
""")
def test_run_file(self):
checkoutput(self.space, testresultoutput, main.run_file, str(testfn))
def test_run_string(self):
checkoutput(self.space, testresultoutput,
main.run_string, testcode, str(testfn))
def test_eval_string(self):
w_x = main.eval_string('2+2', space=self.space)
assert self.space.eq_w(w_x, self.space.wrap(4))
def test_run_module(self):
checkoutput(self.space, testresultoutput, main.run_module,
testmodule, ['hello world'])
checkoutput(self.space, testresultoutput, main.run_module,
testpackage + '.' + testmodule, ['hello world'])
def test__file__file(self):
checkoutput(self.space, testresultoutput, main.run_file, str(test__file__fn))
|