From 3a4c487d0d08f51dca83e31f35aae5ecb3affb82 Mon Sep 17 00:00:00 2001 From: Gustavo Niemeyer Date: Sun, 20 Jun 2010 15:04:54 -0300 Subject: =?UTF-8?q?Fixed=20support=20for=20MockerTestCase.addCleanup()=20i?= =?UTF-8?q?n=20Python=202.3,=20by=20Anders=20F=20Bj=C3=B6rklund=20(#528657?= =?UTF-8?q?).?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- test.py | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) (limited to 'test.py') diff --git a/test.py b/test.py index 21c9d63..864bf7a 100755 --- a/test.py +++ b/test.py @@ -450,6 +450,44 @@ class MockerTestCaseTest(TestCase): self.assertEquals(stash, [[], (1, 2), (3, 4)]) + def test_cleanup_wrapper_in__call__for_2_3(self): + version_info = sys.version_info + __call__ = unittest.TestCase.__call__ + try: + sys.version_info = (2, 3, 5) + stash = [] + def call(self, *args, **kwargs): + self.addCleanup(lambda: stash.append(True)) + unittest.TestCase.__call__ = call + class MyTest(MockerTestCase): + def test_method(self): + pass + MyTest("test_method")() + self.assertEquals(stash, [True]) + finally: + unittest.TestCase.__call__ = __call__ + sys.version_info = version_info + + def test_cleanup_wrapper_in__call__for_2_4(self): + version_info = sys.version_info + __call__ = unittest.TestCase.__call__ + try: + sys.version_info = (2, 4) + stash = [] + def call(self, *args, **kwargs): + self.addCleanup(lambda: stash.append(True)) + unittest.TestCase.__call__ = call + class MyTest(MockerTestCase): + def test_method(self): + pass + MyTest("test_method")() + # Python 2.4+ handles cleanup in run(), registered inside + # MockerTestCase.__init__, so this should *not* work. + self.assertEquals(stash, []) + finally: + unittest.TestCase.__call__ = __call__ + sys.version_info = version_info + def test_twisted_trial_deferred_support(self): calls = [] callbacks = [] -- cgit v1.2.1