101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
|
"""Tests for distutils.dist."""
|
||
|
|
|
||
|
|
import distutils.cmd
|
||
|
|
import distutils.dist
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import sys
|
||
|
|
import tempfile
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from test.test_support import TESTFN
|
||
|
|
|
||
|
|
|
||
|
|
class test_dist(distutils.cmd.Command):
|
||
|
|
"""Sample distutils extension command."""
|
||
|
|
|
||
|
|
user_options = [
|
||
|
|
("sample-option=", "S", "help text"),
|
||
|
|
]
|
||
|
|
|
||
|
|
def initialize_options(self):
|
||
|
|
self.sample_option = None
|
||
|
|
|
||
|
|
|
||
|
|
class TestDistribution(distutils.dist.Distribution):
|
||
|
|
"""Distribution subclasses that avoids the default search for
|
||
|
|
configuration files.
|
||
|
|
|
||
|
|
The ._config_files attribute must be set before
|
||
|
|
.parse_config_files() is called.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def find_config_files(self):
|
||
|
|
return self._config_files
|
||
|
|
|
||
|
|
|
||
|
|
class DistributionTestCase(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.argv = sys.argv[:]
|
||
|
|
del sys.argv[1:]
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
sys.argv[:] = self.argv
|
||
|
|
|
||
|
|
def create_distribution(self, configfiles=()):
|
||
|
|
d = TestDistribution()
|
||
|
|
d._config_files = configfiles
|
||
|
|
d.parse_config_files()
|
||
|
|
d.parse_command_line()
|
||
|
|
return d
|
||
|
|
|
||
|
|
def test_command_packages_unspecified(self):
|
||
|
|
sys.argv.append("build")
|
||
|
|
d = self.create_distribution()
|
||
|
|
self.assertEqual(d.get_command_packages(), ["distutils.command"])
|
||
|
|
|
||
|
|
def test_command_packages_cmdline(self):
|
||
|
|
sys.argv.extend(["--command-packages",
|
||
|
|
"foo.bar,distutils.tests",
|
||
|
|
"test_dist",
|
||
|
|
"-Ssometext",
|
||
|
|
])
|
||
|
|
d = self.create_distribution()
|
||
|
|
# let's actually try to load our test command:
|
||
|
|
self.assertEqual(d.get_command_packages(),
|
||
|
|
["distutils.command", "foo.bar", "distutils.tests"])
|
||
|
|
cmd = d.get_command_obj("test_dist")
|
||
|
|
self.assert_(isinstance(cmd, test_dist))
|
||
|
|
self.assertEqual(cmd.sample_option, "sometext")
|
||
|
|
|
||
|
|
def test_command_packages_configfile(self):
|
||
|
|
sys.argv.append("build")
|
||
|
|
f = open(TESTFN, "w")
|
||
|
|
try:
|
||
|
|
print >>f, "[global]"
|
||
|
|
print >>f, "command_packages = foo.bar, splat"
|
||
|
|
f.close()
|
||
|
|
d = self.create_distribution([TESTFN])
|
||
|
|
self.assertEqual(d.get_command_packages(),
|
||
|
|
["distutils.command", "foo.bar", "splat"])
|
||
|
|
|
||
|
|
# ensure command line overrides config:
|
||
|
|
sys.argv[1:] = ["--command-packages", "spork", "build"]
|
||
|
|
d = self.create_distribution([TESTFN])
|
||
|
|
self.assertEqual(d.get_command_packages(),
|
||
|
|
["distutils.command", "spork"])
|
||
|
|
|
||
|
|
# Setting --command-packages to '' should cause the default to
|
||
|
|
# be used even if a config file specified something else:
|
||
|
|
sys.argv[1:] = ["--command-packages", "", "build"]
|
||
|
|
d = self.create_distribution([TESTFN])
|
||
|
|
self.assertEqual(d.get_command_packages(), ["distutils.command"])
|
||
|
|
|
||
|
|
finally:
|
||
|
|
os.unlink(TESTFN)
|
||
|
|
|
||
|
|
|
||
|
|
def test_suite():
|
||
|
|
return unittest.makeSuite(DistributionTestCase)
|