| "Test autocomplete, coverage 93%." |
| |
| import unittest |
| from unittest.mock import Mock, patch |
| from test.support import requires |
| from tkinter import Tk, Text |
| import os |
| import __main__ |
| |
| import idlelib.autocomplete as ac |
| import idlelib.autocomplete_w as acw |
| from idlelib.idle_test.mock_idle import Func |
| from idlelib.idle_test.mock_tk import Event |
| |
| |
| class DummyEditwin: |
| def __init__(self, root, text): |
| self.root = root |
| self.text = text |
| self.indentwidth = 8 |
| self.tabwidth = 8 |
| self.prompt_last_line = '>>>' # Currently not used by autocomplete. |
| |
| |
| class AutoCompleteTest(unittest.TestCase): |
| |
| @classmethod |
| def setUpClass(cls): |
| requires('gui') |
| cls.root = Tk() |
| cls.root.withdraw() |
| cls.text = Text(cls.root) |
| cls.editor = DummyEditwin(cls.root, cls.text) |
| |
| @classmethod |
| def tearDownClass(cls): |
| del cls.editor, cls.text |
| cls.root.update_idletasks() |
| cls.root.destroy() |
| del cls.root |
| |
| def setUp(self): |
| self.text.delete('1.0', 'end') |
| self.autocomplete = ac.AutoComplete(self.editor) |
| |
| def test_init(self): |
| self.assertEqual(self.autocomplete.editwin, self.editor) |
| self.assertEqual(self.autocomplete.text, self.text) |
| |
| def test_make_autocomplete_window(self): |
| testwin = self.autocomplete._make_autocomplete_window() |
| self.assertIsInstance(testwin, acw.AutoCompleteWindow) |
| |
| def test_remove_autocomplete_window(self): |
| acp = self.autocomplete |
| acp.autocompletewindow = m = Mock() |
| acp._remove_autocomplete_window() |
| m.hide_window.assert_called_once() |
| self.assertIsNone(acp.autocompletewindow) |
| |
| def test_force_open_completions_event(self): |
| # Call _open_completions and break. |
| acp = self.autocomplete |
| open_c = Func() |
| acp.open_completions = open_c |
| self.assertEqual(acp.force_open_completions_event('event'), 'break') |
| self.assertEqual(open_c.args[0], ac.FORCE) |
| |
| def test_autocomplete_event(self): |
| Equal = self.assertEqual |
| acp = self.autocomplete |
| |
| # Result of autocomplete event: If modified tab, None. |
| ev = Event(mc_state=True) |
| self.assertIsNone(acp.autocomplete_event(ev)) |
| del ev.mc_state |
| |
| # If tab after whitespace, None. |
| self.text.insert('1.0', ' """Docstring.\n ') |
| self.assertIsNone(acp.autocomplete_event(ev)) |
| self.text.delete('1.0', 'end') |
| |
| # If active autocomplete window, complete() and 'break'. |
| self.text.insert('1.0', 're.') |
| acp.autocompletewindow = mock = Mock() |
| mock.is_active = Mock(return_value=True) |
| Equal(acp.autocomplete_event(ev), 'break') |
| mock.complete.assert_called_once() |
| acp.autocompletewindow = None |
| |
| # If no active autocomplete window, open_completions(), None/break. |
| open_c = Func(result=False) |
| acp.open_completions = open_c |
| Equal(acp.autocomplete_event(ev), None) |
| Equal(open_c.args[0], ac.TAB) |
| open_c.result = True |
| Equal(acp.autocomplete_event(ev), 'break') |
| Equal(open_c.args[0], ac.TAB) |
| |
| def test_try_open_completions_event(self): |
| Equal = self.assertEqual |
| text = self.text |
| acp = self.autocomplete |
| trycompletions = acp.try_open_completions_event |
| after = Func(result='after1') |
| acp.text.after = after |
| |
| # If no text or trigger, after not called. |
| trycompletions() |
| Equal(after.called, 0) |
| text.insert('1.0', 're') |
| trycompletions() |
| Equal(after.called, 0) |
| |
| # Attribute needed, no existing callback. |
| text.insert('insert', ' re.') |
| acp._delayed_completion_id = None |
| trycompletions() |
| Equal(acp._delayed_completion_index, text.index('insert')) |
| Equal(after.args, |
| (acp.popupwait, acp._delayed_open_completions, ac.TRY_A)) |
| cb1 = acp._delayed_completion_id |
| Equal(cb1, 'after1') |
| |
| # File needed, existing callback cancelled. |
| text.insert('insert', ' "./Lib/') |
| after.result = 'after2' |
| cancel = Func() |
| acp.text.after_cancel = cancel |
| trycompletions() |
| Equal(acp._delayed_completion_index, text.index('insert')) |
| Equal(cancel.args, (cb1,)) |
| Equal(after.args, |
| (acp.popupwait, acp._delayed_open_completions, ac.TRY_F)) |
| Equal(acp._delayed_completion_id, 'after2') |
| |
| def test_delayed_open_completions(self): |
| Equal = self.assertEqual |
| acp = self.autocomplete |
| open_c = Func() |
| acp.open_completions = open_c |
| self.text.insert('1.0', '"dict.') |
| |
| # Set autocomplete._delayed_completion_id to None. |
| # Text index changed, don't call open_completions. |
| acp._delayed_completion_id = 'after' |
| acp._delayed_completion_index = self.text.index('insert+1c') |
| acp._delayed_open_completions('dummy') |
| self.assertIsNone(acp._delayed_completion_id) |
| Equal(open_c.called, 0) |
| |
| # Text index unchanged, call open_completions. |
| acp._delayed_completion_index = self.text.index('insert') |
| acp._delayed_open_completions((1, 2, 3, ac.FILES)) |
| self.assertEqual(open_c.args[0], (1, 2, 3, ac.FILES)) |
| |
| def test_oc_cancel_comment(self): |
| none = self.assertIsNone |
| acp = self.autocomplete |
| |
| # Comment is in neither code or string. |
| acp._delayed_completion_id = 'after' |
| after = Func(result='after') |
| acp.text.after_cancel = after |
| self.text.insert(1.0, '# comment') |
| none(acp.open_completions(ac.TAB)) # From 'else' after 'elif'. |
| none(acp._delayed_completion_id) |
| |
| def test_oc_no_list(self): |
| acp = self.autocomplete |
| fetch = Func(result=([],[])) |
| acp.fetch_completions = fetch |
| self.text.insert('1.0', 'object') |
| self.assertIsNone(acp.open_completions(ac.TAB)) |
| self.text.insert('insert', '.') |
| self.assertIsNone(acp.open_completions(ac.TAB)) |
| self.assertEqual(fetch.called, 2) |
| |
| |
| def test_open_completions_none(self): |
| # Test other two None returns. |
| none = self.assertIsNone |
| acp = self.autocomplete |
| |
| # No object for attributes or need call not allowed. |
| self.text.insert(1.0, '.') |
| none(acp.open_completions(ac.TAB)) |
| self.text.insert('insert', ' int().') |
| none(acp.open_completions(ac.TAB)) |
| |
| # Blank or quote trigger 'if complete ...'. |
| self.text.delete(1.0, 'end') |
| self.assertFalse(acp.open_completions(ac.TAB)) |
| self.text.insert('1.0', '"') |
| self.assertFalse(acp.open_completions(ac.TAB)) |
| self.text.delete('1.0', 'end') |
| |
| class dummy_acw: |
| __init__ = Func() |
| show_window = Func(result=False) |
| hide_window = Func() |
| |
| def test_open_completions(self): |
| # Test completions of files and attributes. |
| acp = self.autocomplete |
| fetch = Func(result=(['tem'],['tem', '_tem'])) |
| acp.fetch_completions = fetch |
| def make_acw(): return self.dummy_acw() |
| acp._make_autocomplete_window = make_acw |
| |
| self.text.insert('1.0', 'int.') |
| acp.open_completions(ac.TAB) |
| self.assertIsInstance(acp.autocompletewindow, self.dummy_acw) |
| self.text.delete('1.0', 'end') |
| |
| # Test files. |
| self.text.insert('1.0', '"t') |
| self.assertTrue(acp.open_completions(ac.TAB)) |
| self.text.delete('1.0', 'end') |
| |
| def test_fetch_completions(self): |
| # Test that fetch_completions returns 2 lists: |
| # For attribute completion, a large list containing all variables, and |
| # a small list containing non-private variables. |
| # For file completion, a large list containing all files in the path, |
| # and a small list containing files that do not start with '.'. |
| acp = self.autocomplete |
| small, large = acp.fetch_completions( |
| '', ac.ATTRS) |
| if hasattr(__main__, '__file__') and __main__.__file__ != ac.__file__: |
| self.assertNotIn('AutoComplete', small) # See issue 36405. |
| |
| # Test attributes |
| s, b = acp.fetch_completions('', ac.ATTRS) |
| self.assertLess(len(small), len(large)) |
| self.assertTrue(all(filter(lambda x: x.startswith('_'), s))) |
| self.assertTrue(any(filter(lambda x: x.startswith('_'), b))) |
| |
| # Test smalll should respect to __all__. |
| with patch.dict('__main__.__dict__', {'__all__': ['a', 'b']}): |
| s, b = acp.fetch_completions('', ac.ATTRS) |
| self.assertEqual(s, ['a', 'b']) |
| self.assertIn('__name__', b) # From __main__.__dict__. |
| self.assertIn('sum', b) # From __main__.__builtins__.__dict__. |
| self.assertIn('nonlocal', b) # From keyword.kwlist. |
| pos = b.index('False') # Test False not included twice. |
| self.assertNotEqual(b[pos+1], 'False') |
| |
| # Test attributes with name entity. |
| mock = Mock() |
| mock._private = Mock() |
| with patch.dict('__main__.__dict__', {'foo': mock}): |
| s, b = acp.fetch_completions('foo', ac.ATTRS) |
| self.assertNotIn('_private', s) |
| self.assertIn('_private', b) |
| self.assertEqual(s, [i for i in sorted(dir(mock)) if i[:1] != '_']) |
| self.assertEqual(b, sorted(dir(mock))) |
| |
| # Test files |
| def _listdir(path): |
| # This will be patch and used in fetch_completions. |
| if path == '.': |
| return ['foo', 'bar', '.hidden'] |
| return ['monty', 'python', '.hidden'] |
| |
| with patch.object(os, 'listdir', _listdir): |
| s, b = acp.fetch_completions('', ac.FILES) |
| self.assertEqual(s, ['bar', 'foo']) |
| self.assertEqual(b, ['.hidden', 'bar', 'foo']) |
| |
| s, b = acp.fetch_completions('~', ac.FILES) |
| self.assertEqual(s, ['monty', 'python']) |
| self.assertEqual(b, ['.hidden', 'monty', 'python']) |
| |
| def test_get_entity(self): |
| # Test that a name is in the namespace of sys.modules and |
| # __main__.__dict__. |
| acp = self.autocomplete |
| Equal = self.assertEqual |
| |
| Equal(acp.get_entity('int'), int) |
| |
| # Test name from sys.modules. |
| mock = Mock() |
| with patch.dict('sys.modules', {'tempfile': mock}): |
| Equal(acp.get_entity('tempfile'), mock) |
| |
| # Test name from __main__.__dict__. |
| di = {'foo': 10, 'bar': 20} |
| with patch.dict('__main__.__dict__', {'d': di}): |
| Equal(acp.get_entity('d'), di) |
| |
| # Test name not in namespace. |
| with patch.dict('__main__.__dict__', {}): |
| with self.assertRaises(NameError): |
| acp.get_entity('not_exist') |
| |
| |
| if __name__ == '__main__': |
| unittest.main(verbosity=2) |