Kaydet (Commit) 38430e2d authored tarafından Victor Stinner's avatar Victor Stinner

Fix os.get_exec_path() (code and tests) for python -bb

Catch BytesWarning exceptions.
üst 9802b39c
......@@ -387,13 +387,13 @@ def get_exec_path(env=None):
try:
path_list = env.get('PATH')
except TypeError:
except (TypeError, BytesWarning):
path_list = None
if supports_bytes_environ:
try:
path_listb = env[b'PATH']
except (KeyError, TypeError):
except (KeyError, TypeError, BytesWarning):
pass
else:
if path_list is not None:
......
......@@ -450,8 +450,12 @@ class EnvironTests(mapping_tests.BasicTestMappingProtocol):
if os.supports_bytes_environ:
# env cannot contain 'PATH' and b'PATH' keys
self.assertRaises(ValueError,
os.get_exec_path, {'PATH': '1', b'PATH': b'2'})
try:
mixed_env = {'PATH': '1', b'PATH': b'2'}
except BytesWarning:
pass
else:
self.assertRaises(ValueError, os.get_exec_path, mixed_env)
# bytes key and/or value
self.assertSequenceEqual(os.get_exec_path({b'PATH': b'abc'}),
......@@ -723,7 +727,10 @@ class ExecTests(unittest.TestCase):
# os.get_exec_path() reads the 'PATH' variable
with _execvpe_mockup() as calls:
env_path = env.copy()
env_path['PATH'] = program_path
if test_type is bytes:
env_path[b'PATH'] = program_path
else:
env_path['PATH'] = program_path
self.assertRaises(OSError,
os._execvpe, program, arguments, env=env_path)
self.assertEqual(len(calls), 1)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment