ensure that PYTHON_HISTORY is not altered

Author: picnixz

Lib/test/support/script_helper.py

@@ -3,11 +3,15 @@
 
 import collections
 import importlib
+import site
+import stat
 import sys
 import os
 import os.path
 import subprocess
 import py_compile
+import unittest
+import unittest.mock as mock
 
 from importlib.util import source_from_cache
 from test import support
@@ -322,3 +326,49 @@ def title(text):
             raise AssertionError(f"{name} failed")
     else:
         assert_python_ok("-u", script, "-v")
+
+
+_site_gethistoryfile = site.gethistoryfile
+def _gethistoryfile():
+    """Patch site.gethistoryfile() to ignore -I for PYTHON_HISTORY.
+
+    The -I option is necessary for test_no_memory() but using it
+    forbids using a custom PYTHON_HISTORY.
+    """
+    history = os.environ.get("PYTHON_HISTORY")
+    return history or os.path.join(os.path.expanduser('~'), '.python_history')
+
+
+def patch_gethistoryfile(sitemodule=site):
+    return mock.patch.object(sitemodule, "gethistoryfile", _gethistoryfile)
+
+
+def _file_signature(file):
+    st = os.stat(file)
+    return (stat.S_IFMT(st.st_mode), st.st_size)
+
+
+class EnsureSafeUserHistory(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        if history_file := _site_gethistoryfile():
+            if os.path.exists(history_file):
+                cls.__history_file = history_file
+                cls.__history_stat = _file_signature(history_file)
+        else:
+            cls.__history_file = cls.__history_stat = None
+
+    def tearDown(self):
+        super().tearDown()
+        if self.__history_file is None:
+            return
+        self.assertTrue(
+            os.path.exists(self.__history_file),
+            f"PYTHON_HISTORY file ({self.__history_file!r}) was deleted"
+        )
+        self.assertEqual(
+            self.__history_stat,
+            _file_signature(self.__history_file),
+            f"PYTHON_HISTORY file ({self.__history_file!r}) was altered"
+        )

Lib/test/test_cmd_line_script.py

@@ -17,7 +17,9 @@
 from test.support import import_helper, is_apple, os_helper
 from test.support.script_helper import (
     make_pkg, make_script, make_zip_pkg, make_zip_script,
-    assert_python_ok, assert_python_failure, spawn_python, kill_python)
+    assert_python_ok, assert_python_failure, spawn_python, kill_python,
+    patch_gethistoryfile, EnsureSafeUserHistory
+)
 
 verbose = support.verbose
 
@@ -90,7 +92,8 @@ def _make_test_zip_pkg(zip_dir, zip_basename, pkg_name, script_basename,
 
 
 @support.force_not_colorized_test_class
-class CmdLineTest(unittest.TestCase):
+@patch_gethistoryfile()
+class CmdLineTest(EnsureSafeUserHistory, unittest.TestCase):
     def _check_output(self, script_name, exit_code, data,
                              expected_file, expected_argv0,
                              expected_path0, expected_package,

Lib/test/test_pyrepl/test_pyrepl.py

@@ -16,6 +16,7 @@
 from test.support import has_subprocess_support, SHORT_TIMEOUT, STDLIB_DIR
 from test.support.import_helper import import_module
 from test.support.os_helper import EnvironmentVarGuard, unlink
+from test.support.script_helper import EnsureSafeUserHistory
 
 from .support import (
     FakeConsole,
@@ -45,7 +46,7 @@
     pty = None
 
 
-class ReplTestCase(TestCase):
+class ReplTestCase(EnsureSafeUserHistory, TestCase):
     def setUp(self):
         if not has_subprocess_support:
             raise SkipTest("test module requires subprocess")
@@ -97,10 +98,11 @@ def _run_repl(
             env["PYTHON_HISTORY"] = os.path.join(cwd, ".regrtest_history")
         if cmdline_args is not None:
             if "PYTHON_HISTORY" in env:
-                self.assertNotIn(
-                    "-I", cmdline_args,
-                    "PYTHON_HISTORY will be ignored by -I"
-                )
+                for bad_option in ('-I', '-E'):
+                    self.assertNotIn(
+                        bad_option, cmdline_args,
+                        f"PYTHON_HISTORY will be ignored by {bad_option}"
+                    )
             cmd.extend(cmdline_args)
 
         try:

Lib/test/test_repl.py

@@ -14,7 +14,7 @@
     SuppressCrashReport,
     SHORT_TIMEOUT,
 )
-from test.support.script_helper import kill_python
+from test.support.script_helper import kill_python, EnsureSafeUserHistory
 from test.support.import_helper import import_module
 
 try:
@@ -71,7 +71,7 @@ def run_on_interactive_mode(source):
 
 
 @support.force_not_colorized_test_class
-class TestInteractiveInterpreter(unittest.TestCase):
+class TestInteractiveInterpreter(EnsureSafeUserHistory, unittest.TestCase):
 
     @cpython_only
     # Python built with Py_TRACE_REFS fail with a fatal error in
@@ -303,42 +303,44 @@ def test_asyncio_repl_reaches_python_startup_script(self):
 
     @unittest.skipUnless(pty, "requires pty")
     def test_asyncio_repl_is_ok(self):
-        m, s = pty.openpty()
-        cmd = [sys.executable, "-I", "-m", "asyncio"]
-        env = os.environ.copy()
-        proc = subprocess.Popen(
-            cmd,
-            stdin=s,
-            stdout=s,
-            stderr=s,
-            text=True,
-            close_fds=True,
-            env=env,
-        )
-        os.close(s)
-        os.write(m, b"await asyncio.sleep(0)\n")
-        os.write(m, b"exit()\n")
-        output = []
-        while select.select([m], [], [], SHORT_TIMEOUT)[0]:
-            try:
-                data = os.read(m, 1024).decode("utf-8")
-                if not data:
+        with os_helper.temp_dir() as tmpdir:
+            m, s = pty.openpty()
+            cmd = [sys.executable, "-m", "asyncio"]
+            env = os.environ.copy()
+            env["PYTHON_HISTORY"] = os.path.join(tmpdir, ".asyncio_history")
+            proc = subprocess.Popen(
+                cmd,
+                stdin=s,
+                stdout=s,
+                stderr=s,
+                text=True,
+                close_fds=True,
+                env=env,
+            )
+            os.close(s)
+            os.write(m, b"await asyncio.sleep(0)\n")
+            os.write(m, b"exit()\n")
+            output = []
+            while select.select([m], [], [], SHORT_TIMEOUT)[0]:
+                try:
+                    data = os.read(m, 1024).decode("utf-8")
+                    if not data:
+                        break
+                except OSError:
                     break
-            except OSError:
-                break
-            output.append(data)
-        os.close(m)
-        try:
-            exit_code = proc.wait(timeout=SHORT_TIMEOUT)
-        except subprocess.TimeoutExpired:
-            proc.kill()
-            exit_code = proc.wait()
+                output.append(data)
+            os.close(m)
+            try:
+                exit_code = proc.wait(timeout=SHORT_TIMEOUT)
+            except subprocess.TimeoutExpired:
+                proc.kill()
+                exit_code = proc.wait()
 
-        self.assertEqual(exit_code, 0, "".join(output))
+            self.assertEqual(exit_code, 0, "".join(output))
 
 
 @support.force_not_colorized_test_class
-class TestInteractiveModeSyntaxErrors(unittest.TestCase):
+class TestInteractiveModeSyntaxErrors(EnsureSafeUserHistory, unittest.TestCase):
 
     def test_interactive_syntax_error_correct_line(self):
         output = run_on_interactive_mode(dedent("""\
@@ -356,7 +358,7 @@ def f():
         self.assertEqual(traceback_lines, expected_lines)
 
 
-class TestAsyncioREPL(unittest.TestCase):
+class TestAsyncioREPL(EnsureSafeUserHistory, unittest.TestCase):
     def test_multiple_statements_fail_early(self):
         user_input = "1 / 0; print(f'afterwards: {1+1}')"
         p = spawn_repl("-m", "asyncio")