54 lines · 1.2 KB · Python
RawDownload
"""Support code for distutils test cases.""" import shutilimport tempfile from distutils import log  class LoggingSilencer(object):     def setUp(self):        super(LoggingSilencer, self).setUp()        self.threshold = log.set_threshold(log.FATAL)     def tearDown(self):        log.set_threshold(self.threshold)        super(LoggingSilencer, self).tearDown()  class TempdirManager(object):    """Mix-in class that handles temporary directories for test cases.     This is intended to be used with unittest.TestCase.    """     def setUp(self):        super(TempdirManager, self).setUp()        self.tempdirs = []     def tearDown(self):        super(TempdirManager, self).tearDown()        while self.tempdirs:            d = self.tempdirs.pop()            shutil.rmtree(d)     def mkdtemp(self):        """Create a temporary directory that will be cleaned up.         Returns the path of the directory.        """        d = tempfile.mkdtemp()        self.tempdirs.append(d)        return d  class DummyCommand:    """Class to store options for retrieval via set_undefined_options()."""     def __init__(self, **kwargs):        for kw, val in kwargs.items():            setattr(self, kw, val)     def ensure_finalized(self):        pass