Initial commit of OpenSPARC T2 architecture model.

[OpenSPARC-T2-SAM] / sam-t2 / devtools / v9 / lib / python2.4 / distutils / tests / test_dist.py

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