/Windows/Python3.8/WPy64-3830/WPy64-3830/python-3.8.3.amd64/Lib/idlelib/idle_test/test_run.py
Python | 325 lines | 307 code | 17 blank | 1 comment | 6 complexity | 08c6fbb0419148566d0e08bbc3fe989f MD5 | raw file
- "Test run, coverage 42%."
-
- from idlelib import run
- import unittest
- from unittest import mock
- from test.support import captured_stderr
-
- import io
- import sys
-
-
- class RunTest(unittest.TestCase):
-
- def test_print_exception_unhashable(self):
- class UnhashableException(Exception):
- def __eq__(self, other):
- return True
-
- ex1 = UnhashableException('ex1')
- ex2 = UnhashableException('ex2')
- try:
- raise ex2 from ex1
- except UnhashableException:
- try:
- raise ex1
- except UnhashableException:
- with captured_stderr() as output:
- with mock.patch.object(run,
- 'cleanup_traceback') as ct:
- ct.side_effect = lambda t, e: t
- run.print_exception()
-
- tb = output.getvalue().strip().splitlines()
- self.assertEqual(11, len(tb))
- self.assertIn('UnhashableException: ex2', tb[3])
- self.assertIn('UnhashableException: ex1', tb[10])
-
-
- # StdioFile tests.
-
- class S(str):
- def __str__(self):
- return '%s:str' % type(self).__name__
- def __unicode__(self):
- return '%s:unicode' % type(self).__name__
- def __len__(self):
- return 3
- def __iter__(self):
- return iter('abc')
- def __getitem__(self, *args):
- return '%s:item' % type(self).__name__
- def __getslice__(self, *args):
- return '%s:slice' % type(self).__name__
-
-
- class MockShell:
- def __init__(self):
- self.reset()
- def write(self, *args):
- self.written.append(args)
- def readline(self):
- return self.lines.pop()
- def close(self):
- pass
- def reset(self):
- self.written = []
- def push(self, lines):
- self.lines = list(lines)[::-1]
-
-
- class StdInputFilesTest(unittest.TestCase):
-
- def test_misc(self):
- shell = MockShell()
- f = run.StdInputFile(shell, 'stdin')
- self.assertIsInstance(f, io.TextIOBase)
- self.assertEqual(f.encoding, 'utf-8')
- self.assertEqual(f.errors, 'strict')
- self.assertIsNone(f.newlines)
- self.assertEqual(f.name, '<stdin>')
- self.assertFalse(f.closed)
- self.assertTrue(f.isatty())
- self.assertTrue(f.readable())
- self.assertFalse(f.writable())
- self.assertFalse(f.seekable())
-
- def test_unsupported(self):
- shell = MockShell()
- f = run.StdInputFile(shell, 'stdin')
- self.assertRaises(OSError, f.fileno)
- self.assertRaises(OSError, f.tell)
- self.assertRaises(OSError, f.seek, 0)
- self.assertRaises(OSError, f.write, 'x')
- self.assertRaises(OSError, f.writelines, ['x'])
-
- def test_read(self):
- shell = MockShell()
- f = run.StdInputFile(shell, 'stdin')
- shell.push(['one\n', 'two\n', ''])
- self.assertEqual(f.read(), 'one\ntwo\n')
- shell.push(['one\n', 'two\n', ''])
- self.assertEqual(f.read(-1), 'one\ntwo\n')
- shell.push(['one\n', 'two\n', ''])
- self.assertEqual(f.read(None), 'one\ntwo\n')
- shell.push(['one\n', 'two\n', 'three\n', ''])
- self.assertEqual(f.read(2), 'on')
- self.assertEqual(f.read(3), 'e\nt')
- self.assertEqual(f.read(10), 'wo\nthree\n')
-
- shell.push(['one\n', 'two\n'])
- self.assertEqual(f.read(0), '')
- self.assertRaises(TypeError, f.read, 1.5)
- self.assertRaises(TypeError, f.read, '1')
- self.assertRaises(TypeError, f.read, 1, 1)
-
- def test_readline(self):
- shell = MockShell()
- f = run.StdInputFile(shell, 'stdin')
- shell.push(['one\n', 'two\n', 'three\n', 'four\n'])
- self.assertEqual(f.readline(), 'one\n')
- self.assertEqual(f.readline(-1), 'two\n')
- self.assertEqual(f.readline(None), 'three\n')
- shell.push(['one\ntwo\n'])
- self.assertEqual(f.readline(), 'one\n')
- self.assertEqual(f.readline(), 'two\n')
- shell.push(['one', 'two', 'three'])
- self.assertEqual(f.readline(), 'one')
- self.assertEqual(f.readline(), 'two')
- shell.push(['one\n', 'two\n', 'three\n'])
- self.assertEqual(f.readline(2), 'on')
- self.assertEqual(f.readline(1), 'e')
- self.assertEqual(f.readline(1), '\n')
- self.assertEqual(f.readline(10), 'two\n')
-
- shell.push(['one\n', 'two\n'])
- self.assertEqual(f.readline(0), '')
- self.assertRaises(TypeError, f.readlines, 1.5)
- self.assertRaises(TypeError, f.readlines, '1')
- self.assertRaises(TypeError, f.readlines, 1, 1)
-
- def test_readlines(self):
- shell = MockShell()
- f = run.StdInputFile(shell, 'stdin')
- shell.push(['one\n', 'two\n', ''])
- self.assertEqual(f.readlines(), ['one\n', 'two\n'])
- shell.push(['one\n', 'two\n', ''])
- self.assertEqual(f.readlines(-1), ['one\n', 'two\n'])
- shell.push(['one\n', 'two\n', ''])
- self.assertEqual(f.readlines(None), ['one\n', 'two\n'])
- shell.push(['one\n', 'two\n', ''])
- self.assertEqual(f.readlines(0), ['one\n', 'two\n'])
- shell.push(['one\n', 'two\n', ''])
- self.assertEqual(f.readlines(3), ['one\n'])
- shell.push(['one\n', 'two\n', ''])
- self.assertEqual(f.readlines(4), ['one\n', 'two\n'])
-
- shell.push(['one\n', 'two\n', ''])
- self.assertRaises(TypeError, f.readlines, 1.5)
- self.assertRaises(TypeError, f.readlines, '1')
- self.assertRaises(TypeError, f.readlines, 1, 1)
-
- def test_close(self):
- shell = MockShell()
- f = run.StdInputFile(shell, 'stdin')
- shell.push(['one\n', 'two\n', ''])
- self.assertFalse(f.closed)
- self.assertEqual(f.readline(), 'one\n')
- f.close()
- self.assertFalse(f.closed)
- self.assertEqual(f.readline(), 'two\n')
- self.assertRaises(TypeError, f.close, 1)
-
-
- class StdOutputFilesTest(unittest.TestCase):
-
- def test_misc(self):
- shell = MockShell()
- f = run.StdOutputFile(shell, 'stdout')
- self.assertIsInstance(f, io.TextIOBase)
- self.assertEqual(f.encoding, 'utf-8')
- self.assertEqual(f.errors, 'strict')
- self.assertIsNone(f.newlines)
- self.assertEqual(f.name, '<stdout>')
- self.assertFalse(f.closed)
- self.assertTrue(f.isatty())
- self.assertFalse(f.readable())
- self.assertTrue(f.writable())
- self.assertFalse(f.seekable())
-
- def test_unsupported(self):
- shell = MockShell()
- f = run.StdOutputFile(shell, 'stdout')
- self.assertRaises(OSError, f.fileno)
- self.assertRaises(OSError, f.tell)
- self.assertRaises(OSError, f.seek, 0)
- self.assertRaises(OSError, f.read, 0)
- self.assertRaises(OSError, f.readline, 0)
-
- def test_write(self):
- shell = MockShell()
- f = run.StdOutputFile(shell, 'stdout')
- f.write('test')
- self.assertEqual(shell.written, [('test', 'stdout')])
- shell.reset()
- f.write('t\xe8\u015b\U0001d599')
- self.assertEqual(shell.written, [('t\xe8\u015b\U0001d599', 'stdout')])
- shell.reset()
-
- f.write(S('t\xe8\u015b\U0001d599'))
- self.assertEqual(shell.written, [('t\xe8\u015b\U0001d599', 'stdout')])
- self.assertEqual(type(shell.written[0][0]), str)
- shell.reset()
-
- self.assertRaises(TypeError, f.write)
- self.assertEqual(shell.written, [])
- self.assertRaises(TypeError, f.write, b'test')
- self.assertRaises(TypeError, f.write, 123)
- self.assertEqual(shell.written, [])
- self.assertRaises(TypeError, f.write, 'test', 'spam')
- self.assertEqual(shell.written, [])
-
- def test_write_stderr_nonencodable(self):
- shell = MockShell()
- f = run.StdOutputFile(shell, 'stderr', 'iso-8859-15', 'backslashreplace')
- f.write('t\xe8\u015b\U0001d599\xa4')
- self.assertEqual(shell.written, [('t\xe8\\u015b\\U0001d599\\xa4', 'stderr')])
- shell.reset()
-
- f.write(S('t\xe8\u015b\U0001d599\xa4'))
- self.assertEqual(shell.written, [('t\xe8\\u015b\\U0001d599\\xa4', 'stderr')])
- self.assertEqual(type(shell.written[0][0]), str)
- shell.reset()
-
- self.assertRaises(TypeError, f.write)
- self.assertEqual(shell.written, [])
- self.assertRaises(TypeError, f.write, b'test')
- self.assertRaises(TypeError, f.write, 123)
- self.assertEqual(shell.written, [])
- self.assertRaises(TypeError, f.write, 'test', 'spam')
- self.assertEqual(shell.written, [])
-
- def test_writelines(self):
- shell = MockShell()
- f = run.StdOutputFile(shell, 'stdout')
- f.writelines([])
- self.assertEqual(shell.written, [])
- shell.reset()
- f.writelines(['one\n', 'two'])
- self.assertEqual(shell.written,
- [('one\n', 'stdout'), ('two', 'stdout')])
- shell.reset()
- f.writelines(['on\xe8\n', 'tw\xf2'])
- self.assertEqual(shell.written,
- [('on\xe8\n', 'stdout'), ('tw\xf2', 'stdout')])
- shell.reset()
-
- f.writelines([S('t\xe8st')])
- self.assertEqual(shell.written, [('t\xe8st', 'stdout')])
- self.assertEqual(type(shell.written[0][0]), str)
- shell.reset()
-
- self.assertRaises(TypeError, f.writelines)
- self.assertEqual(shell.written, [])
- self.assertRaises(TypeError, f.writelines, 123)
- self.assertEqual(shell.written, [])
- self.assertRaises(TypeError, f.writelines, [b'test'])
- self.assertRaises(TypeError, f.writelines, [123])
- self.assertEqual(shell.written, [])
- self.assertRaises(TypeError, f.writelines, [], [])
- self.assertEqual(shell.written, [])
-
- def test_close(self):
- shell = MockShell()
- f = run.StdOutputFile(shell, 'stdout')
- self.assertFalse(f.closed)
- f.write('test')
- f.close()
- self.assertTrue(f.closed)
- self.assertRaises(ValueError, f.write, 'x')
- self.assertEqual(shell.written, [('test', 'stdout')])
- f.close()
- self.assertRaises(TypeError, f.close, 1)
-
-
- class TestSysRecursionLimitWrappers(unittest.TestCase):
-
- def test_bad_setrecursionlimit_calls(self):
- run.install_recursionlimit_wrappers()
- self.addCleanup(run.uninstall_recursionlimit_wrappers)
- f = sys.setrecursionlimit
- self.assertRaises(TypeError, f, limit=100)
- self.assertRaises(TypeError, f, 100, 1000)
- self.assertRaises(ValueError, f, 0)
-
- def test_roundtrip(self):
- run.install_recursionlimit_wrappers()
- self.addCleanup(run.uninstall_recursionlimit_wrappers)
-
- # check that setting the recursion limit works
- orig_reclimit = sys.getrecursionlimit()
- self.addCleanup(sys.setrecursionlimit, orig_reclimit)
- sys.setrecursionlimit(orig_reclimit + 3)
-
- # check that the new limit is returned by sys.getrecursionlimit()
- new_reclimit = sys.getrecursionlimit()
- self.assertEqual(new_reclimit, orig_reclimit + 3)
-
- def test_default_recursion_limit_preserved(self):
- orig_reclimit = sys.getrecursionlimit()
- run.install_recursionlimit_wrappers()
- self.addCleanup(run.uninstall_recursionlimit_wrappers)
- new_reclimit = sys.getrecursionlimit()
- self.assertEqual(new_reclimit, orig_reclimit)
-
- def test_fixdoc(self):
- def func(): "docstring"
- run.fixdoc(func, "more")
- self.assertEqual(func.__doc__, "docstring\n\nmore")
- func.__doc__ = None
- run.fixdoc(func, "more")
- self.assertEqual(func.__doc__, "more")
-
-
- if __name__ == '__main__':
- unittest.main(verbosity=2)