diff options
Diffstat (limited to 'python/helpers/pydev/tests_runfiles/test_pydevdio.py')
-rw-r--r-- | python/helpers/pydev/tests_runfiles/test_pydevdio.py | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/python/helpers/pydev/tests_runfiles/test_pydevdio.py b/python/helpers/pydev/tests_runfiles/test_pydevdio.py new file mode 100644 index 000000000000..7a48a63bd6d4 --- /dev/null +++ b/python/helpers/pydev/tests_runfiles/test_pydevdio.py @@ -0,0 +1,40 @@ +import sys +import os + + +import unittest + +class Test(unittest.TestCase): + + def testIt(self): + #make it as if we were executing from the directory above this one (so that we can use jycompletionserver + #without the need for it being in the pythonpath) + #(twice the dirname to get the previous level from this file.) + import test_pydevdio #@UnresolvedImport - importing itself + ADD_TO_PYTHONPATH = os.path.join(os.path.dirname(os.path.dirname(test_pydevdio.__file__))) + sys.path.insert(0, ADD_TO_PYTHONPATH) + + try: + import pydevd_io + original = sys.stdout + + try: + sys.stdout = pydevd_io.IOBuf() + print('foo') + print('bar') + + self.assertEquals('foo\nbar\n', sys.stdout.getvalue()) #@UndefinedVariable + + print('ww') + print('xx') + self.assertEquals('ww\nxx\n', sys.stdout.getvalue()) #@UndefinedVariable + finally: + sys.stdout = original + finally: + #remove it to leave it ok for other tests + sys.path.remove(ADD_TO_PYTHONPATH) + +if __name__ == '__main__': + #this is so that we can run it frem the jython tests -- because we don't actually have an __main__ module + #(so, it won't try importing the __main__ module) + unittest.TextTestRunner().run(unittest.makeSuite(Test)) |