def testChainedCompleter(self):
matches = list(C1.Matches(['f'], 0, 'f'))
self.assertEqual(['foo.py ', 'foo '], matches)
p = completion.GlobPredicate('*.py')
c2 = completion.ChainedCompleter([A1], predicate=p)
matches = list(c2.Matches(['f'], 0, 'f'))
self.assertEqual([], matches)
def _InitDefaultCompletions(ex, comp_lookup):
# register builtins and words
comp_builtins.Complete(['-E', '-A', 'command'], ex, comp_lookup)
# register path completion
comp_builtins.Complete(['-D', '-A', 'file'], ex, comp_lookup)
# TODO: Move this into demo/slow-completion.sh
if 1:
# Something for fun, to show off. Also: test that you don't repeatedly hit
# the file system / network / coprocess.
A1 = completion.WordsAction(['foo.py', 'foo', 'bar.py'])
A2 = completion.WordsAction(['m%d' % i for i in xrange(5)], delay=0.1)
C1 = completion.ChainedCompleter([A1, A2])
comp_lookup.RegisterName('slowc', C1)
def testChainedCompleter(self):
print(list(C1.Matches(['f'], 0, 'f')))
p = completion.GlobPredicate('*.py')
c2 = completion.ChainedCompleter([A1], predicate=p)
print(list(c2.Matches(['f'], 0, 'f')))
from core import legacy
from core import lexer
from core import state
from core import test_lib
from core import word_eval
from core import ui
from core.id_kind import Id
from osh import ast_ as ast
from osh import parse_lib
assign_op_e = ast.assign_op_e
A1 = completion.WordsAction(['foo.py', 'foo', 'bar.py'])
C1 = completion.ChainedCompleter([A1])
status_lines = [ui.TestStatusLine()] * 10 # A bunch of dummies
mem = state.Mem('', [], {}, None)
exec_opts = state.ExecOpts(mem)
STATUS = completion.StatusOutput(status_lines, exec_opts)
V1 = completion.WordsAction(['$var1', '$var2', '$another_var'])
EMPTY = completion.WordsAction(['grep', 'sed', 'test'])
FIRST = completion.WordsAction(['grep', 'sed', 'test'])
class CompletionTest(unittest.TestCase):
def testLookup(self):
c = completion.CompletionLookup()
def _BuildCompletionChain(argv, arg, ex):
"""Given flags to complete/compgen, built a ChainedCompleter."""
actions = []
# NOTE: bash doesn't actually check the name until completion time, but
# obviously it's better to check here.
if arg.F:
func_name = arg.F
func = ex.funcs.get(func_name)
if func is None:
raise args.UsageError('Function %r not found' % func_name)
actions.append(completion.ShellFuncAction(ex, func))
# NOTE: We need completion for -A action itself!!! bash seems to have it.
for name in arg.actions:
if name == 'alias':
a = _SortedWordsAction(ex.aliases)
elif name == 'binding':
# TODO: Where do we get this from?
a = _SortedWordsAction(['vi-delete'])
elif name == 'command':
# TODO: This needs keywords too
actions.append(_SortedWordsAction(builtin.BUILTIN_NAMES))
a = completion.ExternalCommandAction(ex.mem)
elif name == 'directory':
a = completion.FileSystemAction(dirs_only=True)
elif name == 'file':
a = completion.FileSystemAction()
elif name == 'function':
a = _SortedWordsAction(ex.funcs)
elif name == 'job':
a = _SortedWordsAction(['jobs-not-implemented'])
elif name == 'user':
a = _UsersAction()
elif name == 'variable':
a = completion.VariablesAction(ex.mem)
elif name == 'helptopic':
a = _SortedWordsAction(osh_help.TOPIC_LOOKUP)
elif name == 'setopt':
a = _SortedWordsAction(state.SET_OPTION_NAMES)
elif name == 'shopt':
a = _SortedWordsAction(state.SHOPT_OPTION_NAMES)
elif name == 'signal':
a = _SortedWordsAction(['TODO:signals'])
elif name == 'stopped':
a = _SortedWordsAction(['jobs-not-implemented'])
else:
raise NotImplementedError(name)
actions.append(a)
# e.g. -W comes after -A directory
if arg.W:
# TODO: Split with IFS. Is that done at registration time or completion
# time?
actions.append(completion.WordsAction(arg.W.split()))
if not actions:
raise args.UsageError('No actions defined in completion: %s' % argv)
chain = completion.ChainedCompleter(
actions,
prefix=arg.P or '',
suffix=arg.S or '')
return chain