mirror of https://github.com/aliasrobotics/cai.git
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
"""Tests for REPL follow-up prompts (y/N after prompt_toolkit)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from cai.repl.ui.tty_input import (
|
|
normalize_repl_line,
|
|
read_repl_line,
|
|
read_repl_yes_no,
|
|
)
|
|
|
|
|
|
class TestNormalizeReplLine:
|
|
def test_strips_carriage_return_from_enter(self):
|
|
assert normalize_repl_line("y\r") == "y"
|
|
|
|
def test_strips_whitespace(self):
|
|
assert normalize_repl_line(" yes \r\n") == "yes"
|
|
|
|
|
|
class TestReadReplYesNo:
|
|
@patch("prompt_toolkit.prompt", return_value="y\r")
|
|
@patch("cai.repl.ui.tty_input._restore_tty_after_prompt")
|
|
def test_yes_with_carriage_return(self, _restore, _prompt):
|
|
console = MagicMock()
|
|
assert read_repl_yes_no(console, "Continue") is True
|
|
|
|
@patch("prompt_toolkit.prompt", return_value="")
|
|
@patch("cai.repl.ui.tty_input._restore_tty_after_prompt")
|
|
def test_empty_defaults_to_no(self, _restore, _prompt):
|
|
console = MagicMock()
|
|
assert read_repl_yes_no(console, "Continue", default=False) is False
|
|
|
|
@patch("prompt_toolkit.prompt", return_value="n")
|
|
@patch("cai.repl.ui.tty_input._restore_tty_after_prompt")
|
|
def test_no_answer(self, _restore, _prompt):
|
|
console = MagicMock()
|
|
assert read_repl_yes_no(console, "Continue") is False
|
|
|
|
|
|
class TestReadReplLine:
|
|
@patch("prompt_toolkit.prompt", return_value="RESET\r")
|
|
@patch("cai.repl.ui.tty_input._restore_tty_after_prompt")
|
|
def test_restores_tty_around_prompt(self, restore, _prompt):
|
|
console = MagicMock()
|
|
assert read_repl_line(console, "> ", markup=False) == "RESET"
|
|
assert restore.call_count == 2
|