test_autocomplete.py

"Test autocomplete, coverage 93%."

importunittest
fromunittest.mockimportMock, patch
fromtest.supportimportrequires
fromtkinterimportTk, Text
importos
import__main__

importidlelib.autocompleteasac
importidlelib.autocomplete_wasacw
fromidlelib.idle_test.mock_idleimportFunc
fromidlelib.idle_test.mock_tkimportEvent


classDummyEditwin:
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.


classAutoCompleteTest(unittest.TestCase):

@classmethod
defsetUpClass(cls):
requires('gui')
cls.root=Tk()
cls.root.withdraw()
cls.text=Text(cls.root)
cls.editor=DummyEditwin(cls.root, cls.text)

@classmethod
deftearDownClass(cls):
delcls.editor, cls.text
cls.root.update_idletasks()
cls.root.destroy()
delcls.root

defsetUp(self):
self.text.delete('1.0', 'end')
self.autocomplete=ac.AutoComplete(self.editor)

deftest_init(self):
self.assertEqual(self.autocomplete.editwin, self.editor)
self.assertEqual(self.autocomplete.text, self.text)

deftest_make_autocomplete_window(self):
testwin=self.autocomplete._make_autocomplete_window()
self.assertIsInstance(testwin, acw.AutoCompleteWindow)

deftest_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)

deftest_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)

deftest_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))
delev.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)

deftest_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')

deftest_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))

deftest_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)

deftest_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)


deftest_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')

classdummy_acw:
__init__=Func()
show_window=Func(result=False)
hide_window=Func()

deftest_open_completions(self):
# Test completions of files and attributes.
acp=self.autocomplete
fetch=Func(result=(['tem'],['tem', '_tem']))
acp.fetch_completions=fetch
defmake_acw(): returnself.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')

deftest_completion_kwds(self):
self.assertIn('and', ac.completion_kwds)
self.assertIn('case', ac.completion_kwds)
self.assertNotIn('None', ac.completion_kwds)

deftest_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)
ifhasattr(__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(lambdax: x.startswith('_'), s)))
self.assertTrue(any(filter(lambdax: x.startswith('_'), b)))

# Test smalll should respect to __all__.
withpatch.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()
withpatch.dict('__main__.__dict__', {'foo': mock}):
s, b=acp.fetch_completions('foo', ac.ATTRS)
self.assertNotIn('_private', s)
self.assertIn('_private', b)
self.assertEqual(s, [iforiinsorted(dir(mock)) ifi[:1] !='_'])
self.assertEqual(b, sorted(dir(mock)))

# Test files
def_listdir(path):
# This will be patch and used in fetch_completions.
ifpath=='.':
return ['foo', 'bar', '.hidden']
return ['monty', 'python', '.hidden']

withpatch.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'])

deftest_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()
withpatch.dict('sys.modules', {'tempfile': mock}):
Equal(acp.get_entity('tempfile'), mock)

# Test name from __main__.__dict__.
di= {'foo': 10, 'bar': 20}
withpatch.dict('__main__.__dict__', {'d': di}):
Equal(acp.get_entity('d'), di)

# Test name not in namespace.
withpatch.dict('__main__.__dict__', {}):
withself.assertRaises(NameError):
acp.get_entity('not_exist')


if__name__=='__main__':
unittest.main(verbosity=2)