Merge pull request #772 from jszakmeister/fix-771

Fix #771: attr plugin is broken when parent overrides child method

CHANGELOG

@@ -64,6 +64,9 @@ In Development
 - Display the report summary and stack traces even if Ctrl-C was pressed
   during the test run.
   Patch by Kevin Qiu.
+- Fix #771: attr plugin is broken when parent and child share same method
+  name with different attributes
+  Patch by John Szakmeister.  Test case provided by Thomas Grainger.
 
 1.3.0
 

functional_tests/support/issue771/test.py

@@ -0,0 +1,16 @@
+from nose.plugins.attrib import attr
+
+from unittest import TestCase
+
+@attr("b")
+def test_b():
+    assert 1 == 1
+
+class TestBase(TestCase):
+    def test_a(self):
+        assert 1 == 1
+
+class TestDerived(TestBase):
+    @attr("a")
+    def test_a(self):
+        assert 1 == 1

functional_tests/test_attribute_plugin.py

@@ -156,6 +156,20 @@ def verify(self):
         assert 'test_case_three' not in self.output
 
 
+# Issue #771
+class TestTopLevelNotSelected(AttributePluginTester):
+    suitepath = os.path.join(support, 'issue771')
+    args = ["-a", "!a"]
+
+    def verify(self):
+        # Note: a failure here may mean that the test case selection is broken
+        # rather than the attribute plugin, but the issue more easily manifests
+        # itself when using attributes.
+        assert 'test.test_b ... ok' in self.output
+        assert 'test_a (test.TestBase) ... ok' in self.output
+        assert 'TestDerived' not in self.output
+
+
 if compat_24:
     class TestAttributeEval(AttributePluginTester):
         args = ["-A", "c>20"]

functional_tests/test_program.py

@@ -113,16 +113,8 @@ def test_run_support_twist(self):
         print "-----"
         print repr(res)
 
-        # some versions of twisted.trial.unittest.TestCase have
-        # runTest in the base class -- this is wrong! But we have
-        # to deal with it
-        if hasattr(TestCase, 'runTest'):
-            expect = 5
-        else:
-            expect = 4
-        self.assertEqual(res.testsRun, expect,
-                         "Expected to run %s tests, ran %s" %
-                         (expect, res.testsRun))
+        self.assertEqual(res.testsRun, 4,
+                         "Expected to run 4 tests, ran %s" % (res.testsRun,))
         assert not res.wasSuccessful()
         assert len(res.errors) == 1
 

nose/loader.py

@@ -104,19 +104,17 @@ def getTestCaseNames(self, testCaseClass):
         """
         if self.config.getTestCaseNamesCompat:
             return unittest.TestLoader.getTestCaseNames(self, testCaseClass)
-        
+
         def wanted(attr, cls=testCaseClass, sel=self.selector):
             item = getattr(cls, attr, None)
             if isfunction(item):
                 item = unbound_method(cls, item)
             elif not ismethod(item):
                 return False
             return sel.wantMethod(item)
+
         cases = filter(wanted, dir(testCaseClass))
-        for base in testCaseClass.__bases__:
-            for case in self.getTestCaseNames(base):
-                if case not in cases:
-                    cases.append(case)
+
         # add runTest if nothing else picked
         if not cases and hasattr(testCaseClass, 'runTest'):
             cases = ['runTest']