https://github.com/python/cpython/commit/9e45fd9858a059950f7387b4fda2b00df0e8e537
commit: 9e45fd9858a059950f7387b4fda2b00df0e8e537
branch: main
author: devdanzin <[email protected]>
committer: AlexWaygood <[email protected]>
date: 2024-06-26T10:39:07Z
summary:
gh-121016: Add test for `PYTHON_BASIC_REPL` envioronment variable (#121017)
files:
M Lib/test/support/__init__.py
M Lib/test/test_cmd_line.py
M Lib/test/test_pyrepl/test_pyrepl.py
diff --git a/Lib/test/support/__init__.py b/Lib/test/support/__init__.py
index 4b430f85e7175c..dbea070929be9b 100644
--- a/Lib/test/support/__init__.py
+++ b/Lib/test/support/__init__.py
@@ -2626,3 +2626,9 @@ def wrapper(*args, **kwargs):
if value is not None:
os.environ[key] = value
return wrapper
+
+
+def initialized_with_pyrepl():
+ """Detect whether PyREPL was used during Python initialization."""
+ # If the main module has a __file__ attribute it's a Python module, which
means PyREPL.
+ return hasattr(sys.modules["__main__"], "__file__")
diff --git a/Lib/test/test_cmd_line.py b/Lib/test/test_cmd_line.py
index a9963bf89d2914..ac99dc4f915f7d 100644
--- a/Lib/test/test_cmd_line.py
+++ b/Lib/test/test_cmd_line.py
@@ -969,7 +969,7 @@ def test_python_user_base(self):
self.assertIn(expected.encode(), out)
def test_python_basic_repl(self):
- # Currently this only tests that the env var is set
+ # Currently this only tests that the env var is set. See
test_pyrepl.test_python_basic_repl.
code = "import os; print('PYTHON_BASIC_REPL' in os.environ)"
expected = "True"
rc, out, err = assert_python_ok('-c', code, PYTHON_BASIC_REPL='1')
diff --git a/Lib/test/test_pyrepl/test_pyrepl.py
b/Lib/test/test_pyrepl/test_pyrepl.py
index adc55f28f08a1e..21a570d271dc84 100644
--- a/Lib/test/test_pyrepl/test_pyrepl.py
+++ b/Lib/test/test_pyrepl/test_pyrepl.py
@@ -862,6 +862,31 @@ def test_dumb_terminal_exits_cleanly(self):
self.assertNotIn("Exception", output)
self.assertNotIn("Traceback", output)
+ @force_not_colorized
+ def test_python_basic_repl(self):
+ env = os.environ.copy()
+ commands = ("from test.support import initialized_with_pyrepl\n"
+ "initialized_with_pyrepl()\n"
+ "exit()\n")
+
+ env.pop("PYTHON_BASIC_REPL", None)
+ output, exit_code = self.run_repl(commands, env=env)
+ if "can\'t use pyrepl" in output:
+ self.skipTest("pyrepl not available")
+ self.assertEqual(exit_code, 0)
+ self.assertIn("True", output)
+ self.assertNotIn("False", output)
+ self.assertNotIn("Exception", output)
+ self.assertNotIn("Traceback", output)
+
+ env["PYTHON_BASIC_REPL"] = "1"
+ output, exit_code = self.run_repl(commands, env=env)
+ self.assertEqual(exit_code, 0)
+ self.assertIn("False", output)
+ self.assertNotIn("True", output)
+ self.assertNotIn("Exception", output)
+ self.assertNotIn("Traceback", output)
+
def run_repl(self, repl_input: str | list[str], env: dict | None = None)
-> tuple[str, int]:
master_fd, slave_fd = pty.openpty()
process = subprocess.Popen(
_______________________________________________
Python-checkins mailing list -- [email protected]
To unsubscribe send an email to [email protected]
https://mail.python.org/mailman3/lists/python-checkins.python.org/
Member address: [email protected]
