mirror of
https://github.com/aljazceru/goose.git
synced 2025-12-28 03:24:21 +01:00
50 lines
1.1 KiB
Python
50 lines
1.1 KiB
Python
import pytest
|
|
from goose.cli.prompt.overwrite_session_prompt import OverwriteSessionPrompt
|
|
|
|
|
|
@pytest.fixture
|
|
def prompt():
|
|
return OverwriteSessionPrompt()
|
|
|
|
|
|
def test_init(prompt):
|
|
assert prompt.choices == {
|
|
"yes": "Overwrite the existing session",
|
|
"no": "Pick a new session name",
|
|
"resume": "Resume the existing session",
|
|
}
|
|
assert prompt.default == "resume"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"choice, expected",
|
|
[
|
|
("", False),
|
|
("invalid", False),
|
|
("n", True),
|
|
("N", True),
|
|
("no", True),
|
|
("NO", True),
|
|
("r", True),
|
|
("R", True),
|
|
("resume", True),
|
|
("RESUME", True),
|
|
("y", True),
|
|
("Y", True),
|
|
("yes", True),
|
|
("YES", True),
|
|
],
|
|
)
|
|
def test_check_choice(prompt, choice, expected):
|
|
assert prompt.check_choice(choice) == expected
|
|
|
|
|
|
def test_instantiation():
|
|
prompt = OverwriteSessionPrompt()
|
|
assert prompt.choices == {
|
|
"yes": "Overwrite the existing session",
|
|
"no": "Pick a new session name",
|
|
"resume": "Resume the existing session",
|
|
}
|
|
assert prompt.default == "resume"
|