def test_handle_enter_multi(self) -> None:
prompt = ListPrompt(
message="Select a fruit",
choices=self.choices,
default="watermelon",
style=InquirerPyStyle({"pointer": "#61afef"}),
vi_mode=True,
qmark="[?]",
pointer=">",
instruction="(j/k)",
multiselect=True,
)
self.assertEqual(prompt.status, {
"result": None,
"answered": False,
"skipped": False
})
with patch("prompt_toolkit.utils.Event") as mock:
event = mock.return_value
prompt._handle_enter(event)
self.assertEqual(prompt.status, {
"result": ["melon"],
"answered": True,
"skipped": False
})
def test_prompt_message_with_cursor(self):
prompt = ListPrompt(message="Select one:", choices=[1, 2, 3])
self.assertEqual(
prompt._get_prompt_message_with_cursor(),
[
("class:questionmark", "?"),
("class:question", " Select one:"),
("class:instruction", " "),
("[SetCursorPosition]", ""),
("", " "),
],
)
def test_resolver_transformer(self):
prompt = ListPrompt(
message="Select one", choices=["1", "2", "3"], transformer=lambda x: x * 2
)
self.assertEqual(
prompt._get_prompt_message(),
[
("class:questionmark", "?"),
("class:question", " Select one"),
("class:instruction", " "),
],
)
prompt.status["result"] = "1"
prompt.status["answered"] = True
self.assertEqual(
prompt._get_prompt_message(),
[
("class:answermark", "?"),
("class:answered_question", " Select one"),
("class:answer", " 11"),
],
)
prompt = InputPrompt(
message="Select one",
transformer=lambda x: x * 2,
)
prompt.status["result"] = "2"
prompt.status["answered"] = True
self.assertEqual(
prompt._get_prompt_message(),
[
("class:answermark", "?"),
("class:answered_question", " Select one"),
("class:answer", " 22"),
],
)
prompt = SecretPrompt(
message="Select one",
transformer=lambda x: x * 2,
)
prompt.status["result"] = "2"
prompt.status["answered"] = True
self.assertEqual(
prompt._get_prompt_message(),
[
("class:answermark", "?"),
("class:answered_question", " Select one"),
("class:answer", " 22"),
],
)
def test_choice_combination(self):
prompt = ListPrompt(message="Test combo", choices=["hello"])
self.assertEqual(prompt._qmark, "?")
self.assertEqual(prompt.instruction, "")
self.assertRaises(InvalidArgument, ListPrompt, "",
[Separator(), Separator()])
def test_list_prompt(self, mocked_term):
mocked_term.return_value = (24, 80)
message = 15 * "i"
qmark = "[?]"
instruction = 2 * "i"
prompt = ListPrompt(
message=message,
choices=self.choices,
default="watermelon",
style=InquirerPyStyle({"pointer": "#61afef"}),
vi_mode=True,
qmark=qmark,
pointer=">",
instruction=instruction,
show_cursor=True,
wrap_lines=True,
border=True,
)
self.assertEqual(prompt._editing_mode, EditingMode.VI)
self.assertIsInstance(prompt.content_control, InquirerPyListControl)
self.assertIsInstance(prompt._kb, KeyBindings)
self.assertIsInstance(prompt._style, Style)
self.assertEqual(prompt._message, message)
self.assertEqual(prompt._qmark, qmark)
self.assertEqual(prompt.instruction, instruction)
self.assertTrue(prompt._border, True)
def test_prompt_handle_toggle_all_no_multiselect(self) -> None:
prompt = ListPrompt(message="Select one:", choices=[1, 2, 3])
self.assertEqual(
prompt.content_control.choices,
[
{
"enabled": False,
"name": "1",
"value": 1
},
{
"enabled": False,
"name": "2",
"value": 2
},
{
"enabled": False,
"name": "3",
"value": 3
},
],
)
prompt._handle_toggle_all(None)
self.assertEqual(
prompt.content_control.choices,
[
{
"enabled": False,
"name": "1",
"value": 1
},
{
"enabled": False,
"name": "2",
"value": 2
},
{
"enabled": False,
"name": "3",
"value": 3
},
],
)
def test_list_prompt_message(self):
prompt = ListPrompt(
message="Select a fruit",
choices=self.choices,
default="watermelon",
style=InquirerPyStyle({"pointer": "#61afef"}),
vi_mode=True,
qmark="[?]",
pointer=">",
instruction="(j/k)",
)
self.assertEqual(
prompt._get_prompt_message(),
[
("class:questionmark", "[?]"),
("class:question", " Select a fruit"),
("class:instruction", " (j/k) "),
],
)
def test_wrap_lines_offset(self, mocked_term):
mocked_term.return_value = (24, 80)
message = 15 * "i"
qmark = "[?]"
instruction = 3 * "i"
prompt = ListPrompt(
message=message,
choices=self.choices,
default="watermelon",
style=InquirerPyStyle({"pointer": "#61afef"}),
vi_mode=True,
qmark=qmark,
pointer=">",
instruction=instruction,
show_cursor=True,
wrap_lines=True,
)
self.assertEqual(
prompt.extra_line_count,
(len(qmark) + 1 + len(message) + 1 + len(instruction) + 1) // 24,
)
prompt = ListPrompt(
message=message,
choices=self.choices,
default="watermelon",
style=InquirerPyStyle({"pointer": "#61afef"}),
vi_mode=True,
qmark=qmark,
pointer=">",
instruction=instruction,
show_cursor=False,
wrap_lines=True,
)
self.assertEqual(
prompt.extra_line_count,
(len(qmark) + 1 + len(message) + 1 + len(instruction) + 1 - 1) //
24,
)
prompt._wrap_lines = False
self.assertEqual(prompt.extra_line_count, 0)
def test_prompt_handle_toggle_choice_no_multiselect(self):
prompt = ListPrompt(message="Select one:",
choices=[1, 2, 3],
multiselect=False)
self.assertEqual(
prompt.content_control.selection,
{
"enabled": False,
"name": "1",
"value": 1
},
)
prompt._handle_toggle_choice(None)
self.assertEqual(
prompt.content_control.selection,
{
"enabled": False,
"name": "1",
"value": 1
},
)
def test_list_enter_empty(self):
prompt = ListPrompt(
message="",
choices=["haha", "haah", "what", "I don't know"],
)
with patch("prompt_toolkit.utils.Event") as mock:
event = mock.return_value
prompt._handle_enter(event)
self.assertEqual(prompt.status["result"], "haha")
prompt = ListPrompt(
message="",
choices=["haha", "haah", "what", "I don't know"],
multiselect=True,
)
with patch("prompt_toolkit.utils.Event") as mock:
event = mock.return_value
prompt._handle_enter(event)
self.assertEqual(prompt.status["result"], ["haha"])
prompt.content_control.choices[1]["enabled"] = True
prompt._handle_enter(event)
self.assertEqual(prompt.status["result"], ["haah"])
def test_handle_up_no_cycle(self):
prompt = ListPrompt(
message="Select one:",
choices=[Separator(), 1, 2, 3, Separator()],
cycle=False,
)
self.assertEqual(prompt.content_control.selected_choice_index, 1)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 1)
prompt = ListPrompt(
message="Select one:",
choices=[1, 2, 3, Separator()],
cycle=False,
)
self.assertEqual(prompt.content_control.selected_choice_index, 0)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 0)
def test_handle_down(self) -> None:
prompt = ListPrompt(
message="Select one:",
choices=[1, 2, 3, Separator()],
)
self.assertEqual(prompt.content_control.selected_choice_index, 0)
prompt._handle_down(None)
self.assertEqual(prompt.content_control.selected_choice_index, 1)
prompt._handle_down(None)
prompt._handle_down(None)
self.assertEqual(prompt.content_control.selected_choice_index, 0)
def test_prompt_execute(self, mocked_run):
mocked_run.return_value = "hello"
result = ListPrompt("hello world", ["yes", "no"]).execute()
self.assertEqual(result, "hello")
result = ListPrompt("hello world", ["yes", "no"],
filter=lambda _: "no").execute()
self.assertEqual(result, "no")
mocked_run.return_value = INQUIRERPY_KEYBOARD_INTERRUPT
prompt = ListPrompt("hello world", ["yes", "no"],
filter=lambda _: "no")
self.assertRaises(KeyboardInterrupt, prompt.execute)
mocked_run.return_value = INQUIRERPY_KEYBOARD_INTERRUPT
prompt = ListPrompt("hello world", ["yes", "no"])
try:
result = prompt.execute(raise_keyboard_interrupt=False)
except KeyboardInterrupt:
self.assertFalse(prompt._is_rasing_kbi())
else:
self.fail("should raise kbi")
def test_handle_enter_validator(self):
prompt = ListPrompt(
message="Select a fruit",
choices=self.choices,
default="watermelon",
style=InquirerPyStyle({"pointer": "#61afef"}),
vi_mode=True,
qmark="[?]",
pointer=">",
instruction="(j/k)",
validate=lambda result: result != "watermelon",
)
self.assertFalse(prompt._invalid)
self.assertEqual(prompt.status, {
"result": None,
"answered": False,
"skipped": False
})
with patch("prompt_toolkit.utils.Event") as mock:
event = mock.return_value
prompt._handle_enter(event)
self.assertEqual(prompt.status, {
"result": None,
"answered": False,
"skipped": False
})
self.assertTrue(prompt._invalid)
prompt.content_control.selected_choice_index = 0
with patch("prompt_toolkit.utils.Event") as mock:
event = mock.return_value
prompt._handle_enter(event)
self.assertEqual(prompt.status, {
"result": "apple",
"answered": True,
"skipped": False
})
def test_handle_up(self):
prompt = ListPrompt(message="Select one:",
choices=[1, 2, Separator(), 3])
self.assertEqual(prompt.content_control.selected_choice_index, 0)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 3)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 1)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 0)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 3)
prompt = ListPrompt(message="Select one:",
choices=[1, 2, 3, Separator()])
self.assertEqual(prompt.content_control.selected_choice_index, 0)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 2)
def test_separator_movement(self):
prompt = ListPrompt(message="..", choices=[Separator("hello"), "yes"])
self.assertEqual(prompt.content_control.selected_choice_index, 1)
prompt._handle_down(None)
self.assertEqual(prompt.content_control.selected_choice_index, 1)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 1)
prompt = ListPrompt(
message="..",
choices=[Separator("hello"), "yes",
Separator(), "no"])
self.assertEqual(prompt.content_control.selected_choice_index, 1)
prompt._handle_down(None)
self.assertEqual(prompt.content_control.selected_choice_index, 3)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 1)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 3)
def test_list_bindings(self):
prompt = ListPrompt(
message="Select a fruit",
choices=self.choices,
default="watermelon",
style=InquirerPyStyle({"pointer": "#61afef"}),
vi_mode=True,
qmark="[?]",
pointer=">",
instruction="(j/k)",
)
self.assertEqual(prompt.content_control.selected_choice_index, 2)
prompt._handle_down(None)
self.assertEqual(prompt.content_control.selected_choice_index, 0)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 2)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 1)
prompt._handle_down(None)
self.assertEqual(prompt.content_control.selected_choice_index, 2)
self.assertEqual(prompt.status, {
"result": None,
"answered": False,
"skipped": False
})
with patch("prompt_toolkit.utils.Event") as mock:
event = mock.return_value
prompt._handle_enter(event)
self.assertEqual(prompt.status, {
"result": "melon",
"answered": True,
"skipped": False
})