test(fuzzy): height adjustment

kazhala · Oct 12, 2021 · b7df510 · b7df510
1 parent 4db519b
commit b7df510
Show file tree

Hide file tree

Showing 2 changed files with 91 additions and 1 deletion.
diff --git a/InquirerPy/prompts/fuzzy.py b/InquirerPy/prompts/fuzzy.py
@@ -63,7 +63,7 @@ def __init__(
         max_lines: int,
         session_result: Optional[InquirerPySessionResult],
         multiselect: bool,
-        marker_pl: str = " ",
+        marker_pl: str,
     ) -> None:
         self._pointer = pointer
         self._marker = marker

diff --git a/tests/prompts/test_fuzzy.py b/tests/prompts/test_fuzzy.py
@@ -26,6 +26,7 @@ class TestFuzzy(unittest.TestCase):
         max_lines=80,
         session_result=None,
         multiselect=False,
+        marker_pl=" ",
     )
 
     @patch("InquirerPy.utils.shutil.get_terminal_size")
@@ -177,6 +178,7 @@ def test_prompt_filter1(self):
             max_lines=80,
             session_result=None,
             multiselect=False,
+            marker_pl=" ",
         )
         self.assertEqual(
             content_control._filtered_choices,
@@ -280,6 +282,60 @@ def test_prompt_filter1(self):
         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=" ",
+        )
+        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")
@@ -722,3 +778,37 @@ def test_constructor_keybindings(self, mocked_kb, mocked_rendered):
             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=" ",
+        )
+        # 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)