In MockerTestCase's constructor, copy attributes from the original method

into the wrapper.

1 files changed, 6 insertions, 1 deletions

diff --git a/mocker.py b/mocker.py
index 1a10dda..8929b91 100644
--- a/mocker.py
+++ b/mocker.py
@@ -107,7 +107,12 @@ class MockerTestCase(unittest.TestCase):
                     self.__cleanup()
                     self.mocker.restore()
                     self.mocker.verify()
-            test_method_wrapper.__doc__ = test_method.__doc__
+            # Copy all attributes from the original method..
+            for attr in dir(test_method):
+                # .. unless they're present in our wrapper already.
+                if not hasattr(test_method_wrapper, attr) or attr == "__doc__":
+                    setattr(test_method_wrapper, attr,
+                            getattr(test_method, attr))
             setattr(self, methodName, test_method_wrapper)
 
         self.mocker = Mocker()