"""Tests for distutils.dist.""" import distutils.cmdimport distutils.distimport osimport shutilimport StringIOimport sysimport tempfileimport 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)  class MetadataTestCase(unittest.TestCase):     def test_simple_metadata(self):        attrs = {"name": "package",                 "version": "1.0"}        dist = distutils.dist.Distribution(attrs)        meta = self.format_metadata(dist)        self.assert_("Metadata-Version: 1.0" in meta)        self.assert_("provides:" not in meta.lower())        self.assert_("requires:" not in meta.lower())        self.assert_("obsoletes:" not in meta.lower())     def test_provides(self):        attrs = {"name": "package",                 "version": "1.0",                 "provides": ["package", "package.sub"]}        dist = distutils.dist.Distribution(attrs)        self.assertEqual(dist.metadata.get_provides(),                         ["package", "package.sub"])        self.assertEqual(dist.get_provides(),                         ["package", "package.sub"])        meta = self.format_metadata(dist)        self.assert_("Metadata-Version: 1.1" in meta)        self.assert_("requires:" not in meta.lower())        self.assert_("obsoletes:" not in meta.lower())     def test_provides_illegal(self):        self.assertRaises(ValueError,                          distutils.dist.Distribution,                          {"name": "package",                           "version": "1.0",                           "provides": ["my.pkg (splat)"]})     def test_requires(self):        attrs = {"name": "package",                 "version": "1.0",                 "requires": ["other", "another (==1.0)"]}        dist = distutils.dist.Distribution(attrs)        self.assertEqual(dist.metadata.get_requires(),                         ["other", "another (==1.0)"])        self.assertEqual(dist.get_requires(),                         ["other", "another (==1.0)"])        meta = self.format_metadata(dist)        self.assert_("Metadata-Version: 1.1" in meta)        self.assert_("provides:" not in meta.lower())        self.assert_("Requires: other" in meta)        self.assert_("Requires: another (==1.0)" in meta)        self.assert_("obsoletes:" not in meta.lower())     def test_requires_illegal(self):        self.assertRaises(ValueError,                          distutils.dist.Distribution,                          {"name": "package",                           "version": "1.0",                           "requires": ["my.pkg (splat)"]})     def test_obsoletes(self):        attrs = {"name": "package",                 "version": "1.0",                 "obsoletes": ["other", "another (<1.0)"]}        dist = distutils.dist.Distribution(attrs)        self.assertEqual(dist.metadata.get_obsoletes(),                         ["other", "another (<1.0)"])        self.assertEqual(dist.get_obsoletes(),                         ["other", "another (<1.0)"])        meta = self.format_metadata(dist)        self.assert_("Metadata-Version: 1.1" in meta)        self.assert_("provides:" not in meta.lower())        self.assert_("requires:" not in meta.lower())        self.assert_("Obsoletes: other" in meta)        self.assert_("Obsoletes: another (<1.0)" in meta)     def test_obsoletes_illegal(self):        self.assertRaises(ValueError,                          distutils.dist.Distribution,                          {"name": "package",                           "version": "1.0",                           "obsoletes": ["my.pkg (splat)"]})     def format_metadata(self, dist):        sio = StringIO.StringIO()        dist.metadata.write_pkg_file(sio)        return sio.getvalue()  def test_suite():    suite = unittest.TestSuite()    suite.addTest(unittest.makeSuite(DistributionTestCase))    suite.addTest(unittest.makeSuite(MetadataTestCase))    return suite