Lib/test/test_compileall.py

   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()