← Back
Editing: test_squeezer.py
from collections import namedtuple from tkinter import Text, Tk import unittest from unittest.mock import Mock, NonCallableMagicMock, patch, sentinel, ANY from test.support import requires from idlelib.config import idleConf from idlelib.squeezer import count_lines_with_wrapping, ExpandingButton, \ Squeezer from idlelib import macosx from idlelib.textview import view_text from idlelib.tooltip import Hovertip from idlelib.pyshell import PyShell SENTINEL_VALUE = sentinel.SENTINEL_VALUE def get_test_tk_root(test_instance): """Helper for tests: Create a root Tk object.""" requires('gui') root = Tk() root.withdraw() def cleanup_root(): root.update_idletasks() root.destroy() test_instance.addCleanup(cleanup_root) return root class CountLinesTest(unittest.TestCase): """Tests for the count_lines_with_wrapping function.""" def check(self, expected, text, linewidth, tabwidth): return self.assertEqual( expected, count_lines_with_wrapping(text, linewidth, tabwidth), ) def test_count_empty(self): """Test with an empty string.""" self.assertEqual(count_lines_with_wrapping(""), 0) def test_count_begins_with_empty_line(self): """Test with a string which begins with a newline.""" self.assertEqual(count_lines_with_wrapping("\ntext"), 2) def test_count_ends_with_empty_line(self): """Test with a string which ends with a newline.""" self.assertEqual(count_lines_with_wrapping("text\n"), 1) def test_count_several_lines(self): """Test with several lines of text.""" self.assertEqual(count_lines_with_wrapping("1\n2\n3\n"), 3) def test_tab_width(self): """Test with various tab widths and line widths.""" self.check(expected=1, text='\t' * 1, linewidth=8, tabwidth=4) self.check(expected=1, text='\t' * 2, linewidth=8, tabwidth=4) self.check(expected=2, text='\t' * 3, linewidth=8, tabwidth=4) self.check(expected=2, text='\t' * 4, linewidth=8, tabwidth=4) self.check(expected=3, text='\t' * 5, linewidth=8, tabwidth=4) # test longer lines and various tab widths self.check(expected=4, text='\t' * 10, linewidth=12, tabwidth=4) self.check(expected=10, text='\t' * 10, linewidth=12, tabwidth=8) self.check(expected=2, text='\t' * 4, linewidth=10, tabwidth=3) # test tabwidth=1 self.check(expected=2, text='\t' * 9, linewidth=5, tabwidth=1) self.check(expected=2, text='\t' * 10, linewidth=5, tabwidth=1) self.check(expected=3, text='\t' * 11, linewidth=5, tabwidth=1) # test for off-by-one errors self.check(expected=2, text='\t' * 6, linewidth=12, tabwidth=4) self.check(expected=3, text='\t' * 6, linewidth=11, tabwidth=4) self.check(expected=2, text='\t' * 6, linewidth=13, tabwidth=4) class SqueezerTest(unittest.TestCase): """Tests for the Squeezer class.""" def make_mock_editor_window(self): """Create a mock EditorWindow instance.""" editwin = NonCallableMagicMock() # isinstance(editwin, PyShell) must be true for Squeezer to enable # auto-squeezing; in practice this will always be true editwin.__class__ = PyShell return editwin def make_squeezer_instance(self, editor_window=None): """Create an actual Squeezer instance with a mock EditorWindow.""" if editor_window is None: editor_window = self.make_mock_editor_window() return Squeezer(editor_window) def test_count_lines(self): """Test Squeezer.count_lines() with various inputs. This checks that Squeezer.count_lines() calls the count_lines_with_wrapping() function with the appropriate parameters. """ for tabwidth, linewidth in [(4, 80), (1, 79), (8, 80), (3, 120)]: self._test_count_lines_helper(linewidth=linewidth, tabwidth=tabwidth) def _prepare_mock_editwin_for_count_lines(self, editwin, linewidth, tabwidth): """Prepare a mock EditorWindow object for Squeezer.count_lines.""" CHAR_WIDTH = 10 BORDER_WIDTH = 2 PADDING_WIDTH = 1 # Prepare all the required functionality on the mock EditorWindow object # so that the calculations in Squeezer.count_lines() can run. editwin.get_tk_tabwidth.return_value = tabwidth editwin.text.winfo_width.return_value = \ linewidth * CHAR_WIDTH + 2 * (BORDER_WIDTH + PADDING_WIDTH) text_opts = { 'border': BORDER_WIDTH, 'padx': PADDING_WIDTH, 'font': None, } editwin.text.cget = lambda opt: text_opts[opt] # monkey-path tkinter.font.Font with a mock object, so that # Font.measure('0') returns CHAR_WIDTH mock_font = Mock() def measure(char): if char == '0': return CHAR_WIDTH raise ValueError("measure should only be called on '0'!") mock_font.return_value.measure = measure patcher = patch('idlelib.squeezer.Font', mock_font) patcher.start() self.addCleanup(patcher.stop) def _test_count_lines_helper(self, linewidth, tabwidth): """Helper for test_count_lines.""" editwin = self.make_mock_editor_window() self._prepare_mock_editwin_for_count_lines(editwin, linewidth, tabwidth) squeezer = self.make_squeezer_instance(editwin) mock_count_lines = Mock(return_value=SENTINEL_VALUE) text = 'TEXT' with patch('idlelib.squeezer.count_lines_with_wrapping', mock_count_lines): self.assertIs(squeezer.count_lines(text), SENTINEL_VALUE) mock_count_lines.assert_called_with(text, linewidth, tabwidth) def test_init(self): """Test the creation of Squeezer instances.""" editwin = self.make_mock_editor_window() squeezer = self.make_squeezer_instance(editwin) self.assertIs(squeezer.editwin, editwin) self.assertEqual(squeezer.expandingbuttons, []) def test_write_no_tags(self): """Test Squeezer's overriding of the EditorWindow's write() method.""" editwin = self.make_mock_editor_window() for text in ['', 'TEXT', 'LONG TEXT' * 1000, 'MANY_LINES\n' * 100]: editwin.write = orig_write = Mock(return_value=SENTINEL_VALUE) squeezer = self.make_squeezer_instance(editwin) self.assertEqual(squeezer.editwin.write(text, ()), SENTINEL_VALUE) self.assertEqual(orig_write.call_count, 1) orig_write.assert_called_with(text, ()) self.assertEqual(len(squeezer.expandingbuttons), 0) def test_write_not_stdout(self): """Test Squeezer's overriding of the EditorWindow's write() method.""" for text in ['', 'TEXT', 'LONG TEXT' * 1000, 'MANY_LINES\n' * 100]: editwin = self.make_mock_editor_window() editwin.write.return_value = SENTINEL_VALUE orig_write = editwin.write squeezer = self.make_squeezer_instance(editwin) self.assertEqual(squeezer.editwin.write(text, "stderr"), SENTINEL_VALUE) self.assertEqual(orig_write.call_count, 1) orig_write.assert_called_with(text, "stderr") self.assertEqual(len(squeezer.expandingbuttons), 0) def test_write_stdout(self): """Test Squeezer's overriding of the EditorWindow's write() method.""" editwin = self.make_mock_editor_window() self._prepare_mock_editwin_for_count_lines(editwin, linewidth=80, tabwidth=8) for text in ['', 'TEXT']: editwin.write = orig_write = Mock(return_value=SENTINEL_VALUE) squeezer = self.make_squeezer_instance(editwin) squeezer.auto_squeeze_min_lines = 50 self.assertEqual(squeezer.editwin.write(text, "stdout"), SENTINEL_VALUE) self.assertEqual(orig_write.call_count, 1) orig_write.assert_called_with(text, "stdout") self.assertEqual(len(squeezer.expandingbuttons), 0) for text in ['LONG TEXT' * 1000, 'MANY_LINES\n' * 100]: editwin.write = orig_write = Mock(return_value=SENTINEL_VALUE) squeezer = self.make_squeezer_instance(editwin) squeezer.auto_squeeze_min_lines = 50 self.assertEqual(squeezer.editwin.write(text, "stdout"), None) self.assertEqual(orig_write.call_count, 0) self.assertEqual(len(squeezer.expandingbuttons), 1) def test_auto_squeeze(self): """Test that the auto-squeezing creates an ExpandingButton properly.""" root = get_test_tk_root(self) text_widget = Text(root) text_widget.mark_set("iomark", "1.0") editwin = self.make_mock_editor_window() editwin.text = text_widget squeezer = self.make_squeezer_instance(editwin) squeezer.auto_squeeze_min_lines = 5 squeezer.count_lines = Mock(return_value=6) editwin.write('TEXT\n'*6, "stdout") self.assertEqual(text_widget.get('1.0', 'end'), '\n') self.assertEqual(len(squeezer.expandingbuttons), 1) def test_squeeze_current_text_event(self): """Test the squeeze_current_text event.""" root = get_test_tk_root(self) # squeezing text should work for both stdout and stderr for tag_name in ["stdout", "stderr"]: text_widget = Text(root) text_widget.mark_set("iomark", "1.0") editwin = self.make_mock_editor_window() editwin.text = editwin.per.bottom = text_widget squeezer = self.make_squeezer_instance(editwin) squeezer.count_lines = Mock(return_value=6) # prepare some text in the Text widget text_widget.insert("1.0", "SOME\nTEXT\n", tag_name) text_widget.mark_set("insert", "1.0") self.assertEqual(text_widget.get('1.0', 'end'), 'SOME\nTEXT\n\n') self.assertEqual(len(squeezer.expandingbuttons), 0) # test squeezing the current text retval = squeezer.squeeze_current_text_event(event=Mock()) self.assertEqual(retval, "break") self.assertEqual(text_widget.get('1.0', 'end'), '\n\n') self.assertEqual(len(squeezer.expandingbuttons), 1) self.assertEqual(squeezer.expandingbuttons[0].s, 'SOME\nTEXT') # test that expanding the squeezed text works and afterwards the # Text widget contains the original text squeezer.expandingbuttons[0].expand(event=Mock()) self.assertEqual(text_widget.get('1.0', 'end'), 'SOME\nTEXT\n\n') self.assertEqual(len(squeezer.expandingbuttons), 0) def test_squeeze_current_text_event_no_allowed_tags(self): """Test that the event doesn't squeeze text without a relevant tag.""" root = get_test_tk_root(self) text_widget = Text(root) text_widget.mark_set("iomark", "1.0") editwin = self.make_mock_editor_window() editwin.text = editwin.per.bottom = text_widget squeezer = self.make_squeezer_instance(editwin) squeezer.count_lines = Mock(return_value=6) # prepare some text in the Text widget text_widget.insert("1.0", "SOME\nTEXT\n", "TAG") text_widget.mark_set("insert", "1.0") self.assertEqual(text_widget.get('1.0', 'end'), 'SOME\nTEXT\n\n') self.assertEqual(len(squeezer.expandingbuttons), 0) # test squeezing the current text retval = squeezer.squeeze_current_text_event(event=Mock()) self.assertEqual(retval, "break") self.assertEqual(text_widget.get('1.0', 'end'), 'SOME\nTEXT\n\n') self.assertEqual(len(squeezer.expandingbuttons), 0) def test_squeeze_text_before_existing_squeezed_text(self): """Test squeezing text before existing squeezed text.""" root = get_test_tk_root(self) text_widget = Text(root) text_widget.mark_set("iomark", "1.0") editwin = self.make_mock_editor_window() editwin.text = editwin.per.bottom = text_widget squeezer = self.make_squeezer_instance(editwin) squeezer.count_lines = Mock(return_value=6) # prepare some text in the Text widget and squeeze it text_widget.insert("1.0", "SOME\nTEXT\n", "stdout") text_widget.mark_set("insert", "1.0") squeezer.squeeze_current_text_event(event=Mock()) self.assertEqual(len(squeezer.expandingbuttons), 1) # test squeezing the current text text_widget.insert("1.0", "MORE\nSTUFF\n", "stdout") text_widget.mark_set("insert", "1.0") retval = squeezer.squeeze_current_text_event(event=Mock()) self.assertEqual(retval, "break") self.assertEqual(text_widget.get('1.0', 'end'), '\n\n\n') self.assertEqual(len(squeezer.expandingbuttons), 2) self.assertTrue(text_widget.compare( squeezer.expandingbuttons[0], '<', squeezer.expandingbuttons[1], )) GetOptionSignature = namedtuple('GetOptionSignature', 'configType section option default type warn_on_default raw') @classmethod def _make_sig(cls, configType, section, option, default=sentinel.NOT_GIVEN, type=sentinel.NOT_GIVEN, warn_on_default=sentinel.NOT_GIVEN, raw=sentinel.NOT_GIVEN): return cls.GetOptionSignature(configType, section, option, default, type, warn_on_default, raw) @classmethod def get_GetOption_signature(cls, mock_call_obj): args, kwargs = mock_call_obj[-2:] return cls._make_sig(*args, **kwargs) def test_reload(self): """Test the reload() class-method.""" self.assertIsInstance(Squeezer.auto_squeeze_min_lines, int) idleConf.SetOption('main', 'PyShell', 'auto-squeeze-min-lines', '42') Squeezer.reload() self.assertEqual(Squeezer.auto_squeeze_min_lines, 42) class ExpandingButtonTest(unittest.TestCase): """Tests for the ExpandingButton class.""" # In these tests the squeezer instance is a mock, but actual tkinter # Text and Button instances are created. def make_mock_squeezer(self): """Helper for tests: Create a mock Squeezer object.""" root = get_test_tk_root(self) squeezer = Mock() squeezer.editwin.text = Text(root) # Set default values for the configuration settings squeezer.auto_squeeze_min_lines = 50 return squeezer @patch('idlelib.squeezer.Hovertip', autospec=Hovertip) def test_init(self, MockHovertip): """Test the simplest creation of an ExpandingButton.""" squeezer = self.make_mock_squeezer() text_widget = squeezer.editwin.text expandingbutton = ExpandingButton('TEXT', 'TAGS', 50, squeezer) self.assertEqual(expandingbutton.s, 'TEXT') # check that the underlying tkinter.Button is properly configured self.assertEqual(expandingbutton.master, text_widget) self.assertTrue('50 lines' in expandingbutton.cget('text')) # check that the text widget still contains no text self.assertEqual(text_widget.get('1.0', 'end'), '\n') # check that the mouse events are bound self.assertIn('<Double-Button-1>', expandingbutton.bind()) right_button_code = '<Button-%s>' % ('2' if macosx.isAquaTk() else '3') self.assertIn(right_button_code, expandingbutton.bind()) # check that ToolTip was called once, with appropriate values self.assertEqual(MockHovertip.call_count, 1) MockHovertip.assert_called_with(expandingbutton, ANY, hover_delay=ANY) # check that 'right-click' appears in the tooltip text tooltip_text = MockHovertip.call_args[0][1] self.assertIn('right-click', tooltip_text.lower()) def test_expand(self): """Test the expand event.""" squeezer = self.make_mock_squeezer() expandingbutton = ExpandingButton('TEXT', 'TAGS', 50, squeezer) # insert the button into the text widget # (this is normally done by the Squeezer class) text_widget = expandingbutton.text text_widget.window_create("1.0", window=expandingbutton) # set base_text to the text widget, so that changes are actually made # to it (by ExpandingButton) and we can inspect these changes afterwards expandingbutton.base_text = expandingbutton.text # trigger the expand event retval = expandingbutton.expand(event=Mock()) self.assertEqual(retval, None) # check that the text was inserted into the text widget self.assertEqual(text_widget.get('1.0', 'end'), 'TEXT\n') # check that the 'TAGS' tag was set on the inserted text text_end_index = text_widget.index('end-1c') self.assertEqual(text_widget.get('1.0', text_end_index), 'TEXT') self.assertEqual(text_widget.tag_nextrange('TAGS', '1.0'), ('1.0', text_end_index)) # check that the button removed itself from squeezer.expandingbuttons self.assertEqual(squeezer.expandingbuttons.remove.call_count, 1) squeezer.expandingbuttons.remove.assert_called_with(expandingbutton) def test_expand_dangerous_oupput(self): """Test that expanding very long output asks user for confirmation.""" squeezer = self.make_mock_squeezer() text = 'a' * 10**5 expandingbutton = ExpandingButton(text, 'TAGS', 50, squeezer) expandingbutton.set_is_dangerous() self.assertTrue(expandingbutton.is_dangerous) # insert the button into the text widget # (this is normally done by the Squeezer class) text_widget = expandingbutton.text text_widget.window_create("1.0", window=expandingbutton) # set base_text to the text widget, so that changes are actually made # to it (by ExpandingButton) and we can inspect these changes afterwards expandingbutton.base_text = expandingbutton.text # patch the message box module to always return False with patch('idlelib.squeezer.tkMessageBox') as mock_msgbox: mock_msgbox.askokcancel.return_value = False mock_msgbox.askyesno.return_value = False # trigger the expand event retval = expandingbutton.expand(event=Mock()) # check that the event chain was broken and no text was inserted self.assertEqual(retval, 'break') self.assertEqual(expandingbutton.text.get('1.0', 'end-1c'), '') # patch the message box module to always return True with patch('idlelib.squeezer.tkMessageBox') as mock_msgbox: mock_msgbox.askokcancel.return_value = True mock_msgbox.askyesno.return_value = True # trigger the expand event retval = expandingbutton.expand(event=Mock()) # check that the event chain wasn't broken and the text was inserted self.assertEqual(retval, None) self.assertEqual(expandingbutton.text.get('1.0', 'end-1c'), text) def test_copy(self): """Test the copy event.""" # testing with the actual clipboard proved problematic, so this test # replaces the clipboard manipulation functions with mocks and checks # that they are called appropriately squeezer = self.make_mock_squeezer() expandingbutton = ExpandingButton('TEXT', 'TAGS', 50, squeezer) expandingbutton.clipboard_clear = Mock() expandingbutton.clipboard_append = Mock() # trigger the copy event retval = expandingbutton.copy(event=Mock()) self.assertEqual(retval, None) # check that the expanding button called clipboard_clear() and # clipboard_append('TEXT') once each self.assertEqual(expandingbutton.clipboard_clear.call_count, 1) self.assertEqual(expandingbutton.clipboard_append.call_count, 1) expandingbutton.clipboard_append.assert_called_with('TEXT') def test_view(self): """Test the view event.""" squeezer = self.make_mock_squeezer() expandingbutton = ExpandingButton('TEXT', 'TAGS', 50, squeezer) expandingbutton.selection_own = Mock() with patch('idlelib.squeezer.view_text', autospec=view_text)\ as mock_view_text: # trigger the view event expandingbutton.view(event=Mock()) # check that the expanding button called view_text self.assertEqual(mock_view_text.call_count, 1) # check that the proper text was passed self.assertEqual(mock_view_text.call_args[0][2], 'TEXT') def test_rmenu(self): """Test the context menu.""" squeezer = self.make_mock_squeezer() expandingbutton = ExpandingButton('TEXT', 'TAGS', 50, squeezer) with patch('tkinter.Menu') as mock_Menu: mock_menu = Mock() mock_Menu.return_value = mock_menu mock_event = Mock() mock_event.x = 10 mock_event.y = 10 expandingbutton.context_menu_event(event=mock_event) self.assertEqual(mock_menu.add_command.call_count, len(expandingbutton.rmenu_specs)) for label, *data in expandingbutton.rmenu_specs: mock_menu.add_command.assert_any_call(label=label, command=ANY) if __name__ == '__main__': unittest.main(verbosity=2)
Save File
Cancel