@attr can now be used as a class decorator (Issue 292)

2 files changed, 14 insertions, 0 deletions

diff --git a/functional_tests/support/att/test_attr.py b/functional_tests/support/att/test_attr.py
index 20e1e49..f9b8ea5 100644
--- a/functional_tests/support/att/test_attr.py
+++ b/functional_tests/support/att/test_attr.py
@@ -1,3 +1,4 @@
+from nose.plugins.attrib import attr
 import unittest
 
 def test_one():
@@ -41,3 +42,14 @@ class Something(unittest.TestCase):
     
     def test_case_three(self):
         pass
+
+
+class TestAttrClass:
+
+    def test_one(self):
+        pass
+
+    def test_two(self):
+        pass
+
+TestAttrClass = attr('a')(TestAttrClass)
diff --git a/functional_tests/test_attribute_plugin.py b/functional_tests/test_attribute_plugin.py
index f62594e..d3bd779 100644
--- a/functional_tests/test_attribute_plugin.py
+++ b/functional_tests/test_attribute_plugin.py
@@ -28,6 +28,8 @@ class TestSimpleAttribute(PluginTester, unittest.TestCase):
         assert 'test_case_two' not in self.output
         assert 'test_case_one' not in self.output
         assert 'test_case_three' not in self.output
+        assert 'TestAttrClass.test_one ... ok' in self.output
+        assert 'TestAttrClass.test_two ... ok' in self.output
 
 
 class TestNotSimpleAttribute(PluginTester, unittest.TestCase):