diff options
author | Fred Drake <fdrake@acm.org> | 2004-06-17 20:14:50 +0000 |
---|---|---|
committer | Fred Drake <fdrake@acm.org> | 2004-06-17 20:14:50 +0000 |
commit | b8ab8b6da809f9ceecd715113b2c860235186b02 (patch) | |
tree | f788648daf81cb0a80838a5f0a622a3f91c792f7 /Lib/distutils/tests/support.py | |
parent | 9e1ac2496fd3a5e1126162b7734ff26aed4495df (diff) | |
download | cpython-git-b8ab8b6da809f9ceecd715113b2c860235186b02.tar.gz |
move support code to a helper module to ease re-use
Diffstat (limited to 'Lib/distutils/tests/support.py')
-rw-r--r-- | Lib/distutils/tests/support.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/Lib/distutils/tests/support.py b/Lib/distutils/tests/support.py new file mode 100644 index 0000000000..cef985d79a --- /dev/null +++ b/Lib/distutils/tests/support.py @@ -0,0 +1,41 @@ +"""Support code for distutils test cases.""" + +import shutil +import tempfile + + +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 |