diff options
| -rw-r--r-- | python/samba/tests/__init__.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/python/samba/tests/__init__.py b/python/samba/tests/__init__.py index 86ec3dff13..65a727aaf6 100644 --- a/python/samba/tests/__init__.py +++ b/python/samba/tests/__init__.py @@ -53,6 +53,7 @@ class TestCase(unittest.TestCase): def get_credentials(self): return cmdline_credentials + # These functions didn't exist before Python2.7: if not getattr(unittest.TestCase, "skipTest", None): def skipTest(self, reason): raise SkipTest(reason) @@ -65,6 +66,16 @@ class TestCase(unittest.TestCase): def assertIsNot(self, a, b): self.assertTrue(a is not b) + if not getattr(unittest.TestCase, "addCleanup", None): + def addCleanup(self, fn, *args, **kwargs): + self._cleanups = getattr(self, "_cleanups", []) + [ + (fn, args, kwargs)] + + def tearDown(self): + super(TestCase, self).tearDown() + for (fn, args, kwargs) in reversed(getattr(self, "_cleanups", [])): + fn(*args, **kwargs) + class LdbTestCase(unittest.TestCase): """Trivial test case for running tests against a LDB.""" |