Fixed support for MockerTestCase.addCleanup() in Python 2.3,

by Anders F Björklund (#528657).

1 files changed, 9 insertions, 0 deletions

diff --git a/mocker.py b/mocker.py
index 336e696..ae7970c 100644
--- a/mocker.py
+++ b/mocker.py
@@ -189,6 +189,15 @@ class MockerTestCase(unittest.TestCase):
 
         super(MockerTestCase, self).__init__(methodName)
 
+    def __call__(self, *args, **kwargs):
+        # This is necessary for Python 2.3 only, because it didn't use run(),
+        # which is supported above.
+        try:
+            super(MockerTestCase, self).__call__(*args, **kwargs)
+        finally:
+            if sys.version_info < (2, 4):
+                self.__cleanup()
+
     def __cleanup(self):
         for path in self.__cleanup_paths:
             if os.path.isfile(path):