mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-04 08:24:40 +01:00
38295a1dd9
* refactor: remove usage of global variables in abbrevs.py * chore: add flake8-mutable to prevent mutable defaults * fix: abbrevs expand test * refactor: add xonsh session singleton * refactor: fix circular errors when using xonshSession as singleton * refactor: remove black magicked builtin attributes * style: black format tests as well * refactor: update tests to use xonsh-session singleton * refactor: update abbrevs to not use builtins * test: remove DummyCommandsCache and patch orig class * fix: failing test_command_completers * test: use monkeypatch to update xession fixture * fix: failing test_pipelines * fix: failing test_main * chore: run test suit as single invocation * test: fix tests/test_xonsh.xsh * refactor: remove builtins from docs/conf.py * fix: mypy error in jobs * fix: test error from test_main * test: close xession error in test_command_completers * chore: use pytest-cov for reporting coverage this will include subprocess calls, and will increase coverage * style:
199 lines
5.1 KiB
Python
199 lines
5.1 KiB
Python
import pytest
|
|
from inspect import signature
|
|
from unittest.mock import MagicMock
|
|
from prompt_toolkit.document import Document
|
|
from prompt_toolkit.completion import Completion as PTKCompletion
|
|
|
|
from xonsh.aliases import Aliases
|
|
from xonsh.completer import Completer
|
|
from xonsh.completers.tools import RichCompletion
|
|
from xonsh.ptk_shell.completer import PromptToolkitCompleter
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"completion, lprefix, ptk_completion",
|
|
[
|
|
(RichCompletion("x", 0, "x()", "func"), 0, None),
|
|
(RichCompletion("x", 1, "xx", "instance"), 0, None),
|
|
(
|
|
RichCompletion("x", description="wow"),
|
|
5,
|
|
PTKCompletion(RichCompletion("x"), -5, "x", "wow"),
|
|
),
|
|
(RichCompletion("x"), 5, PTKCompletion(RichCompletion("x"), -5, "x")),
|
|
("x", 5, PTKCompletion("x", -5, "x")),
|
|
],
|
|
)
|
|
def test_rich_completion(completion, lprefix, ptk_completion, monkeypatch, xession):
|
|
xonsh_completer_mock = MagicMock()
|
|
xonsh_completer_mock.complete.return_value = {completion}, lprefix
|
|
|
|
ptk_completer = PromptToolkitCompleter(xonsh_completer_mock, None, None)
|
|
ptk_completer.reserve_space = lambda: None
|
|
ptk_completer.suggestion_completion = lambda _, __: None
|
|
|
|
document_mock = MagicMock()
|
|
document_mock.text = ""
|
|
document_mock.current_line = ""
|
|
document_mock.cursor_position_col = 0
|
|
|
|
monkeypatch.setattr(xession, "aliases", Aliases())
|
|
|
|
completions = list(ptk_completer.get_completions(document_mock, MagicMock()))
|
|
if isinstance(completion, RichCompletion) and not ptk_completion:
|
|
assert completions == [
|
|
PTKCompletion(
|
|
completion,
|
|
-completion.prefix_len,
|
|
completion.display,
|
|
completion.description,
|
|
)
|
|
]
|
|
else:
|
|
assert completions == [ptk_completion]
|
|
|
|
|
|
EXPANSION_CASES = (
|
|
(
|
|
"sanity",
|
|
6,
|
|
dict(
|
|
prefix="sanity",
|
|
line="sanity",
|
|
begidx=0,
|
|
endidx=6,
|
|
multiline_text="sanity",
|
|
cursor_index=6,
|
|
),
|
|
),
|
|
(
|
|
"gb ",
|
|
3,
|
|
dict(
|
|
prefix="",
|
|
line="git branch ",
|
|
begidx=11,
|
|
endidx=11,
|
|
multiline_text="git branch ",
|
|
cursor_index=11,
|
|
),
|
|
),
|
|
(
|
|
"gb ",
|
|
1,
|
|
dict(
|
|
prefix="g",
|
|
line="gb ",
|
|
begidx=0,
|
|
endidx=1,
|
|
multiline_text="gb ",
|
|
cursor_index=1,
|
|
),
|
|
),
|
|
(
|
|
"gb",
|
|
0,
|
|
dict(
|
|
prefix="",
|
|
line="gb",
|
|
begidx=0,
|
|
endidx=0,
|
|
multiline_text="gb",
|
|
cursor_index=0,
|
|
),
|
|
),
|
|
(
|
|
" gb ",
|
|
0,
|
|
dict(
|
|
prefix="",
|
|
line=" gb ", # the PTK completer `lstrip`s the line
|
|
begidx=0,
|
|
endidx=0,
|
|
multiline_text=" gb ",
|
|
cursor_index=0,
|
|
),
|
|
),
|
|
(
|
|
"gb --",
|
|
5,
|
|
dict(
|
|
prefix="--",
|
|
line="git branch --",
|
|
begidx=11,
|
|
endidx=13,
|
|
multiline_text="git branch --",
|
|
cursor_index=13,
|
|
),
|
|
),
|
|
(
|
|
"nice\ngb --",
|
|
10,
|
|
dict(
|
|
prefix="--",
|
|
line="git branch --",
|
|
begidx=11,
|
|
endidx=13,
|
|
multiline_text="nice\ngit branch --",
|
|
cursor_index=18,
|
|
),
|
|
),
|
|
(
|
|
"nice\n gb --",
|
|
11,
|
|
dict(
|
|
prefix="--",
|
|
line=" git branch --",
|
|
begidx=12,
|
|
endidx=14,
|
|
multiline_text="nice\n git branch --",
|
|
cursor_index=19,
|
|
),
|
|
),
|
|
(
|
|
"gb -- wow",
|
|
5,
|
|
dict(
|
|
prefix="--",
|
|
line="git branch -- wow",
|
|
begidx=11,
|
|
endidx=13,
|
|
multiline_text="git branch -- wow",
|
|
cursor_index=13,
|
|
),
|
|
),
|
|
(
|
|
"gb --wow",
|
|
5,
|
|
dict(
|
|
prefix="--",
|
|
line="git branch --wow",
|
|
begidx=11,
|
|
endidx=13,
|
|
multiline_text="git branch --wow",
|
|
cursor_index=13,
|
|
),
|
|
),
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("code, index, expected_args", EXPANSION_CASES)
|
|
def test_alias_expansion(code, index, expected_args, monkeypatch, xession):
|
|
xonsh_completer_mock = MagicMock(spec=Completer)
|
|
xonsh_completer_mock.complete.return_value = set(), 0
|
|
|
|
ptk_completer = PromptToolkitCompleter(xonsh_completer_mock, None, None)
|
|
ptk_completer.reserve_space = lambda: None
|
|
ptk_completer.suggestion_completion = lambda _, __: None
|
|
|
|
monkeypatch.setattr(xession, "aliases", Aliases(gb=["git branch"]))
|
|
|
|
list(ptk_completer.get_completions(Document(code, index), MagicMock()))
|
|
mock_call = xonsh_completer_mock.complete.call_args
|
|
args, kwargs = mock_call
|
|
expected_args["self"] = None
|
|
expected_args["ctx"] = None
|
|
assert (
|
|
signature(Completer.complete).bind(None, *args, **kwargs).arguments
|
|
== expected_args
|
|
)
|