"Test pyshell, coverage 12%." # Plus coverage of test_warning. Was 20% with test_openshell. from idlelib import pyshell import os import sys import unittest from unittest import mock from test.support import os_helper, requires from test.support.script_helper import assert_python_ok from tkinter import Tk class FunctionTest(unittest.TestCase): # Test stand-alone module level non-gui functions. def test_restart_line_wide(self): eq = self.assertEqual for file, mul, extra in (('', 22, ''), ('finame', 21, '=')): width = 60 bar = mul * '=' with self.subTest(file=file, bar=bar): file = file or 'Shell' line = pyshell.restart_line(width, file) eq(len(line), width) eq(line, f"{bar+extra} RESTART: {file} {bar}") def test_restart_line_narrow(self): expect, taglen = "= RESTART: Shell", 16 for width in (taglen-1, taglen, taglen+1): with self.subTest(width=width): self.assertEqual(pyshell.restart_line(width, ''), expect) self.assertEqual(pyshell.restart_line(taglen+2, ''), expect+' =') def test_fix_user_path(self): # gh-134300: the idlelib directory is removed, other entries kept. eq = self.assertEqual idlelib_dir = os.path.dirname(os.path.abspath(pyshell.__file__)) eq(pyshell.fix_user_path(['', '/a', idlelib_dir, '/b']), ['', '/a', '/b']) eq(pyshell.fix_user_path(['/a', '/b']), ['/a', '/b']) eq(pyshell.fix_user_path([idlelib_dir]), []) def test_shadowed_stdlib(self): # gh-70331: user files in the current directory must not shadow # the stdlib modules imported by IDLE. with os_helper.temp_dir() as cwd: for name in ('os', 'random', 'tkinter'): os_helper.create_empty_file(os.path.join(cwd, f'{name}.py')) for module in 'idlelib', 'idlelib.idle', 'idlelib.pyshell': with self.subTest(module=module): assert_python_ok('-m', module, '-h', __isolated=False, __cwd=cwd) def test_build_subprocess_arglist(self): interp = mock.Mock(port=1234) args = pyshell.ModifiedInterpreter.build_subprocess_arglist(interp) # gh-70331: -P keeps the current directory out of sys.path. self.assertEqual(args[:2], [sys.executable, '-P']) self.assertEqual(args[-1], '1234') class PyShellFileListTest(unittest.TestCase): @classmethod def setUpClass(cls): requires('gui') cls.root = Tk() cls.root.withdraw() @classmethod def tearDownClass(cls): #cls.root.update_idletasks() ## for id in cls.root.after_info(): ## cls.root.after_cancel(id) # Need for EditorWindow. cls.root.destroy() del cls.root def test_init(self): psfl = pyshell.PyShellFileList(self.root) self.assertEqual(psfl.EditorWindow, pyshell.PyShellEditorWindow) self.assertIsNone(psfl.pyshell) # The following sometimes causes 'invalid command name "109734456recolorize"'. # Uncommenting after_cancel above prevents this, but results in # TclError: bad window path name ".!listedtoplevel.!frame.text" # which is normally prevented by after_cancel. ## def test_openshell(self): ## pyshell.use_subprocess = False ## ps = pyshell.PyShellFileList(self.root).open_shell() ## self.assertIsInstance(ps, pyshell.PyShell) class PyShellTest(unittest.TestCase): @classmethod def setUpClass(cls): requires('gui') cls.root = Tk() cls.root.withdraw() cls.shell = pyshell.PyShell(pyshell.PyShellFileList(cls.root)) @classmethod def tearDownClass(cls): cls.shell.close() del cls.shell cls.root.destroy() del cls.root def setUp(self): text = self.shell.text self.shell.per.bottom.delete('1.0', 'end') text.mark_set('iomark', '1.0') # As after begin(). text.mark_gravity('iomark', 'left') self.shell.undo.reset_undo() def test_output_at_prompt(self): # gh-75512: output from a thread while a prompt is shown goes # before the prompt, which stays with the input in progress. shell = self.shell text = shell.text shell.write('x\n', 'stdout') shell.resetoutput() text.tag_add('console', 'iomark-1c') # As showprompt() does. text.insert('end-1c', 'a = (') shell.write('hel', 'stdout') shell.write('lo\n', 'stdout') self.assertEqual(text.get('iomark-6c', 'end-1c'), 'hello\na = (') self.assertIn('console', text.tag_names('iomark-1c')) self.assertNotIn('console', text.tag_names('iomark-7c')) self.assertEqual(shell.shell_sidebar.line_prompts, {3: '>>>'}) class InputStatementlTest(unittest.TestCase): # Test handling of response to input statements in user code. @classmethod def setUpClass(cls): requires('gui') cls.root = Tk() cls.root.withdraw() cls.shell = pyshell.PyShell(pyshell.PyShellFileList(cls.root)) # As after begin(). cls.shell.text.mark_set('iomark', 'insert') cls.shell.text.mark_gravity('iomark', 'left') @classmethod def tearDownClass(cls): cls.shell.close() del cls.shell cls.root.destroy() del cls.root def test_input_not_colorized(self): # gh-64007: input for input() is not colorized, unlike code. shell = self.shell text = shell.text color = shell.color shell.resetoutput() color.reading = True text.insert('end-1c', 'for x in y') color.recolorize_main() self.assertEqual(text.tag_ranges('KEYWORD'), ()) color.reading = False color.notify_range('iomark', 'end') color.recolorize_main() self.assertEqual(len(text.tag_ranges('KEYWORD')), 4) class PyShellRemoveLastNewlineAndSurroundingWhitespaceTest(unittest.TestCase): regexp = pyshell.PyShell._last_newline_re def all_removed(self, text): self.assertEqual('', self.regexp.sub('', text)) def none_removed(self, text): self.assertEqual(text, self.regexp.sub('', text)) def check_result(self, text, expected): self.assertEqual(expected, self.regexp.sub('', text)) def test_empty(self): self.all_removed('') def test_newline(self): self.all_removed('\n') def test_whitespace_no_newline(self): self.all_removed(' ') self.all_removed(' ') self.all_removed(' ') self.all_removed(' ' * 20) self.all_removed('\t') self.all_removed('\t\t') self.all_removed('\t\t\t') self.all_removed('\t' * 20) self.all_removed('\t ') self.all_removed(' \t') self.all_removed(' \t \t ') self.all_removed('\t \t \t') def test_newline_with_whitespace(self): self.all_removed(' \n') self.all_removed('\t\n') self.all_removed(' \t\n') self.all_removed('\t \n') self.all_removed('\n ') self.all_removed('\n\t') self.all_removed('\n \t') self.all_removed('\n\t ') self.all_removed(' \n ') self.all_removed('\t\n ') self.all_removed(' \n\t') self.all_removed('\t\n\t') self.all_removed('\t \t \t\n') self.all_removed(' \t \t \n') self.all_removed('\n\t \t \t') self.all_removed('\n \t \t ') def test_multiple_newlines(self): self.check_result('\n\n', '\n') self.check_result('\n' * 5, '\n' * 4) self.check_result('\n' * 5 + '\t', '\n' * 4) self.check_result('\n' * 20, '\n' * 19) self.check_result('\n' * 20 + ' ', '\n' * 19) self.check_result(' \n \n ', ' \n') self.check_result(' \n\n ', ' \n') self.check_result(' \n\n', ' \n') self.check_result('\t\n\n', '\t\n') self.check_result('\n\n ', '\n') self.check_result('\n\n\t', '\n') self.check_result(' \n \n ', ' \n') self.check_result('\t\n\t\n\t', '\t\n') def test_non_whitespace(self): self.none_removed('a') self.check_result('a\n', 'a') self.check_result('a\n ', 'a') self.check_result('a \n ', 'a') self.check_result('a \n\t', 'a') self.none_removed('-') self.check_result('-\n', '-') self.none_removed('.') self.check_result('.\n', '.') def test_unsupported_whitespace(self): self.none_removed('\v') self.none_removed('\n\v') self.check_result('\v\n', '\v') self.none_removed(' \n\v') self.check_result('\v\n ', '\v') if __name__ == '__main__': unittest.main(verbosity=2)