- Added a new unit test to check for a bug in mocker where the original

  attribute gets swallowed by a subsequent exception check.
- Added a fix for the bug.

1 files changed, 14 insertions, 0 deletions

diff --git a/test.py b/test.py
index e98acb8..ec9578f 100755
--- a/test.py
+++ b/test.py
@@ -3983,6 +3983,20 @@ class PatcherTest(TestCase):
         self.assertEquals(obj.method(), "original")
         self.assertRaises(AssertionError, obj.method)
 
+    def test_original_exception_raised(self):
+        class MyClass(object):
+            def non_existing_attribute(self):
+                return self.bad_attribute
+
+        mock_class = self.mocker.patch(MyClass)
+        mock_class.run()
+        self.mocker.replay()
+        my_class = MyClass()
+        try:
+            my_class.non_existing_attribute()
+        except AttributeError, error:
+            message = "'MyClass' object has no attribute 'bad_attribute'"
+            self.assertEquals(message, error.message)
 
 def main():
     try: