]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
gh-140482: Preserve and restore `stty echo` as a test environment (#140519)
authorBarry Warsaw <barry@python.org>
Fri, 24 Oct 2025 18:07:08 +0000 (11:07 -0700)
committerGitHub <noreply@github.com>
Fri, 24 Oct 2025 18:07:08 +0000 (11:07 -0700)
gh-140482: Restore `stty echo` as a test environment

Lib/test/libregrtest/save_env.py
Misc/NEWS.d/next/Tests/2025-10-23-16-39-49.gh-issue-140482.ZMtyeD.rst [new file with mode: 0644]

index 4cf1a075b300135e043745f132da1bd37cb840df..138465012a252c5209bbc86efc0b2ef96f5d698f 100644 (file)
@@ -9,6 +9,13 @@ from test.support import os_helper
 
 from .utils import print_warning
 
+# Import termios to save and restore terminal echo.  This is only available on
+# Unix, and it's fine if the module can't be found.
+try:
+    import termios                                # noqa: F401
+except ModuleNotFoundError:
+    pass
+
 
 class SkipTestEnvironment(Exception):
     pass
@@ -65,6 +72,7 @@ class saved_test_environment:
                  'shutil_archive_formats', 'shutil_unpack_formats',
                  'asyncio.events._event_loop_policy',
                  'urllib.requests._url_tempfiles', 'urllib.requests._opener',
+                 'stty_echo',
                 )
 
     def get_module(self, name):
@@ -292,6 +300,24 @@ class saved_test_environment:
         warnings = self.get_module('warnings')
         warnings.showwarning = fxn
 
+    def get_stty_echo(self):
+        termios = self.try_get_module('termios')
+        if not os.isatty(fd := sys.__stdin__.fileno()):
+            return None
+        attrs = termios.tcgetattr(fd)
+        lflags = attrs[3]
+        return bool(lflags & termios.ECHO)
+    def restore_stty_echo(self, echo):
+        termios = self.get_module('termios')
+        attrs = termios.tcgetattr(fd := sys.__stdin__.fileno())
+        if echo:
+            # Turn echo on.
+            attrs[3] |= termios.ECHO
+        else:
+            # Turn echo off.
+            attrs[3] &= ~termios.ECHO
+        termios.tcsetattr(fd, termios.TCSADRAIN, attrs)
+
     def resource_info(self):
         for name in self.resources:
             method_suffix = name.replace('.', '_')
diff --git a/Misc/NEWS.d/next/Tests/2025-10-23-16-39-49.gh-issue-140482.ZMtyeD.rst b/Misc/NEWS.d/next/Tests/2025-10-23-16-39-49.gh-issue-140482.ZMtyeD.rst
new file mode 100644 (file)
index 0000000..20747ad
--- /dev/null
@@ -0,0 +1 @@
+Preserve and restore the state of ``stty echo`` as part of the test environment.