← Back to branch summary

~malept/ubuntu/lucid/python2.6/dev-dependency-fix

« back to all changes in this revision

Viewing changes to Lib/distutils/tests/test_build_py.py

  • Committer: Bazaar Package Importer
  • Author(s): Matthias Klose
  • Date: 2009-02-13 12:51:00 UTC
  • Revision ID: james.westby@ubuntu.com-20090213125100-uufgcb9yeqzujpqw
Tags: upstream-2.6.1
Import upstream version 2.6.1

Show diffs side-by-side

added added

removed removed

Lines of Context:
Lib/distutils/tests/test_build_py.py
 
1
"""Tests for distutils.command.build_py."""
 
2
 
 
3
import os
 
4
import sys
 
5
import StringIO
 
6
import unittest
 
7
 
 
8
from distutils.command.build_py import build_py
 
9
from distutils.core import Distribution
 
10
from distutils.errors import DistutilsFileError
 
11
 
 
12
from distutils.tests import support
 
13
 
 
14
 
 
15
class BuildPyTestCase(support.TempdirManager,
 
16
                      support.LoggingSilencer,
 
17
                      unittest.TestCase):
 
18
 
 
19
    def test_package_data(self):
 
20
        sources = self.mkdtemp()
 
21
        f = open(os.path.join(sources, "__init__.py"), "w")
 
22
        f.write("# Pretend this is a package.")
 
23
        f.close()
 
24
        f = open(os.path.join(sources, "README.txt"), "w")
 
25
        f.write("Info about this package")
 
26
        f.close()
 
27
 
 
28
        destination = self.mkdtemp()
 
29
 
 
30
        dist = Distribution({"packages": ["pkg"],
 
31
                             "package_dir": {"pkg": sources}})
 
32
        # script_name need not exist, it just need to be initialized
 
33
        dist.script_name = os.path.join(sources, "setup.py")
 
34
        dist.command_obj["build"] = support.DummyCommand(
 
35
            force=0,
 
36
            build_lib=destination)
 
37
        dist.packages = ["pkg"]
 
38
        dist.package_data = {"pkg": ["README.txt"]}
 
39
        dist.package_dir = {"pkg": sources}
 
40
 
 
41
        cmd = build_py(dist)
 
42
        cmd.compile = 1
 
43
        cmd.ensure_finalized()
 
44
        self.assertEqual(cmd.package_data, dist.package_data)
 
45
 
 
46
        cmd.run()
 
47
 
 
48
        # This makes sure the list of outputs includes byte-compiled
 
49
        # files for Python modules but not for package data files
 
50
        # (there shouldn't *be* byte-code files for those!).
 
51
        #
 
52
        self.assertEqual(len(cmd.get_outputs()), 3)
 
53
        pkgdest = os.path.join(destination, "pkg")
 
54
        files = os.listdir(pkgdest)
 
55
        self.assert_("__init__.py" in files)
 
56
        self.assert_("__init__.pyc" in files)
 
57
        self.assert_("README.txt" in files)
 
58
 
 
59
    def test_empty_package_dir (self):
 
60
        # See SF 1668596/1720897.
 
61
        cwd = os.getcwd()
 
62
 
 
63
        # create the distribution files.
 
64
        sources = self.mkdtemp()
 
65
        open(os.path.join(sources, "__init__.py"), "w").close()
 
66
 
 
67
        testdir = os.path.join(sources, "doc")
 
68
        os.mkdir(testdir)
 
69
        open(os.path.join(testdir, "testfile"), "w").close()
 
70
 
 
71
        os.chdir(sources)
 
72
        sys.stdout = StringIO.StringIO()
 
73
 
 
74
        try:
 
75
            dist = Distribution({"packages": ["pkg"],
 
76
                                 "package_dir": {"pkg": ""},
 
77
                                 "package_data": {"pkg": ["doc/*"]}})
 
78
            # script_name need not exist, it just need to be initialized
 
79
            dist.script_name = os.path.join(sources, "setup.py")
 
80
            dist.script_args = ["build"]
 
81
            dist.parse_command_line()
 
82
 
 
83
            try:
 
84
                dist.run_commands()
 
85
            except DistutilsFileError:
 
86
                self.fail("failed package_data test when package_dir is ''")
 
87
        finally:
 
88
            # Restore state.
 
89
            os.chdir(cwd)
 
90
            sys.stdout = sys.__stdout__
 
91
 
 
92
def test_suite():
 
93
    return unittest.makeSuite(BuildPyTestCase)
 
94
 
 
95
if __name__ == "__main__":
 
96
    unittest.main(defaultTest="test_suite")