diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/samba/tests/__init__.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/python/samba/tests/__init__.py b/python/samba/tests/__init__.py index b73308ecd4..99cea490a1 100644 --- a/python/samba/tests/__init__.py +++ b/python/samba/tests/__init__.py @@ -75,10 +75,11 @@ class TestCase(unittest.TestCase): self._cleanups = getattr(self, "_cleanups", []) + [ (fn, args, kwargs)] - def tearDown(self): - super(TestCase, self).tearDown() + def run(self, result=None): + ret = super(TestCase, self).run(result=result) for (fn, args, kwargs) in reversed(getattr(self, "_cleanups", [])): fn(*args, **kwargs) + return ret class LdbTestCase(TestCase): |