def test_kb(self):
prompt = CheckboxPrompt(message="", choices=self.choices)
prompt._invalid = True
@prompt.register_kb("b")
def test(_):
pass
test("") # type: ignore
self.assertEqual(prompt._invalid, False)
def test_validator(self):
prompt = CheckboxPrompt(
message="",
choices=self.choices,
validate=lambda x: len(x) > 2,
invalid_message="hello",
)
with patch("prompt_toolkit.utils.Event") as mock:
self.assertEqual(prompt._invalid, False)
event = mock.return_value
prompt._handle_enter(event)
self.assertEqual(prompt._invalid, True)
self.assertEqual(prompt._invalid_message, "hello")
def test_checkbox_prompt_message(self):
prompt = CheckboxPrompt(
message="Select something",
choices=self.choices,
instruction="TAB",
)
self.assertEqual(
prompt._get_prompt_message(),
[
("class:questionmark", "?"),
("class:question", " Select something"),
("class:instruction", " TAB "),
],
)
def test_checkbox_prompt(self):
prompt = CheckboxPrompt(
message="Select something",
choices=self.choices,
default="boy&girl",
style=None,
vi_mode=False,
qmark="!",
pointer="<",
instruction="TAB",
)
self.assertEqual(prompt._editing_mode, EditingMode.EMACS)
self.assertIsInstance(prompt.content_control,
InquirerPyCheckboxControl)
self.assertIsInstance(prompt._kb, KeyBindings)
self.assertIsInstance(prompt._style, Style)
self.assertEqual(prompt._message, "Select something")
self.assertEqual(prompt._qmark, "!")
self.assertEqual(prompt.instruction, "TAB")
def test_checkbox_enter_empty(self):
prompt = CheckboxPrompt(message="", choices=["haah", "haha", "what"])
with patch("prompt_toolkit.utils.Event") as mock:
event = mock.return_value
prompt._handle_enter(event)
self.assertEqual(prompt.status["result"], [])
def test_checkbox_bindings(self):
prompt = CheckboxPrompt(message="", choices=self.choices)
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)
self.assertEqual(prompt.content_control.selected_choice_index, 3)
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, 3)
prompt._handle_up(None)
self.assertEqual(prompt.content_control.selected_choice_index, 1)
self.assertEqual(prompt.status["result"], None)
self.assertEqual(prompt.status["answered"], False)
self.assertEqual(prompt.status["skipped"], False)
with patch("prompt_toolkit.utils.Event") as mock:
event = mock.return_value
prompt._handle_enter(event)
self.assertEqual(prompt.status["result"], ["mix"])
self.assertEqual(prompt.status["answered"], True)
prompt._handle_toggle_choice(None)
self.assertEqual(
prompt.content_control.choices,
[
{
"enabled": False,
"name": "boy",
"value": "boy"
},
{
"enabled": True,
"name": "girl",
"value": "girl"
},
{
"enabled": False,
"name": "---------------",
"value": ANY
},
{
"enabled": True,
"name": "mix",
"value": "boy&girl"
},
],
)
prompt._handle_toggle_all(None)
self.assertEqual(
prompt.content_control.choices,
[
{
"enabled": True,
"name": "boy",
"value": "boy"
},
{
"enabled": False,
"name": "girl",
"value": "girl"
},
{
"enabled": False,
"name": "---------------",
"value": ANY
},
{
"enabled": False,
"name": "mix",
"value": "boy&girl"
},
],
)
prompt._handle_toggle_all(None, True)
self.assertEqual(
prompt.content_control.choices,
[
{
"enabled": True,
"name": "boy",
"value": "boy"
},
{
"enabled": True,
"name": "girl",
"value": "girl"
},
{
"enabled": False,
"name": "---------------",
"value": ANY
},
{
"enabled": True,
"name": "mix",
"value": "boy&girl"
},
],
)
def test_minimum_args(self):
CheckboxPrompt(message="yes", choices=self.choices)