Add tests

Author: tanloong

Lib/test/test_pyrepl/test_pyrepl.py

@@ -11,7 +11,7 @@
 import tempfile
 from pkgutil import ModuleInfo
 from unittest import TestCase, skipUnless, skipIf, SkipTest
-from unittest.mock import patch
+from unittest.mock import Mock, patch
 from test.support import force_not_colorized, make_clean_env, Py_DEBUG
 from test.support import has_subprocess_support, SHORT_TIMEOUT, STDLIB_DIR
 from test.support.import_helper import import_module
@@ -1908,3 +1908,43 @@ def test_ctrl_d_single_line_end_no_newline(self):
         )
         reader, _ = handle_all_events(events)
         self.assertEqual("hello", "".join(reader.buffer))
+
+
+@skipUnless(sys.platform == "win32", "windows console only")
+class TestWindowsConsoleEolWrap(TestCase):
+    def _make_mock_console(self, width=80):
+        from _pyrepl import windows_console as wc
+
+        console = object.__new__(wc.WindowsConsole)
+
+        console.width = width
+        console.posxy = (0, 0)
+        console.screen = [""]
+
+        console._hide_cursor = Mock()
+        console._show_cursor = Mock()
+        console._erase_to_end = Mock()
+        console._move_relative = Mock()
+        console.move_cursor = Mock()
+        console._WindowsConsole__write = Mock()
+
+        return console, wc
+
+    def test_normal_line_not_full(self):
+        width = 10
+        console, wc = self._make_mock_console(width=width)
+        new_line = "a" * 3
+        wc.WindowsConsole._WindowsConsole__write_changed_line(
+            console, 0, "", new_line, 0
+        )
+        self.assertEqual(console.posxy, (3, 0))
+
+    def test_exact_width_line_wrap_behavior(self):
+        width = 10
+        console, wc = self._make_mock_console(width=width)
+        new_line = "a" * width
+
+        wc.WindowsConsole._WindowsConsole__write_changed_line(
+            console, 0, "", new_line, 0
+        )
+        self.assertEqual(console.posxy, (width - 1, 0))