Home | History | Annotate | Download | only in test 
      1 import compileall
      2 import imp
      3 import os
      4 import py_compile
      5 import shutil
      6 import struct
      7 import tempfile
      8 from test import test_support
      9 import unittest
     10 
     11 
     12 class CompileallTests(unittest.TestCase):
     13 
     14     def setUp(self):
     15         self.directory = tempfile.mkdtemp()
     16         self.source_path = os.path.join(self.directory, '_test.py')
     17         self.bc_path = self.source_path + ('c' if __debug__ else 'o')
     18         with open(self.source_path, 'w') as file:
     19             file.write('x = 123\n')
     20         self.source_path2 = os.path.join(self.directory, '_test2.py')
     21         self.bc_path2 = self.source_path2 + ('c' if __debug__ else 'o')
     22         shutil.copyfile(self.source_path, self.source_path2)
     23 
     24     def tearDown(self):
     25         shutil.rmtree(self.directory)
     26 
     27     def data(self):
     28         with open(self.bc_path, 'rb') as file:
     29             data = file.read(8)
     30         mtime = int(os.stat(self.source_path).st_mtime)
     31         compare = struct.pack('<4sl', imp.get_magic(), mtime)
     32         return data, compare
     33 
     34     def recreation_check(self, metadata):
     35         """Check that compileall recreates bytecode when the new metadata is
     36         used."""
     37         if not hasattr(os, 'stat'):
     38             return
     39         py_compile.compile(self.source_path)
     40         self.assertEqual(*self.data())
     41         with open(self.bc_path, 'rb') as file:
     42             bc = file.read()[len(metadata):]
     43         with open(self.bc_path, 'wb') as file:
     44             file.write(metadata)
     45             file.write(bc)
     46         self.assertNotEqual(*self.data())
     47         compileall.compile_dir(self.directory, force=False, quiet=True)
     48         self.assertTrue(*self.data())
     49 
     50     def test_mtime(self):
     51         # Test a change in mtime leads to a new .pyc.
     52         self.recreation_check(struct.pack('<4sl', imp.get_magic(), 1))
     53 
     54     def test_magic_number(self):
     55         # Test a change in mtime leads to a new .pyc.
     56         self.recreation_check(b'\0\0\0\0')
     57 
     58     def test_compile_files(self):
     59         # Test compiling a single file, and complete directory
     60         for fn in (self.bc_path, self.bc_path2):
     61             try:
     62                 os.unlink(fn)
     63             except:
     64                 pass
     65         compileall.compile_file(self.source_path, force=False, quiet=True)
     66         self.assertTrue(os.path.isfile(self.bc_path) \
     67                         and not os.path.isfile(self.bc_path2))
     68         os.unlink(self.bc_path)
     69         compileall.compile_dir(self.directory, force=False, quiet=True)
     70         self.assertTrue(os.path.isfile(self.bc_path) \
     71                         and os.path.isfile(self.bc_path2))
     72         os.unlink(self.bc_path)
     73         os.unlink(self.bc_path2)
     74 
     75 def test_main():
     76     test_support.run_unittest(CompileallTests)
     77 
     78 
     79 if __name__ == "__main__":
     80     test_main()
     81