mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-06 17:30:59 +01:00
108 lines
2.7 KiB
Python
108 lines
2.7 KiB
Python
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
|
|
from xonsh.environ import Env
|
|
from xonsh.prompt.base import PromptFormatter
|
|
|
|
|
|
@pytest.fixture
|
|
def formatter(xonsh_builtins):
|
|
return PromptFormatter()
|
|
|
|
|
|
@pytest.mark.parametrize('fields', [{
|
|
'a_string': 'cat',
|
|
'none': (lambda: None),
|
|
'f': (lambda: 'wakka'),
|
|
}])
|
|
@pytest.mark.parametrize('inp, exp', [
|
|
('my {a_string}', 'my cat'),
|
|
('my {none}{a_string}', 'my cat'),
|
|
('{f} jawaka', 'wakka jawaka'),
|
|
])
|
|
def test_format_prompt(inp, exp, fields, formatter):
|
|
obs = formatter(template=inp, fields=fields)
|
|
assert exp == obs
|
|
|
|
|
|
@pytest.mark.parametrize('fields', [{
|
|
'a_string': 'cats',
|
|
'a_number': 7,
|
|
'empty': '',
|
|
'current_job': (lambda: 'sleep'),
|
|
'none': (lambda: None),
|
|
}])
|
|
@pytest.mark.parametrize('inp, exp', [
|
|
('{a_number:{0:^3}}cats', ' 7 cats'),
|
|
('{current_job:{} | }xonsh', 'sleep | xonsh'),
|
|
('{none:{} | }{a_string}{empty:!}', 'cats!'),
|
|
('{none:{}}', ''),
|
|
('{{{a_string:{{{}}}}}}', '{{cats}}'),
|
|
('{{{none:{{{}}}}}}', '{}'),
|
|
])
|
|
def test_format_prompt_with_format_spec(inp, exp, fields, formatter):
|
|
obs = formatter(template=inp, fields=fields)
|
|
assert exp == obs
|
|
|
|
|
|
def test_format_prompt_with_broken_template(formatter):
|
|
for p in ('{user', '{user}{hostname'):
|
|
assert formatter(p) == p
|
|
|
|
# '{{user' will be parsed to '{user'
|
|
for p in ('{{user}', '{{user'):
|
|
assert 'user' in formatter(p)
|
|
|
|
|
|
@pytest.mark.parametrize('inp', [
|
|
'{user',
|
|
'{{user',
|
|
'{{user}',
|
|
'{user}{hostname',
|
|
])
|
|
def test_format_prompt_with_broken_template_in_func(inp, formatter):
|
|
# '{{user' will be parsed to '{user'
|
|
assert '{user' in formatter(lambda: inp)
|
|
|
|
|
|
def test_format_prompt_with_invalid_func(formatter, xonsh_builtins):
|
|
xonsh_builtins.__xonsh_env__ = Env()
|
|
|
|
def p():
|
|
foo = bar # raises exception # noqa
|
|
return '{user}'
|
|
|
|
assert isinstance(formatter(p), str)
|
|
|
|
|
|
def test_format_prompt_with_func_that_raises(formatter, capsys, xonsh_builtins):
|
|
xonsh_builtins.__xonsh_env__ = Env()
|
|
template = 'tt {zerodiv} tt'
|
|
exp = 'tt (ERROR:zerodiv) tt'
|
|
fields = {'zerodiv': lambda: 1/0}
|
|
obs = formatter(template, fields)
|
|
assert exp == obs
|
|
out, err = capsys.readouterr()
|
|
assert 'prompt: error' in err
|
|
|
|
|
|
def test_promptformatter_cache(formatter):
|
|
spam = Mock()
|
|
template = '{spam} and {spam}'
|
|
fields = {'spam': spam}
|
|
|
|
formatter(template, fields)
|
|
|
|
assert spam.call_count == 1
|
|
|
|
|
|
def test_promptformatter_clears_cache(formatter):
|
|
spam = Mock()
|
|
template = '{spam} and {spam}'
|
|
fields = {'spam': spam}
|
|
|
|
formatter(template, fields)
|
|
formatter(template, fields)
|
|
|
|
assert spam.call_count == 2
|