aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/Lib/test/test_pyrepl/test_reader.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_pyrepl/test_reader.py')
-rw-r--r--Lib/test/test_pyrepl/test_reader.py17
1 files changed, 17 insertions, 0 deletions
diff --git a/Lib/test/test_pyrepl/test_reader.py b/Lib/test/test_pyrepl/test_reader.py
index 079c963d19a..78b11323d60 100644
--- a/Lib/test/test_pyrepl/test_reader.py
+++ b/Lib/test/test_pyrepl/test_reader.py
@@ -2,8 +2,10 @@ import itertools
import functools
import rlcompleter
from unittest import TestCase
+from unittest.mock import MagicMock, patch
from .support import handle_all_events, handle_events_narrow_console, code_to_events, prepare_reader
+from test.support import import_helper
from _pyrepl.console import Event
from _pyrepl.reader import Reader
@@ -179,6 +181,21 @@ class TestReader(TestCase):
self.assert_screen_equals(reader, expected)
self.assertTrue(reader.finished)
+ def test_input_hook_is_called_if_set(self):
+ input_hook = MagicMock()
+ def _prepare_console(events):
+ console = MagicMock()
+ console.get_event.side_effect = events
+ console.height = 100
+ console.width = 80
+ console.input_hook = input_hook
+ return console
+
+ events = code_to_events("a")
+ reader, _ = handle_all_events(events, prepare_console=_prepare_console)
+
+ self.assertEqual(len(input_hook.mock_calls), 4)
+
def test_keyboard_interrupt_clears_screen(self):
namespace = {"itertools": itertools}
code = "import itertools\nitertools."