mirror of
https://github.com/python/cpython.git
synced 2025-07-24 11:44:31 +00:00
[3.13] gh-122478: Remove internal frames from tracebacks in REPL (GH-122528) (#123227)
Frames of methods in code and codeop modules was show with non-default
sys.excepthook.
Save correct tracebacks in sys.last_traceback and update __traceback__
attribute of sys.last_value and sys.last_exc.
(cherry picked from commit e73e7a7abd)
Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
This commit is contained in:
CF Bolz-Tereick
GitHub
parent
5271f8fead
commit
30eee22d3f
3 changed files with 162 additions and 52 deletions
|
|
@ -1,5 +1,6 @@
|
|||
"Test InteractiveConsole and InteractiveInterpreter from code module"
|
||||
import sys
|
||||
import traceback
|
||||
import unittest
|
||||
from textwrap import dedent
|
||||
from contextlib import ExitStack
|
||||
|
|
@ -30,6 +31,7 @@ class MockSys:
|
|||
|
||||
|
||||
class TestInteractiveConsole(unittest.TestCase, MockSys):
|
||||
maxDiff = None
|
||||
|
||||
def setUp(self):
|
||||
self.console = code.InteractiveConsole()
|
||||
|
|
@ -61,21 +63,118 @@ class TestInteractiveConsole(unittest.TestCase, MockSys):
|
|||
raise AssertionError("no console stdout")
|
||||
|
||||
def test_syntax_error(self):
|
||||
self.infunc.side_effect = ["undefined", EOFError('Finished')]
|
||||
self.infunc.side_effect = ["def f():",
|
||||
" x = ?",
|
||||
"",
|
||||
EOFError('Finished')]
|
||||
self.console.interact()
|
||||
for call in self.stderr.method_calls:
|
||||
if 'NameError' in ''.join(call[1]):
|
||||
break
|
||||
else:
|
||||
raise AssertionError("No syntax error from console")
|
||||
output = ''.join(''.join(call[1]) for call in self.stderr.method_calls)
|
||||
output = output[output.index('(InteractiveConsole)'):]
|
||||
output = output[:output.index('\nnow exiting')]
|
||||
self.assertEqual(output.splitlines()[1:], [
|
||||
' File "<console>", line 2',
|
||||
' x = ?',
|
||||
' ^',
|
||||
'SyntaxError: invalid syntax'])
|
||||
self.assertIs(self.sysmod.last_type, SyntaxError)
|
||||
self.assertIs(type(self.sysmod.last_value), SyntaxError)
|
||||
self.assertIsNone(self.sysmod.last_traceback)
|
||||
self.assertIsNone(self.sysmod.last_value.__traceback__)
|
||||
self.assertIs(self.sysmod.last_exc, self.sysmod.last_value)
|
||||
|
||||
def test_indentation_error(self):
|
||||
self.infunc.side_effect = [" 1", EOFError('Finished')]
|
||||
self.console.interact()
|
||||
output = ''.join(''.join(call[1]) for call in self.stderr.method_calls)
|
||||
output = output[output.index('(InteractiveConsole)'):]
|
||||
output = output[:output.index('\nnow exiting')]
|
||||
self.assertEqual(output.splitlines()[1:], [
|
||||
' File "<console>", line 1',
|
||||
' 1',
|
||||
'IndentationError: unexpected indent'])
|
||||
self.assertIs(self.sysmod.last_type, IndentationError)
|
||||
self.assertIs(type(self.sysmod.last_value), IndentationError)
|
||||
self.assertIsNone(self.sysmod.last_traceback)
|
||||
self.assertIsNone(self.sysmod.last_value.__traceback__)
|
||||
self.assertIs(self.sysmod.last_exc, self.sysmod.last_value)
|
||||
|
||||
def test_unicode_error(self):
|
||||
self.infunc.side_effect = ["'\ud800'", EOFError('Finished')]
|
||||
self.console.interact()
|
||||
output = ''.join(''.join(call[1]) for call in self.stderr.method_calls)
|
||||
output = output[output.index('(InteractiveConsole)'):]
|
||||
output = output[output.index('\n') + 1:]
|
||||
self.assertTrue(output.startswith('UnicodeEncodeError: '), output)
|
||||
self.assertIs(self.sysmod.last_type, UnicodeEncodeError)
|
||||
self.assertIs(type(self.sysmod.last_value), UnicodeEncodeError)
|
||||
self.assertIsNone(self.sysmod.last_traceback)
|
||||
self.assertIsNone(self.sysmod.last_value.__traceback__)
|
||||
self.assertIs(self.sysmod.last_exc, self.sysmod.last_value)
|
||||
|
||||
def test_sysexcepthook(self):
|
||||
self.infunc.side_effect = ["raise ValueError('')",
|
||||
self.infunc.side_effect = ["def f():",
|
||||
" raise ValueError('BOOM!')",
|
||||
"",
|
||||
"f()",
|
||||
EOFError('Finished')]
|
||||
hook = mock.Mock()
|
||||
self.sysmod.excepthook = hook
|
||||
self.console.interact()
|
||||
self.assertTrue(hook.called)
|
||||
hook.assert_called()
|
||||
hook.assert_called_with(self.sysmod.last_type,
|
||||
self.sysmod.last_value,
|
||||
self.sysmod.last_traceback)
|
||||
self.assertIs(self.sysmod.last_type, ValueError)
|
||||
self.assertIs(type(self.sysmod.last_value), ValueError)
|
||||
self.assertIs(self.sysmod.last_traceback, self.sysmod.last_value.__traceback__)
|
||||
self.assertIs(self.sysmod.last_exc, self.sysmod.last_value)
|
||||
self.assertEqual(traceback.format_exception(self.sysmod.last_exc), [
|
||||
'Traceback (most recent call last):\n',
|
||||
' File "<console>", line 1, in <module>\n',
|
||||
' File "<console>", line 2, in f\n',
|
||||
'ValueError: BOOM!\n'])
|
||||
|
||||
def test_sysexcepthook_syntax_error(self):
|
||||
self.infunc.side_effect = ["def f():",
|
||||
" x = ?",
|
||||
"",
|
||||
EOFError('Finished')]
|
||||
hook = mock.Mock()
|
||||
self.sysmod.excepthook = hook
|
||||
self.console.interact()
|
||||
hook.assert_called()
|
||||
hook.assert_called_with(self.sysmod.last_type,
|
||||
self.sysmod.last_value,
|
||||
self.sysmod.last_traceback)
|
||||
self.assertIs(self.sysmod.last_type, SyntaxError)
|
||||
self.assertIs(type(self.sysmod.last_value), SyntaxError)
|
||||
self.assertIsNone(self.sysmod.last_traceback)
|
||||
self.assertIsNone(self.sysmod.last_value.__traceback__)
|
||||
self.assertIs(self.sysmod.last_exc, self.sysmod.last_value)
|
||||
self.assertEqual(traceback.format_exception(self.sysmod.last_exc), [
|
||||
' File "<console>", line 2\n',
|
||||
' x = ?\n',
|
||||
' ^\n',
|
||||
'SyntaxError: invalid syntax\n'])
|
||||
|
||||
def test_sysexcepthook_indentation_error(self):
|
||||
self.infunc.side_effect = [" 1", EOFError('Finished')]
|
||||
hook = mock.Mock()
|
||||
self.sysmod.excepthook = hook
|
||||
self.console.interact()
|
||||
hook.assert_called()
|
||||
hook.assert_called_with(self.sysmod.last_type,
|
||||
self.sysmod.last_value,
|
||||
self.sysmod.last_traceback)
|
||||
self.assertIs(self.sysmod.last_type, IndentationError)
|
||||
self.assertIs(type(self.sysmod.last_value), IndentationError)
|
||||
self.assertIsNone(self.sysmod.last_traceback)
|
||||
self.assertIsNone(self.sysmod.last_value.__traceback__)
|
||||
self.assertIs(self.sysmod.last_exc, self.sysmod.last_value)
|
||||
self.assertEqual(traceback.format_exception(self.sysmod.last_exc), [
|
||||
' File "<console>", line 1\n',
|
||||
' 1\n',
|
||||
'IndentationError: unexpected indent\n'])
|
||||
|
||||
def test_sysexcepthook_crashing_doesnt_close_repl(self):
|
||||
self.infunc.side_effect = ["1/0", "a = 123", "print(a)", EOFError('Finished')]
|
||||
|
|
@ -167,6 +266,11 @@ class TestInteractiveConsole(unittest.TestCase, MockSys):
|
|||
ValueError
|
||||
""")
|
||||
self.assertIn(expected, output)
|
||||
self.assertIs(self.sysmod.last_type, ValueError)
|
||||
self.assertIs(type(self.sysmod.last_value), ValueError)
|
||||
self.assertIs(self.sysmod.last_traceback, self.sysmod.last_value.__traceback__)
|
||||
self.assertIsNotNone(self.sysmod.last_traceback)
|
||||
self.assertIs(self.sysmod.last_exc, self.sysmod.last_value)
|
||||
|
||||
def test_context_tb(self):
|
||||
self.infunc.side_effect = ["try: ham\nexcept: eggs\n",
|
||||
|
|
@ -185,6 +289,11 @@ class TestInteractiveConsole(unittest.TestCase, MockSys):
|
|||
NameError: name 'eggs' is not defined
|
||||
""")
|
||||
self.assertIn(expected, output)
|
||||
self.assertIs(self.sysmod.last_type, NameError)
|
||||
self.assertIs(type(self.sysmod.last_value), NameError)
|
||||
self.assertIs(self.sysmod.last_traceback, self.sysmod.last_value.__traceback__)
|
||||
self.assertIsNotNone(self.sysmod.last_traceback)
|
||||
self.assertIs(self.sysmod.last_exc, self.sysmod.last_value)
|
||||
|
||||
|
||||
class TestInteractiveConsoleLocalExit(unittest.TestCase, MockSys):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue