301 lines
11 KiB
Python
301 lines
11 KiB
Python
|
"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)
|