diff options
| -rw-r--r-- | Lib/_pyrepl/readline.py | 4 | ||||
| -rw-r--r-- | Lib/test/test_pyrepl/test_pyrepl.py | 30 | ||||
| -rw-r--r-- | Lib/test/test_pyrepl/test_unix_eventqueue.py | 2 |
3 files changed, 28 insertions, 8 deletions
diff --git a/Lib/_pyrepl/readline.py b/Lib/_pyrepl/readline.py index 9c85ce1..787dbc0 100644 --- a/Lib/_pyrepl/readline.py +++ b/Lib/_pyrepl/readline.py @@ -28,6 +28,7 @@ extensions for multiline input. from __future__ import annotations +import warnings from dataclasses import dataclass, field import os @@ -301,7 +302,8 @@ class _ReadlineWrapper: reader.more_lines = more_lines reader.ps1 = reader.ps2 = ps1 reader.ps3 = reader.ps4 = ps2 - return reader.readline(), reader.was_paste_mode_activated + with warnings.catch_warnings(action="ignore"): + return reader.readline(), reader.was_paste_mode_activated finally: reader.more_lines = saved reader.paste_mode = False diff --git a/Lib/test/test_pyrepl/test_pyrepl.py b/Lib/test/test_pyrepl/test_pyrepl.py index bc0a997..b643ae5 100644 --- a/Lib/test/test_pyrepl/test_pyrepl.py +++ b/Lib/test/test_pyrepl/test_pyrepl.py @@ -1,12 +1,15 @@ import itertools +import io import os import rlcompleter -import unittest from unittest import TestCase +from unittest.mock import patch -from .support import FakeConsole, handle_all_events, handle_events_narrow_console, multiline_input, code_to_events +from .support import FakeConsole, handle_all_events, handle_events_narrow_console +from .support import more_lines, multiline_input, code_to_events from _pyrepl.console import Event from _pyrepl.readline import ReadlineAlikeReader, ReadlineConfig +from _pyrepl.readline import multiline_input as readline_multiline_input class TestCursorPosition(TestCase): @@ -475,6 +478,25 @@ class TestPyReplCompleter(TestCase): output = multiline_input(reader, namespace) self.assertEqual(output, "os.") + @patch("_pyrepl.readline._ReadlineWrapper.get_reader") + @patch("sys.stderr", new_callable=io.StringIO) + def test_completion_with_warnings(self, mock_stderr, mock_get_reader): + class Dummy: + @property + def test_func(self): + import warnings + warnings.warn("warnings\n") + return None + + dummy = Dummy() + events = code_to_events("dummy.test_func.\t\n\n") + namespace = {"dummy": dummy} + reader = self.prepare_reader(events, namespace) + mock_get_reader.return_value = reader + output = readline_multiline_input(more_lines, ">>>", "...") + self.assertEqual(output[0], "dummy.test_func.__") + self.assertEqual(mock_stderr.getvalue(), "") + class TestPasteEvent(TestCase): def prepare_reader(self, events): @@ -633,7 +655,3 @@ class TestPasteEvent(TestCase): reader = self.prepare_reader(events) output = multiline_input(reader) self.assertEqual(output, input_code) - - -if __name__ == "__main__": - unittest.main() diff --git a/Lib/test/test_pyrepl/test_unix_eventqueue.py b/Lib/test/test_pyrepl/test_unix_eventqueue.py index be2bc00..c06536b 100644 --- a/Lib/test/test_pyrepl/test_unix_eventqueue.py +++ b/Lib/test/test_pyrepl/test_unix_eventqueue.py @@ -7,7 +7,7 @@ from _pyrepl.unix_eventqueue import EventQueue @patch("_pyrepl.curses.tigetstr", lambda x: b"") -class TestUnivEventQueue(unittest.TestCase): +class TestUnixEventQueue(unittest.TestCase): def setUp(self): self.file = tempfile.TemporaryFile() |