test(number): negative toggle

kazhala · Dec 7, 2021 · 7fa9432 · 7fa9432
1 parent 318b2bf
commit 7fa9432
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/tests/prompts/test_number.py b/tests/prompts/test_number.py
@@ -236,3 +236,41 @@ def test_on_text_change(self, mocked_text) -> None:
         self.assertEqual(self.float_prompt._integral_width, 3)
         self.float_prompt._integral_buffer.text = "100"
         self.assertEqual(self.float_prompt._integral_width, 4)
+
+    def test_handle_negative_toggle(self) -> None:
+        self.assertEqual(self.prompt._whole_buffer.text, "1")
+        self.assertEqual(self.prompt._whole_buffer.cursor_position, 1)
+        self.prompt._handle_negative_toggle(None)
+        self.assertEqual(self.prompt._whole_buffer.text, "-1")
+        self.assertEqual(self.prompt._whole_buffer.cursor_position, 2)
+        self.prompt._handle_negative_toggle(None)
+        self.assertEqual(self.prompt._whole_buffer.text, "1")
+        self.assertEqual(self.prompt._whole_buffer.cursor_position, 1)
+
+        self.prompt._min = -10
+        self.prompt._max = 10
+        self.prompt._whole_buffer.text = "10"
+
+        self.prompt._whole_buffer.cursor_position = 2
+        self.prompt._handle_negative_toggle(None)
+        self.assertEqual(self.prompt._whole_buffer.text, "-10")
+        self.assertEqual(self.prompt._whole_buffer.cursor_position, 3)
+        self.prompt._handle_negative_toggle(None)
+        self.assertEqual(self.prompt._whole_buffer.text, "10")
+        self.assertEqual(self.prompt._whole_buffer.cursor_position, 2)
+
+        self.prompt._whole_buffer.cursor_position = 1
+        self.prompt._handle_negative_toggle(None)
+        self.assertEqual(self.prompt._whole_buffer.text, "-10")
+        self.assertEqual(self.prompt._whole_buffer.cursor_position, 2)
+        self.prompt._handle_negative_toggle(None)
+        self.assertEqual(self.prompt._whole_buffer.text, "10")
+        self.assertEqual(self.prompt._whole_buffer.cursor_position, 1)
+
+        self.prompt._whole_buffer.cursor_position = 0
+        self.prompt._handle_negative_toggle(None)
+        self.assertEqual(self.prompt._whole_buffer.text, "-10")
+        self.assertEqual(self.prompt._whole_buffer.cursor_position, 1)
+        self.prompt._handle_negative_toggle(None)
+        self.assertEqual(self.prompt._whole_buffer.text, "10")
+        self.assertEqual(self.prompt._whole_buffer.cursor_position, 0)