mirror of https://github.com/python/cpython
431 lines
15 KiB
Python
431 lines
15 KiB
Python
"Test run, coverage 54%."
|
|
|
|
from idlelib import run
|
|
import io
|
|
import sys
|
|
from test.support import captured_output, captured_stderr
|
|
import unittest
|
|
from unittest import mock
|
|
import idlelib
|
|
from idlelib.idle_test.mock_idle import Func
|
|
|
|
idlelib.testing = True # Use {} for executing test user code.
|
|
|
|
|
|
class ExceptionTest(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])
|
|
|
|
data = (('1/0', ZeroDivisionError, "division by zero\n"),
|
|
('abc', NameError, "name 'abc' is not defined. "
|
|
"Did you mean: 'abs'? "
|
|
"Or did you forget to import 'abc'?\n"),
|
|
('int.reel', AttributeError,
|
|
"type object 'int' has no attribute 'reel'. "
|
|
"Did you mean: 'real'?\n"),
|
|
)
|
|
|
|
def test_get_message(self):
|
|
for code, exc, msg in self.data:
|
|
with self.subTest(code=code):
|
|
try:
|
|
eval(compile(code, '', 'eval'))
|
|
except exc:
|
|
typ, val, tb = sys.exc_info()
|
|
actual = run.get_message_lines(typ, val, tb)[0]
|
|
expect = f'{exc.__name__}: {msg}'
|
|
self.assertEqual(actual, expect)
|
|
|
|
@mock.patch.object(run, 'cleanup_traceback',
|
|
new_callable=lambda: (lambda t, e: None))
|
|
def test_get_multiple_message(self, mock):
|
|
d = self.data
|
|
data2 = ((d[0], d[1]), (d[1], d[2]), (d[2], d[0]))
|
|
subtests = 0
|
|
for (code1, exc1, msg1), (code2, exc2, msg2) in data2:
|
|
with self.subTest(codes=(code1,code2)):
|
|
try:
|
|
eval(compile(code1, '', 'eval'))
|
|
except exc1:
|
|
try:
|
|
eval(compile(code2, '', 'eval'))
|
|
except exc2:
|
|
with captured_stderr() as output:
|
|
run.print_exception()
|
|
actual = output.getvalue()
|
|
self.assertIn(msg1, actual)
|
|
self.assertIn(msg2, actual)
|
|
subtests += 1
|
|
self.assertEqual(subtests, len(data2)) # All subtests ran?
|
|
|
|
# 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 RecursionLimitTest(unittest.TestCase):
|
|
# Test (un)install_recursionlimit_wrappers and fixdoc.
|
|
|
|
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):
|
|
# Put here until better place for miscellaneous test.
|
|
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")
|
|
|
|
|
|
class HandleErrorTest(unittest.TestCase):
|
|
# Method of MyRPCServer
|
|
def test_fatal_error(self):
|
|
eq = self.assertEqual
|
|
with captured_output('__stderr__') as err,\
|
|
mock.patch('idlelib.run.thread.interrupt_main',
|
|
new_callable=Func) as func:
|
|
try:
|
|
raise EOFError
|
|
except EOFError:
|
|
run.MyRPCServer.handle_error(None, 'abc', '123')
|
|
eq(run.exit_now, True)
|
|
run.exit_now = False
|
|
eq(err.getvalue(), '')
|
|
|
|
try:
|
|
raise IndexError
|
|
except IndexError:
|
|
run.MyRPCServer.handle_error(None, 'abc', '123')
|
|
eq(run.quitting, True)
|
|
run.quitting = False
|
|
msg = err.getvalue()
|
|
self.assertIn('abc', msg)
|
|
self.assertIn('123', msg)
|
|
self.assertIn('IndexError', msg)
|
|
eq(func.called, 2)
|
|
|
|
|
|
class ExecRuncodeTest(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.addClassCleanup(setattr,run,'print_exception',run.print_exception)
|
|
cls.prt = Func() # Need reference.
|
|
run.print_exception = cls.prt
|
|
mockrpc = mock.Mock()
|
|
mockrpc.console.getvar = Func(result=False)
|
|
cls.ex = run.Executive(mockrpc)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
assert sys.excepthook == sys.__excepthook__
|
|
|
|
def test_exceptions(self):
|
|
ex = self.ex
|
|
ex.runcode('1/0')
|
|
self.assertIs(ex.user_exc_info[0], ZeroDivisionError)
|
|
|
|
self.addCleanup(setattr, sys, 'excepthook', sys.__excepthook__)
|
|
sys.excepthook = lambda t, e, tb: run.print_exception(t)
|
|
ex.runcode('1/0')
|
|
self.assertIs(self.prt.args[0], ZeroDivisionError)
|
|
|
|
sys.excepthook = lambda: None
|
|
ex.runcode('1/0')
|
|
t, e, tb = ex.user_exc_info
|
|
self.assertIs(t, TypeError)
|
|
self.assertTrue(isinstance(e.__context__, ZeroDivisionError))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(verbosity=2)
|