Issue #19637: fix test_undecodable_env() of test_subprocess on AIX

On AIX, the C locale encoding uses the ISO-8859-1 encoding, not ASCII.
This commit is contained in:
Victor Stinner 2013-11-19 23:46:06 +01:00
parent 85fe8a645e
commit 5323fb0990
1 changed files with 12 additions and 5 deletions

View File

@ -1681,31 +1681,38 @@ class POSIXProcessTestCase(BaseTestCase):
def test_undecodable_env(self): def test_undecodable_env(self):
for key, value in (('test', 'abc\uDCFF'), ('test\uDCFF', '42')): for key, value in (('test', 'abc\uDCFF'), ('test\uDCFF', '42')):
encoded_value = value.encode("ascii", "surrogateescape")
# test str with surrogates # test str with surrogates
script = "import os; print(ascii(os.getenv(%s)))" % repr(key) script = "import os; print(ascii(os.getenv(%s)))" % repr(key)
env = os.environ.copy() env = os.environ.copy()
env[key] = value env[key] = value
# Use C locale to get ascii for the locale encoding to force # Use C locale to get ASCII for the locale encoding to force
# surrogate-escaping of \xFF in the child process; otherwise it can # surrogate-escaping of \xFF in the child process; otherwise it can
# be decoded as-is if the default locale is latin-1. # be decoded as-is if the default locale is latin-1.
env['LC_ALL'] = 'C' env['LC_ALL'] = 'C'
if sys.platform.startswith("aix"):
# On AIX, the C locale uses the Latin1 encoding
decoded_value = encoded_value.decode("latin1", "surrogateescape")
else:
# On other UNIXes, the C locale uses the ASCII encoding
decoded_value = value
stdout = subprocess.check_output( stdout = subprocess.check_output(
[sys.executable, "-c", script], [sys.executable, "-c", script],
env=env) env=env)
stdout = stdout.rstrip(b'\n\r') stdout = stdout.rstrip(b'\n\r')
self.assertEqual(stdout.decode('ascii'), ascii(value)) self.assertEqual(stdout.decode('ascii'), ascii(decoded_value))
# test bytes # test bytes
key = key.encode("ascii", "surrogateescape") key = key.encode("ascii", "surrogateescape")
value = value.encode("ascii", "surrogateescape")
script = "import os; print(ascii(os.getenvb(%s)))" % repr(key) script = "import os; print(ascii(os.getenvb(%s)))" % repr(key)
env = os.environ.copy() env = os.environ.copy()
env[key] = value env[key] = encoded_value
stdout = subprocess.check_output( stdout = subprocess.check_output(
[sys.executable, "-c", script], [sys.executable, "-c", script],
env=env) env=env)
stdout = stdout.rstrip(b'\n\r') stdout = stdout.rstrip(b'\n\r')
self.assertEqual(stdout.decode('ascii'), ascii(value)) self.assertEqual(stdout.decode('ascii'), ascii(encoded_value))
def test_bytes_program(self): def test_bytes_program(self):
abs_program = os.fsencode(sys.executable) abs_program = os.fsencode(sys.executable)