Add test_pythonsafepath_env in test_cmd_line

naonus · youknowone · commit 16218cfae44a · 2023-08-30T18:25:40.000+09:00
diff --git a/Lib/subprocess.py b/Lib/subprocess.py
@@ -309,6 +309,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
@@ -1008,6 +1008,25 @@ def test_sys_flags_not_set(self):
             PYTHONSAFEPATH="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)
 
 class SyntaxErrorTests(unittest.TestCase):
     def check_string(self, code):

Original file line number	Diff line number	Diff line change
`@@ -309,6 +309,7 @@ def _args_from_interpreter_flags():`
`309`	`309`	`'verbose': 'v',`
`310`	`310`	`'bytes_warning': 'b',`
`311`	`311`	`'quiet': 'q',`
	`312`	`+ 'safe_path': 'P'`
`312`	`313`	`# -O is handled in _optim_args_from_interpreter_flags()`
`313`	`314`	`}`
`314`	`315`	`args = _optim_args_from_interpreter_flags()`