[Python-checkins] r86104 - in python/branches/release27-maint: Lib/unittest/result.py Lib/unittest/suite.py Lib/unittest/test/test_suite.py Misc/NEWS

[michael.foord]

Author: michael.foord
Date: Mon Nov  1 23:11:53 2010
New Revision: 86104

Log:
Merged revisions 86101 via svnmerge from 
svn+ssh://pythondev@svn.python.org/python/branches/py3k

........
  r86101 | michael.foord | 2010-11-01 21:09:03 +0000 (Mon, 01 Nov 2010) | 1 line
  
  Fix issue 9926. TestSuite subclasses that override __call__ are called correctly.
........


Modified:
   python/branches/release27-maint/   (props changed)
   python/branches/release27-maint/Lib/unittest/result.py
   python/branches/release27-maint/Lib/unittest/suite.py
   python/branches/release27-maint/Lib/unittest/test/test_suite.py
   python/branches/release27-maint/Misc/NEWS

Modified: python/branches/release27-maint/Lib/unittest/result.py
==============================================================================
--- python/branches/release27-maint/Lib/unittest/result.py	(original)
+++ python/branches/release27-maint/Lib/unittest/result.py	Mon Nov  1 23:11:53 2010
@@ -35,6 +35,7 @@
     formatted traceback of the error that occurred.
     """
     _previousTestClass = None
+    _testRunEntered = False
     _moduleSetUpFailed = False
     def __init__(self, stream=None, descriptions=None, verbosity=None):
         self.failfast = False

Modified: python/branches/release27-maint/Lib/unittest/suite.py
==============================================================================
--- python/branches/release27-maint/Lib/unittest/suite.py	(original)
+++ python/branches/release27-maint/Lib/unittest/suite.py	Mon Nov  1 23:11:53 2010
@@ -80,23 +80,11 @@
     subclassing, do not forget to call the base class constructor.
     """
 
+    def run(self, result, debug=False):
+        topLevel = False
+        if getattr(result, '_testRunEntered', False) is False:
+            result._testRunEntered = topLevel = True
 
-    def run(self, result):
-        self._wrapped_run(result)
-        self._tearDownPreviousClass(None, result)
-        self._handleModuleTearDown(result)
-        return result
-
-    def debug(self):
-        """Run the tests without collecting errors in a TestResult"""
-        debug = _DebugResult()
-        self._wrapped_run(debug, True)
-        self._tearDownPreviousClass(None, debug)
-        self._handleModuleTearDown(debug)
-
-    ################################
-    # private methods
-    def _wrapped_run(self, result, debug=False):
         for test in self:
             if result.shouldStop:
                 break
@@ -111,13 +99,23 @@
                     getattr(result, '_moduleSetUpFailed', False)):
                     continue
 
-            if hasattr(test, '_wrapped_run'):
-                test._wrapped_run(result, debug)
-            elif not debug:
+            if not debug:
                 test(result)
             else:
                 test.debug()
 
+        if topLevel:
+            self._tearDownPreviousClass(None, result)
+            self._handleModuleTearDown(result)
+        return result
+
+    def debug(self):
+        """Run the tests without collecting errors in a TestResult"""
+        debug = _DebugResult()
+        self.run(debug, True)
+
+    ################################
+
     def _handleClassSetUp(self, test, result):
         previousClass = getattr(result, '_previousTestClass', None)
         currentClass = test.__class__

Modified: python/branches/release27-maint/Lib/unittest/test/test_suite.py
==============================================================================
--- python/branches/release27-maint/Lib/unittest/test/test_suite.py	(original)
+++ python/branches/release27-maint/Lib/unittest/test/test_suite.py	Mon Nov  1 23:11:53 2010
@@ -345,5 +345,19 @@
         self.assertEqual(result.testsRun, 2)
 
 
+    def test_overriding_call(self):
+        class MySuite(unittest.TestSuite):
+            called = False
+            def __call__(self, *args, **kw):
+                self.called = True
+                unittest.TestSuite.__call__(self, *args, **kw)
+
+        suite = MySuite()
+        wrapper = unittest.TestSuite()
+        wrapper.addTest(suite)
+        wrapper(unittest.TestResult())
+        self.assertTrue(suite.called)
+
+
 if __name__ == '__main__':
     unittest.main()

Modified: python/branches/release27-maint/Misc/NEWS
==============================================================================
--- python/branches/release27-maint/Misc/NEWS	(original)
+++ python/branches/release27-maint/Misc/NEWS	Mon Nov  1 23:11:53 2010
@@ -66,6 +66,8 @@
 Library
 -------
 
+- Issue 120176: Wrapped TestSuite subclass does not get __call__ executed
+
 - Issue 6706: asyncore accept() method no longer raises EWOULDBLOCK/ECONNABORTED
   on incomplete connection attempt but returns None instead.