xonsh/tests/test_readline_shell.py
Noorhteen Raja NJ e25ab34e32
Add test for readline shell (#4447)
* fix: ReadlineShell pass stdin/out to Cmd's constructor

* test: add test for readline shell

* chore: increase required coverage

* test:

* fix: failing ci pipeline

* fix: failing ci test
2021-09-02 22:15:20 +03:00

46 lines
1.3 KiB
Python

import pytest
from xonsh.completers.tools import RichCompletion
from xonsh.readline_shell import _render_completions
@pytest.mark.parametrize(
"prefix, completion, prefix_len, readline_completion",
[
("", "a", 0, "a"),
("a", "b", 0, "ab"),
("a", "b", 1, "b"),
("adc", "bc", 2, "abc"),
("", RichCompletion("x", 0), 0, "x"),
("", RichCompletion("x", 0, "aaa", "aaa"), 0, "x"),
("a", RichCompletion("b", 1), 0, "b"),
("a", RichCompletion("b", 0), 1, "ab"),
("a", RichCompletion("b"), 0, "ab"),
("a", RichCompletion("b"), 1, "b"),
],
)
def test_render_completions(prefix, completion, prefix_len, readline_completion):
assert _render_completions({completion}, prefix, prefix_len) == [
readline_completion
]
@pytest.mark.parametrize(
"line, exp",
[
[repr("hello"), "hello"],
["2 * 3", "6"],
],
)
def test_rl_prompt_cmdloop(line, exp, readline_shell, capsys):
shell = readline_shell
shell.use_rawinput = False
shell.stdin.write(f"{line}\nexit\n") # note: terminate with '\n'
shell.stdin.seek(0)
shell.cmdloop()
# xonsh, doesn't write all its output to shell.stdout
# so capture sys.stdout
out, err = capsys.readouterr()
# sometimes the output has ansii color codes
assert exp in out.strip()