Add test_pythonsafepath_env in test_cmd_line

naonus · naonus · commit ad6553c43c61 · 2023-03-02T23:52:13.000+09:00
diff --git a/Lib/subprocess.py b/Lib/subprocess.py
@@ -288,6 +288,7 @@ def _args_from_interpreter_flags():
         'verbose': 'v',
         'bytes_warning': 'b',
         'quiet': 'q',
+        'safe_path': 'P'
         # -O is handled in _optim_args_from_interpreter_flags()
     }
     args = _optim_args_from_interpreter_flags()
diff --git a/Lib/test/test_cmd_line.py b/Lib/test/test_cmd_line.py
@@ -884,6 +884,25 @@ def test_sys_flags_not_set(self):
             PYTHONVERBOSE="1",
         )
 
+    # TODO: RUSTPYTHON
+    @unittest.expectedFailure
+    def test_pythonsafepath_env(self):
+        # Test the PYTHONSAFEPATH environment variable
+        code = "import sys; print(sys.flags.safe_path)"
+        env = dict(os.environ)
+        env.pop('PYTHONSAFEPATH', None)
+        args = (sys.executable, '-P', code)
+
+        proc = subprocess.run(args, stdout=subprocess.PIPE,
+                              universal_newlines=True, env=env)
+        self.assertEqual(proc.stdout.rstrip(), 'False')
+        self.assertEqual(proc.returncode, 0, proc)
+
+        env['PYTHONSAFEPATH'] = '1'
+        proc = subprocess.run(args, stdout=subprocess.PIPE,
+                              universal_newlines=True, env=env)
+        self.assertEqual(proc.stdout.rstrip(), 'True')
+        self.assertEqual(proc.returncode, 0, proc)
 
 def test_main():
     support.run_unittest(CmdLineTest, IgnoreEnvironmentTest)

Original file line number	Diff line number	Diff line change
`@@ -288,6 +288,7 @@ def _args_from_interpreter_flags():`
`288`	`288`	`'verbose': 'v',`
`289`	`289`	`'bytes_warning': 'b',`
`290`	`290`	`'quiet': 'q',`
	`291`	`+ 'safe_path': 'P'`
`291`	`292`	`# -O is handled in _optim_args_from_interpreter_flags()`
`292`	`293`	`}`
`293`	`294`	`args = _optim_args_from_interpreter_flags()`