mirror of https://github.com/pypa/pip
105 lines
2.8 KiB
Python
105 lines
2.8 KiB
Python
import os
|
|
from tests.lib import reset_env
|
|
|
|
|
|
def test_completion_for_bash():
|
|
"""
|
|
Test getting completion for bash shell
|
|
"""
|
|
script = reset_env()
|
|
bash_completion = """\
|
|
_pip_completion()
|
|
{
|
|
COMPREPLY=( $( COMP_WORDS="${COMP_WORDS[*]}" \\
|
|
COMP_CWORD=$COMP_CWORD \\
|
|
PIP_AUTO_COMPLETE=1 $1 ) )
|
|
}
|
|
complete -o default -F _pip_completion pip"""
|
|
|
|
result = script.pip('completion', '--bash')
|
|
assert bash_completion in result.stdout, 'bash completion is wrong'
|
|
|
|
|
|
def test_completion_for_zsh():
|
|
"""
|
|
Test getting completion for zsh shell
|
|
"""
|
|
script = reset_env()
|
|
zsh_completion = """\
|
|
function _pip_completion {
|
|
local words cword
|
|
read -Ac words
|
|
read -cn cword
|
|
reply=( $( COMP_WORDS="$words[*]" \\
|
|
COMP_CWORD=$(( cword-1 )) \\
|
|
PIP_AUTO_COMPLETE=1 $words[1] ) )
|
|
}
|
|
compctl -K _pip_completion pip"""
|
|
|
|
result = script.pip('completion', '--zsh')
|
|
assert zsh_completion in result.stdout, 'zsh completion is wrong'
|
|
|
|
|
|
def test_completion_for_unknown_shell():
|
|
"""
|
|
Test getting completion for an unknown shell
|
|
"""
|
|
script = reset_env()
|
|
error_msg = 'no such option: --myfooshell'
|
|
result = script.pip('completion', '--myfooshell', expect_error=True)
|
|
assert error_msg in result.stderr, 'tests for an unknown shell failed'
|
|
|
|
|
|
def test_completion_alone():
|
|
"""
|
|
Test getting completion for none shell, just pip completion
|
|
"""
|
|
script = reset_env()
|
|
result = script.pip('completion', expect_error=True)
|
|
assert 'ERROR: You must pass --bash or --zsh' in result.stderr, \
|
|
'completion alone failed -- ' + result.stderr
|
|
|
|
|
|
def setup_completion(words, cword):
|
|
environ = os.environ.copy()
|
|
script = reset_env(environ)
|
|
environ['PIP_AUTO_COMPLETE'] = '1'
|
|
environ['COMP_WORDS'] = words
|
|
environ['COMP_CWORD'] = cword
|
|
|
|
# expect_error is True because autocomplete exists with 1 status code
|
|
result = script.run('python', '-c', 'import pip;pip.autocomplete()',
|
|
expect_error=True)
|
|
|
|
return result, script
|
|
|
|
|
|
def test_completion_for_un_snippet():
|
|
"""
|
|
Test getting completion for ``un`` should return
|
|
uninstall and unzip
|
|
"""
|
|
|
|
res, env = setup_completion('pip un', '1')
|
|
assert res.stdout.strip().split() == ['uninstall', 'unzip'], res.stdout
|
|
|
|
|
|
def test_completion_for_default_parameters():
|
|
"""
|
|
Test getting completion for ``--`` should contain --help
|
|
"""
|
|
|
|
res, env = setup_completion('pip --', '1')
|
|
assert '--help' in res.stdout,\
|
|
"autocomplete function could not complete ``--``"
|
|
|
|
|
|
def test_completion_option_for_command():
|
|
"""
|
|
Test getting completion for ``--`` in command (eg. pip search --)
|
|
"""
|
|
|
|
res, env = setup_completion('pip search --', '2')
|
|
assert '--help' in res.stdout,\
|
|
"autocomplete function could not complete ``--``"
|