import asyncio import unittest from typing import Callable, NamedTuple from unittest.mock import ANY, MagicMock, call, patch from pfzy.score import fzy_scorer, substr_scorer from prompt_toolkit.application.application import Application from prompt_toolkit.buffer import Buffer from prompt_toolkit.layout.layout import Layout from InquirerPy.base.complex import BaseComplexPrompt from InquirerPy.enum import INQUIRERPY_POINTER_SEQUENCE from InquirerPy.prompts.fuzzy import FuzzyPrompt, InquirerPyFuzzyControl class AsyncMock(MagicMock): async def __call__(self, *args, **kwargs): return super().__call__(*args, **kwargs) class TestFuzzy(unittest.TestCase): content_control = InquirerPyFuzzyControl( choices=["haah", "haha", "what", "waht", "weaht"], pointer=INQUIRERPY_POINTER_SEQUENCE, marker=INQUIRERPY_POINTER_SEQUENCE, current_text=lambda: "yes", max_lines=80, session_result=None, multiselect=False, marker_pl=" ", match_exact=False, ) @patch("InquirerPy.utils.shutil.get_terminal_size") def setUp(self, mocked_term): mocked_term.return_value = (24, 80) self.prompt = FuzzyPrompt( message="Select one of them", choices=[ "haah", "haha", "what", "waht", {"name": "weaht", "value": "weaht", "enabled": True}, ], ) def test_content_control_init(self): self.assertEqual(self.content_control._pointer, INQUIRERPY_POINTER_SEQUENCE) self.assertEqual(self.content_control._marker, INQUIRERPY_POINTER_SEQUENCE) self.assertEqual(self.content_control._current_text(), "yes") self.assertEqual( self.content_control.choices, [ { "enabled": False, "index": 0, "indices": [], "name": "haah", "value": "haah", }, { "enabled": False, "index": 1, "indices": [], "name": "haha", "value": "haha", }, { "enabled": False, "index": 2, "indices": [], "name": "what", "value": "what", }, { "enabled": False, "index": 3, "indices": [], "name": "waht", "value": "waht", }, { "enabled": False, "index": 4, "indices": [], "name": "weaht", "value": "weaht", }, ], ) self.assertEqual( self.content_control._filtered_choices, [ { "enabled": False, "index": 0, "indices": [], "name": "haah", "value": "haah", }, { "enabled": False, "index": 1, "indices": [], "name": "haha", "value": "haha", }, { "enabled": False, "index": 2, "indices": [], "name": "what", "value": "what", }, { "enabled": False, "index": 3, "indices": [], "name": "waht", "value": "waht", }, { "enabled": False, "index": 4, "indices": [], "name": "weaht", "value": "weaht", }, ], ) self.assertEqual(self.content_control.selected_choice_index, 0) self.assertEqual( self.content_control.selection, { "index": 0, "name": "haah", "enabled": False, "value": "haah", "indices": [], }, ) self.assertEqual(self.content_control.choice_count, 5) def test_content_control_text(self): self.content_control.choices[0]["enabled"] = True self.assertEqual( self.content_control._get_formatted_choices(), [ ("class:pointer", "❯"), ("class:marker", "❯"), ("[SetCursorPosition]", ""), ("class:pointer", "haah"), ("", "\n"), ("class:pointer", " "), ("class:marker", " "), ("", "haha"), ("", "\n"), ("class:pointer", " "), ("class:marker", " "), ("", "what"), ("", "\n"), ("class:pointer", " "), ("class:marker", " "), ("", "waht"), ("", "\n"), ("class:pointer", " "), ("class:marker", " "), ("", "weaht"), ], ) self.content_control.choices[0]["enabled"] = False def test_prompt_filter1(self): content_control = InquirerPyFuzzyControl( choices=["meat", "what", "whaaah", "weather", "haha"], pointer=INQUIRERPY_POINTER_SEQUENCE, marker=INQUIRERPY_POINTER_SEQUENCE, current_text=lambda: "wh", max_lines=80, session_result=None, multiselect=False, marker_pl=" ", match_exact=False, ) self.assertEqual( content_control._filtered_choices, [ { "enabled": False, "index": 0, "indices": [], "name": "meat", "value": "meat", }, { "enabled": False, "index": 1, "indices": [], "name": "what", "value": "what", }, { "enabled": False, "index": 2, "indices": [], "name": "whaaah", "value": "whaaah", }, { "enabled": False, "index": 3, "indices": [], "name": "weather", "value": "weather", }, { "enabled": False, "index": 4, "indices": [], "name": "haha", "value": "haha", }, ], ) result = asyncio.run(content_control._filter_choices(0.0)) self.assertEqual( result, [ { "enabled": False, "index": 1, "indices": [0, 1], "name": "what", "value": "what", }, { "enabled": False, "index": 2, "indices": [0, 1], "name": "whaaah", "value": "whaaah", }, { "enabled": False, "index": 3, "indices": [0, 4], "name": "weather", "value": "weather", }, ], ) content_control._filtered_choices = result self.assertEqual( content_control._get_formatted_choices(), [ ("class:pointer", "❯"), ("class:marker", " "), ("[SetCursorPosition]", ""), ("class:fuzzy_match", "w"), ("class:fuzzy_match", "h"), ("class:pointer", "a"), ("class:pointer", "t"), ("", "\n"), ("class:pointer", " "), ("class:marker", " "), ("class:fuzzy_match", "w"), ("class:fuzzy_match", "h"), ("", "a"), ("", "a"), ("", "a"), ("", "h"), ("", "\n"), ("class:pointer", " "), ("class:marker", " "), ("class:fuzzy_match", "w"), ("", "e"), ("", "a"), ("", "t"), ("class:fuzzy_match", "h"), ("", "e"), ("", "r"), ], ) self.assertEqual(content_control.choice_count, 3) self.assertEqual(content_control.selected_choice_index, 0) def test_prompt_filter2(self): content_control = InquirerPyFuzzyControl( choices=["meat", "what", "whaaah", "weather", "haha"], pointer=INQUIRERPY_POINTER_SEQUENCE, marker=INQUIRERPY_POINTER_SEQUENCE, current_text=lambda: "", max_lines=80, session_result=None, multiselect=False, marker_pl=" ", match_exact=False, ) content_control.choices[0]["indices"] = [1, 2, 3] asyncio.run(content_control._filter_choices(0.0)) self.assertEqual( content_control._filtered_choices, [ { "enabled": False, "index": 0, "indices": [], "name": "meat", "value": "meat", }, { "enabled": False, "index": 1, "indices": [], "name": "what", "value": "what", }, { "enabled": False, "index": 2, "indices": [], "name": "whaaah", "value": "whaaah", }, { "enabled": False, "index": 3, "indices": [], "name": "weather", "value": "weather", }, { "enabled": False, "index": 4, "indices": [], "name": "haha", "value": "haha", }, ], ) @patch("InquirerPy.prompts.fuzzy.InquirerPyFuzzyControl") @patch("InquirerPy.prompts.fuzzy.calculate_height") @patch("InquirerPy.utils.shutil.get_terminal_size") def test_constructor(self, mocked_term, mocked_height, mocked_control): mocked_term.return_value = (24, 80) mocked_height.return_value = (80, 80) message = 15 * "i" qmark = "[?]" instruction = 2 * "i" prompt = FuzzyPrompt( message=message, qmark=qmark, instruction=instruction, choices=[ "haah", "haha", "what", "waht", {"name": "weaht", "value": "weaht", "enabled": True}, ], ) self.assertEqual(prompt._instruction, instruction) self.assertEqual(prompt._multiselect, False) self.assertEqual(prompt._prompt, INQUIRERPY_POINTER_SEQUENCE) self.assertFalse(prompt._border) self.assertEqual(prompt._info, True) self.assertIsInstance(prompt._buffer, Buffer) self.assertIsInstance(prompt._layout, Layout) self.assertIsInstance(prompt._application, Application) mocked_control.assert_called_with( choices=[ "haah", "haha", "what", "waht", {"name": "weaht", "value": "weaht", "enabled": True}, ], pointer=INQUIRERPY_POINTER_SEQUENCE, marker=INQUIRERPY_POINTER_SEQUENCE, current_text=ANY, max_lines=80, session_result=None, multiselect=False, marker_pl=" ", match_exact=False, ) prompt = FuzzyPrompt( message=message, qmark=qmark, instruction=instruction, choices=[ "haah", "haha", "what", "waht", {"name": "weaht", "value": "weaht", "enabled": True}, ], border=True, ) mocked_control.assert_called_with( choices=[ "haah", "haha", "what", "waht", {"name": "weaht", "value": "weaht", "enabled": True}, ], pointer=INQUIRERPY_POINTER_SEQUENCE, marker=INQUIRERPY_POINTER_SEQUENCE, current_text=ANY, max_lines=80, session_result=None, multiselect=False, marker_pl=" ", match_exact=False, ) def test_prompt_after_input(self): prompt = FuzzyPrompt( message="Select one of them", choices=["haah", "haha", "what", "waht", "weaht"], ) self.assertEqual( prompt._generate_after_input(), [("", " "), ("class:fuzzy_info", "5/5")] ) prompt = FuzzyPrompt( message="Select one of them", choices=["haah", "haha", "what", "waht", "weaht"], info=False, ) self.assertEqual(prompt._generate_after_input(), []) def test_prompt_after_input2(self): prompt = FuzzyPrompt( message="", choices=["1", "2", "3"], match_exact=True, exact_symbol=" *" ) self.assertEqual( prompt._generate_after_input(), [("", " "), ("class:fuzzy_info", "3/3"), ("class:fuzzy_info", " *")], ) def test_prompt_before_input(self): prompt = FuzzyPrompt( message="Select one of them", choices=["haah", "haha", "what", "waht", "weaht"], ) self.assertEqual( prompt._generate_before_input(), [("class:fuzzy_prompt", "❯ ")] ) prompt = FuzzyPrompt( message="Select one of them", choices=["haah", "haha", "what", "waht", "weaht"], prompt=">", ) self.assertEqual( prompt._generate_before_input(), [("class:fuzzy_prompt", "> ")] ) @patch("asyncio.create_task") def test_prompt_on_text_changed(self, mocked): self.assertEqual(self.prompt.content_control.selected_choice_index, 0) self.prompt.content_control.selected_choice_index = 4 self.prompt._buffer.text = "ha" mocked.assert_called() def test_prompt_filter_callback(self): class Hello(NamedTuple): cancelled: Callable result: Callable hello = Hello(cancelled=lambda: True, result=lambda: []) self.prompt._filter_callback(hello) self.assertEqual( self.prompt.content_control._filtered_choices, [ { "enabled": False, "index": 0, "indices": [], "name": "haah", "value": "haah", }, { "enabled": False, "index": 1, "indices": [], "name": "haha", "value": "haha", }, { "enabled": False, "index": 2, "indices": [], "name": "what", "value": "what", }, { "enabled": False, "index": 3, "indices": [], "name": "waht", "value": "waht", }, { "enabled": False, "index": 4, "indices": [], "name": "weaht", "value": "weaht", }, ], ) self.assertEqual(self.prompt.content_control.selected_choice_index, 0) self.prompt.content_control.selected_choice_index = 4 hello = Hello(cancelled=lambda: False, result=lambda: []) self.prompt._filter_callback(hello) self.prompt.content_control._get_formatted_choices() self.assertEqual(self.prompt.content_control._filtered_choices, []) self.assertEqual(self.prompt.content_control.selected_choice_index, 0) self.prompt.content_control.selected_choice_index = -1 hello = Hello( cancelled=lambda: False, result=lambda: [ { "enabled": False, "index": i, "indices": [], "name": "weaht", "value": "weaht", } for i in range(3) ], ) self.prompt._filter_callback(hello) self.prompt.content_control._get_formatted_choices() self.assertEqual(self.prompt.content_control.selected_choice_index, 0) self.assertEqual(self.prompt.content_control._first_line, 0) self.assertEqual(self.prompt.content_control._last_line, 3) self.prompt.content_control.selected_choice_index = 5 hello = Hello( cancelled=lambda: False, result=lambda: [ { "enabled": False, "index": i, "indices": [], "name": "weaht", "value": "weaht", } for i in range(3) ], ) self.prompt._filter_callback(hello) self.prompt.content_control._get_formatted_choices() self.assertEqual(self.prompt.content_control.selected_choice_index, 2) self.assertEqual(self.prompt.content_control._first_line, 0) self.assertEqual(self.prompt.content_control._last_line, 3) def test_prompt_bindings(self): self.assertEqual(self.prompt.content_control.selected_choice_index, 0) with patch("prompt_toolkit.utils.Event") as mock: event = mock.return_value self.prompt._handle_enter(event) self.assertEqual(self.prompt.status["answered"], True) self.assertEqual(self.prompt.status["result"], "haah") self.assertEqual(self.prompt.status["skipped"], False) prompt = FuzzyPrompt( message="Select one of them", choices=["haah", "haha", "what", "waht", "weaht"], multiselect=True, ) with patch("prompt_toolkit.utils.Event") as mock: event = mock.return_value prompt._handle_enter(event) self.assertEqual(prompt.status["answered"], True) self.assertEqual(prompt.status["result"], ["haah"]) self.assertEqual(prompt.status["skipped"], False) prompt.status["answered"] = False prompt.status["result"] = None prompt._handle_toggle_choice(None) prompt._handle_down(None) prompt._handle_toggle_choice(None) prompt._handle_down(None) with patch("prompt_toolkit.utils.Event") as mock: event = mock.return_value prompt._handle_enter(event) self.assertEqual(prompt.status["answered"], True) self.assertEqual(prompt.status["result"], ["haah", "haha"]) self.assertEqual(prompt.status["skipped"], False) prompt = FuzzyPrompt( message="Select one of them", choices=["haah", "haha", "what", "waht", "weaht"], multiselect=True, ) prompt.content_control._filtered_choices = [] with patch("prompt_toolkit.utils.Event") as mock: event = mock.return_value prompt._handle_enter(event) self.assertEqual(prompt.status["answered"], True) self.assertEqual(prompt.status["result"], []) @patch("asyncio.create_task") def test_prompt_validator(self, _): prompt = FuzzyPrompt( message="Select one", choices=["haha", "asa", "132132"], validate=lambda x: len(x) > 1, invalid_message="what", multiselect=True, ) self.assertEqual(prompt._invalid, False) with patch("prompt_toolkit.utils.Event") as mock: event = mock.return_value prompt._handle_enter(event) self.assertEqual(prompt._invalid_message, "what") self.assertEqual(prompt._invalid, True) prompt._on_text_changed("") self.assertEqual(prompt._invalid, False) def test_prompt_handle_toggle_no_multiselect(self): prompt = FuzzyPrompt(message="", choices=[1], multiselect=False) self.assertEqual( prompt.content_control.choices, [ { "enabled": False, "index": 0, "indices": [], "name": "1", "value": 1, } ], ) prompt._handle_toggle_choice(None) self.assertEqual( prompt.content_control.choices, [ { "enabled": False, "index": 0, "indices": [], "name": "1", "value": 1, } ], ) def test_prompt_handle_toggle_all_no_multiselect(self): prompt = FuzzyPrompt(message="", choices=[1], multiselect=False) self.assertEqual( prompt.content_control.choices, [ { "enabled": False, "index": 0, "indices": [], "name": "1", "value": 1, } ], ) prompt._handle_toggle_all(None) self.assertEqual( prompt.content_control.choices, [ { "enabled": False, "index": 0, "indices": [], "name": "1", "value": 1, } ], ) def test_prompt_handle_toggle_all_filter(self) -> None: prompt = FuzzyPrompt( message="", choices=["haha", "asdfa", "112321fd"], multiselect=True ) prompt.content_control._current_text = lambda: "haha" prompt.content_control._filtered_choices = asyncio.run( prompt.content_control._filter_choices(0.0) ) self.assertEqual( prompt.content_control._filtered_choices, [ { "enabled": False, "index": 0, "indices": [0, 1, 2, 3], "name": "haha", "value": "haha", } ], ) prompt._handle_toggle_all(None) self.assertEqual(prompt.content_control.choices[0]["enabled"], True) self.assertEqual(prompt.content_control.choices[1]["enabled"], False) self.assertEqual(prompt.content_control.choices[2]["enabled"], False) def test_prompt_handle_toggle_all(self): prompt = FuzzyPrompt( message="", choices=["haha", "asdfa", "112321fd"], multiselect=True ) self.assertEqual( prompt.content_control.choices, [ { "enabled": False, "index": 0, "indices": [], "name": "haha", "value": "haha", }, { "enabled": False, "index": 1, "indices": [], "name": "asdfa", "value": "asdfa", }, { "enabled": False, "index": 2, "indices": [], "name": "112321fd", "value": "112321fd", }, ], ) prompt._handle_toggle_all(None) self.assertEqual( prompt.content_control.choices, [ { "enabled": True, "index": 0, "indices": [], "name": "haha", "value": "haha", }, { "enabled": True, "index": 1, "indices": [], "name": "asdfa", "value": "asdfa", }, { "enabled": True, "index": 2, "indices": [], "name": "112321fd", "value": "112321fd", }, ], ) prompt._handle_toggle_all(None, True) self.assertEqual( prompt.content_control.choices, [ { "enabled": True, "index": 0, "indices": [], "name": "haha", "value": "haha", }, { "enabled": True, "index": 1, "indices": [], "name": "asdfa", "value": "asdfa", }, { "enabled": True, "index": 2, "indices": [], "name": "112321fd", "value": "112321fd", }, ], ) prompt._handle_toggle_all(None) self.assertEqual( prompt.content_control.choices, [ { "enabled": False, "index": 0, "indices": [], "name": "haha", "value": "haha", }, { "enabled": False, "index": 1, "indices": [], "name": "asdfa", "value": "asdfa", }, { "enabled": False, "index": 2, "indices": [], "name": "112321fd", "value": "112321fd", }, ], ) def test_wait_time(self): self.prompt.content_control.choices = [] self.assertEqual(self.prompt._calculate_wait_time(), 0.0) self.prompt.content_control.choices = [{} for _ in range(9)] self.assertEqual(self.prompt._calculate_wait_time(), 0.0) self.prompt.content_control.choices = [{} for _ in range(50)] self.assertEqual(self.prompt._calculate_wait_time(), 0.05) self.prompt.content_control.choices = [{} for _ in range(100)] self.assertEqual(self.prompt._calculate_wait_time(), 0.1) self.prompt.content_control.choices = [{} for _ in range(1000)] self.assertEqual(self.prompt._calculate_wait_time(), 0.2) self.prompt.content_control.choices = [{} for _ in range(10000)] self.assertEqual(self.prompt._calculate_wait_time(), 0.3) self.prompt.content_control.choices = [{} for _ in range(100000)] self.assertEqual(self.prompt._calculate_wait_time(), 0.6) self.prompt.content_control.choices = [{} for _ in range(1000000)] self.assertEqual(self.prompt._calculate_wait_time(), 1.2) def test_prompt_validator_index(self): class Hello(NamedTuple): cancelled: Callable result: Callable class App(NamedTuple): exit: Callable class Event(NamedTuple): app: NamedTuple hello = Hello(cancelled=lambda: False, result=lambda: []) self.prompt._filter_callback(hello) event = Event(App(exit=lambda result: True)) self.prompt._handle_enter(event) @patch.object(FuzzyPrompt, "_on_text_changed") def test_on_rendered(self, _): prompt = FuzzyPrompt(message="", choices=[1, 2, 3], default="yes") self.assertEqual(prompt._buffer.text, "") self.assertEqual(prompt._buffer.cursor_position, 0) prompt._on_rendered(None) self.assertEqual(prompt._buffer.text, "yes") self.assertEqual(prompt._buffer.cursor_position, 3) @patch.object(FuzzyPrompt, "_on_rendered") @patch.object(BaseComplexPrompt, "register_kb") def test_constructor_keybindings(self, mocked_kb, mocked_rendered): prompt = FuzzyPrompt(message="", choices=[1, 2, 3]) prompt._after_render(None) try: mocked_kb.assert_has_calls([call("space", filter=ANY)]) mocked_kb.assert_has_calls([call("j", filter=ANY)]) mocked_kb.assert_has_calls([call("k", filter=ANY)]) except AssertionError: pass else: self.fail("space/j/k kb was registered") def test_control_line_render(self) -> None: control = InquirerPyFuzzyControl( choices=[i for i in range(6)], pointer="", marker="", current_text=lambda: "", max_lines=4, session_result=None, multiselect=True, marker_pl=" ", match_exact=False, ) # range 0-4 self.assertEqual(control._last_line, 4) self.assertEqual(control._first_line, 0) control._get_formatted_choices() self.assertEqual(control._last_line, 4) self.assertEqual(control._first_line, 0) # range 1-5 control.selected_choice_index = 4 control._get_formatted_choices() self.assertEqual(control._last_line, 5) self.assertEqual(control._first_line, 1) control.selected_choice_index = 6 control._get_formatted_choices() self.assertEqual(control._last_line, 6) self.assertEqual(control._first_line, 2) control.selected_choice_index = 7 control._get_formatted_choices() self.assertEqual(control._last_line, 6) self.assertEqual(control._first_line, 2) def test_control_exact_match(self) -> None: content_control = InquirerPyFuzzyControl( choices=["meat", "what", "whaaah", "weather", "haha"], pointer=INQUIRERPY_POINTER_SEQUENCE, marker=INQUIRERPY_POINTER_SEQUENCE, current_text=lambda: "aa", max_lines=80, session_result=None, multiselect=False, marker_pl=" ", match_exact=True, ) self.assertEqual( content_control._filtered_choices, [ { "enabled": False, "index": 0, "indices": [], "name": "meat", "value": "meat", }, { "enabled": False, "index": 1, "indices": [], "name": "what", "value": "what", }, { "enabled": False, "index": 2, "indices": [], "name": "whaaah", "value": "whaaah", }, { "enabled": False, "index": 3, "indices": [], "name": "weather", "value": "weather", }, { "enabled": False, "index": 4, "indices": [], "name": "haha", "value": "haha", }, ], ) result = asyncio.run(content_control._filter_choices(0.0)) self.assertEqual( result, [ { "enabled": False, "index": 2, "indices": [2, 3], "name": "whaaah", "value": "whaaah", } ], ) def test_toggle_exact(self): self.assertEqual(self.prompt.content_control._scorer, fzy_scorer) self.prompt._toggle_exact(None) self.assertEqual(self.prompt.content_control._scorer, substr_scorer) self.prompt._toggle_exact(None) self.assertEqual(self.prompt.content_control._scorer, fzy_scorer) self.prompt._toggle_exact(None, True) self.assertEqual(self.prompt.content_control._scorer, substr_scorer) self.prompt._toggle_exact(None, False) self.assertEqual(self.prompt.content_control._scorer, fzy_scorer)