gh-117389: Fix test_compileall.EncodingTest (#117390)

This commit is contained in:
Nikita Sobolev 2024-05-05 21:46:37 +03:00 committed by GitHub
parent 5092ea238e
commit 44f67916da
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -502,19 +502,25 @@ class EncodingTest(unittest.TestCase):
self.directory = tempfile.mkdtemp()
self.source_path = os.path.join(self.directory, '_test.py')
with open(self.source_path, 'w', encoding='utf-8') as file:
file.write('# -*- coding: utf-8 -*-\n')
file.write('print u"\u20ac"\n')
# Intentional syntax error: bytes can only contain
# ASCII literal characters.
file.write('b"\u20ac"')
def tearDown(self):
shutil.rmtree(self.directory)
def test_error(self):
try:
orig_stdout = sys.stdout
sys.stdout = io.TextIOWrapper(io.BytesIO(),encoding='ascii')
compileall.compile_dir(self.directory)
finally:
sys.stdout = orig_stdout
buffer = io.TextIOWrapper(io.BytesIO(), encoding='ascii')
with contextlib.redirect_stdout(buffer):
compiled = compileall.compile_dir(self.directory)
self.assertFalse(compiled) # should not be successful
buffer.seek(0)
res = buffer.read()
self.assertIn(
'SyntaxError: bytes can only contain ASCII literal characters',
res,
)
self.assertNotIn('UnicodeEncodeError', res)
class CommandLineTestsBase: