mirror of
https://github.com/python/cpython.git
synced 2025-10-31 02:15:10 +00:00
e96ac3a33a
svn+ssh://pythondev@svn.python.org/python/branches/py3k
................
r78991 | martin.v.loewis | 2010-03-16 12:03:13 +0100 (Di, 16 Mär 2010) | 9 lines
Merged revisions 78976 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk
........
r78976 | martin.v.loewis | 2010-03-15 14:00:17 +0100 (Mo, 15 Mär 2010) | 1 line
Issue #6716: Quote -x arguments of compileall in MSI installer.
........
................
r78992 | martin.v.loewis | 2010-03-16 14:19:21 +0100 (Di, 16 Mär 2010) | 2 lines
Issue #6716/2: Backslash-replace error output in compilall.
................
r78994 | martin.v.loewis | 2010-03-16 17:19:47 +0100 (Di, 16 Mär 2010) | 1 line
Issue #6716/3: Exclude 2to3 tests from compileall.
................
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
import sys
|
|
import compileall
|
|
import imp
|
|
import os
|
|
import py_compile
|
|
import shutil
|
|
import struct
|
|
import sys
|
|
import tempfile
|
|
import time
|
|
from test import support
|
|
import unittest
|
|
import io
|
|
|
|
|
|
class CompileallTests(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.directory = tempfile.mkdtemp()
|
|
self.source_path = os.path.join(self.directory, '_test.py')
|
|
self.bc_path = self.source_path + ('c' if __debug__ else 'o')
|
|
with open(self.source_path, 'w') as file:
|
|
file.write('x = 123\n')
|
|
|
|
def tearDown(self):
|
|
shutil.rmtree(self.directory)
|
|
|
|
def data(self):
|
|
with open(self.bc_path, 'rb') as file:
|
|
data = file.read(8)
|
|
mtime = int(os.stat(self.source_path).st_mtime)
|
|
compare = struct.pack('<4sl', imp.get_magic(), mtime)
|
|
return data, compare
|
|
|
|
def recreation_check(self, metadata):
|
|
"""Check that compileall recreates bytecode when the new metadata is
|
|
used."""
|
|
if not hasattr(os, 'stat'):
|
|
return
|
|
py_compile.compile(self.source_path)
|
|
self.assertEqual(*self.data())
|
|
with open(self.bc_path, 'rb') as file:
|
|
bc = file.read()[len(metadata):]
|
|
with open(self.bc_path, 'wb') as file:
|
|
file.write(metadata)
|
|
file.write(bc)
|
|
self.assertNotEqual(*self.data())
|
|
compileall.compile_dir(self.directory, force=False, quiet=True)
|
|
self.assertTrue(*self.data())
|
|
|
|
def test_mtime(self):
|
|
# Test a change in mtime leads to a new .pyc.
|
|
self.recreation_check(struct.pack('<4sl', imp.get_magic(), 1))
|
|
|
|
def test_magic_number(self):
|
|
# Test a change in mtime leads to a new .pyc.
|
|
self.recreation_check(b'\0\0\0\0')
|
|
|
|
|
|
class EncodingTest(unittest.TestCase):
|
|
'Issue 6716: compileall should escape source code when printing errors to stdout.'
|
|
|
|
def setUp(self):
|
|
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')
|
|
|
|
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
|
|
|
|
def test_main():
|
|
support.run_unittest(CompileallTests,
|
|
EncodingTest)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|