diff options
Diffstat (limited to 'tests/test_cmd2.py')
| -rw-r--r-- | tests/test_cmd2.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/test_cmd2.py b/tests/test_cmd2.py index d2f9c934..2c3a6253 100644 --- a/tests/test_cmd2.py +++ b/tests/test_cmd2.py @@ -1097,11 +1097,13 @@ def test_default_to_shell_failure(capsys): def test_ansi_prompt_not_esacped(base_app): + from cmd2.rl_utils import rl_make_safe_prompt prompt = '(Cmd) ' - assert base_app._surround_ansi_escapes(prompt) == prompt + assert rl_make_safe_prompt(prompt) == prompt def test_ansi_prompt_escaped(): + from cmd2.rl_utils import rl_make_safe_prompt app = cmd2.Cmd() color = Fore.CYAN prompt = 'InColor' @@ -1110,7 +1112,7 @@ def test_ansi_prompt_escaped(): readline_hack_start = "\x01" readline_hack_end = "\x02" - readline_safe_prompt = app._surround_ansi_escapes(color_prompt) + readline_safe_prompt = rl_make_safe_prompt(color_prompt) assert prompt != color_prompt assert readline_safe_prompt.startswith(readline_hack_start + color + readline_hack_end) assert readline_safe_prompt.endswith(readline_hack_start + Fore.RESET + readline_hack_end) |