Mock getheightwidth

lysnikolaou · lysnikolaou · commit f2e3e8627141 · 2024-05-07T09:52:17.000+02:00
diff --git a/Lib/test/test_pyrepl.py b/Lib/test/test_pyrepl.py
@@ -104,7 +104,10 @@ def handle_all_events(
 
 
 handle_events_narrow_console = partial(
-    handle_all_events, prepare_console=partial(prepare_mock_console, width=10)
+    handle_all_events,
+    prepare_console=partial(
+        prepare_mock_console, width=10, getheightwidth=MagicMock(lambda self: (25, 10))
+    ),
 )
 
 
@@ -942,15 +945,27 @@ def unix_console(events, **kwargs):
     return console
 
 
-handle_events_unix_console = partial(handle_all_events, prepare_console=unix_console)
+handle_events_unix_console = partial(
+    handle_all_events,
+    prepare_console=partial(unix_console, getheightwidth=MagicMock(25, 80)),
+)
 handle_events_narrow_unix_console = partial(
-    handle_all_events, prepare_console=partial(unix_console, width=5)
+    handle_all_events,
+    prepare_console=partial(
+        unix_console, width=5, getheightwidth=MagicMock(lambda self: (25, 5))
+    ),
 )
 handle_events_short_unix_console = partial(
-    handle_all_events, prepare_console=partial(unix_console, height=1)
+    handle_all_events,
+    prepare_console=partial(
+        unix_console, height=1, getheightwidth=MagicMock(lambda self: (1, 80))
+    ),
 )
 handle_events_unix_console_height_3 = partial(
-    handle_all_events, prepare_console=partial(unix_console, height=2)
+    handle_all_events,
+    prepare_console=partial(
+        unix_console, height=3, getheightwidth=MagicMock(lambda self: (3, 80))
+    ),
 )
 
 
