[Python-checkins] distutils2: Add a simple test for the packaging RECORD file.

[eric.araujo]

http://hg.python.org/distutils2/rev/1272d8fbe018
changeset:   1161:1272d8fbe018
user:        Éric Araujo <merwok at netwok.org>
date:        Mon Sep 19 03:10:40 2011 +0200
summary:
  Add a simple test for the packaging RECORD file.

The existing test_record is not easily extendable to add script files or
extension modules: it collects all files from fake_dists and generates a
RECORD file at runtime.  I felt more comfortable adding a new test
written from scratch, more self-contained (just one project with
well-defined files) and more stupid (the checksums and sizes are
computed once and hard-coded).

files:
  distutils2/tests/test_command_install_dist.py     |   2 +-
  distutils2/tests/test_command_install_distinfo.py |  80 +++++++++-
  2 files changed, 76 insertions(+), 6 deletions(-)


diff --git a/distutils2/tests/test_command_install_dist.py b/distutils2/tests/test_command_install_dist.py
--- a/distutils2/tests/test_command_install_dist.py
+++ b/distutils2/tests/test_command_install_dist.py
@@ -229,7 +229,7 @@
         cmd.ensure_finalized()
         cmd.run()
 
-        f =open(cmd.record)
+        f = open(cmd.record)
         try:
             content = f.read()
         finally:
diff --git a/distutils2/tests/test_command_install_distinfo.py b/distutils2/tests/test_command_install_distinfo.py
--- a/distutils2/tests/test_command_install_distinfo.py
+++ b/distutils2/tests/test_command_install_distinfo.py
@@ -2,16 +2,19 @@
 
 import os
 import csv
+import codecs
+
+from distutils2.command.install_distinfo import install_distinfo
+from distutils2.command.cmd import Command
+from distutils2.compiler.extension import Extension
+from distutils2.metadata import Metadata
+from distutils2.tests import unittest, support
+from distutils2._backport import sysconfig
 try:
     import hashlib
 except:
     from distutils2._backport import hashlib
 
-from distutils2.command.install_distinfo import install_distinfo
-from distutils2.command.cmd import Command
-from distutils2.metadata import Metadata
-from distutils2.tests import unittest, support
-
 
 class DummyInstallCmd(Command):
 
@@ -134,6 +137,73 @@
         self.checkLists(os.listdir(dist_info),
                         ['METADATA', 'REQUESTED', 'INSTALLER'])
 
+    def test_record_basic(self):
+        install_dir = self.mkdtemp()
+        modules_dest = os.path.join(install_dir, 'lib')
+        scripts_dest = os.path.join(install_dir, 'bin')
+        project_dir, dist = self.create_dist(
+            name='Spamlib', version='0.1',
+            py_modules=['spam'], scripts=['spamd'],
+            ext_modules=[Extension('_speedspam', ['_speedspam.c'])])
+
+        # using a real install_dist command is too painful, so we use a mock
+        # class that's only a holder for options to be used by install_distinfo
+        # and we create placeholder files manually instead of using build_*.
+        # the install_* commands will still be consulted by install_distinfo.
+        os.chdir(project_dir)
+        self.write_file('spam', '# Python module')
+        self.write_file('spamd', '# Python script')
+        extmod = '_speedspam' + sysconfig.get_config_var('SO')
+        self.write_file(extmod, '')
+
+        install = DummyInstallCmd(dist)
+        install.outputs = ['spam', 'spamd', extmod]
+        install.install_lib = modules_dest
+        install.install_scripts = scripts_dest
+        dist.command_obj['install_dist'] = install
+
+        cmd = install_distinfo(dist)
+        cmd.ensure_finalized()
+        dist.command_obj['install_distinfo'] = cmd
+        cmd.run()
+
+        # checksum and size are not hard-coded for METADATA as it is
+        # platform-dependent (line endings)
+        metadata = os.path.join(modules_dest, 'Spamlib-0.1.dist-info',
+                                'METADATA')
+        fp = open(metadata, 'rb')
+        try:
+            content = fp.read()
+        finally:
+            fp.close()
+
+        metadata_size = str(len(content))
+        metadata_md5 = hashlib.md5(content).hexdigest()
+
+        record = os.path.join(modules_dest, 'Spamlib-0.1.dist-info', 'RECORD')
+        fp = codecs.open(record, encoding='utf-8')
+        try:
+            content = fp.read()
+        finally:
+            fp.close()
+
+        found = []
+        for line in content.splitlines():
+            filename, checksum, size = line.split(',')
+            filename = os.path.basename(filename)
+            found.append((filename, checksum, size))
+
+        expected = [
+            ('spam', '6ab2f288ef2545868effe68757448b45', '15'),
+            ('spamd', 'd13e6156ce78919a981e424b2fdcd974', '15'),
+            (extmod, 'd41d8cd98f00b204e9800998ecf8427e', '0'),
+            ('METADATA', metadata_md5, metadata_size),
+            ('INSTALLER', '44e3fde05f3f537ed85831969acf396d', '9'),
+            ('REQUESTED', 'd41d8cd98f00b204e9800998ecf8427e', '0'),
+            ('RECORD', '', ''),
+        ]
+        self.assertEqual(found, expected)
+
     def test_record(self):
         pkg_dir, dist = self.create_dist(name='foo',
                                          version='1.0')

-- 
Repository URL: http://hg.python.org/distutils2