From 44c1f801a3068647673c7e827827bab7cc72264c Mon Sep 17 00:00:00 2001 From: Duncan McGreggor Date: Fri, 13 Feb 2009 05:49:02 -0600 Subject: - 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. --- test.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'test.py') 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: -- cgit v1.2.1