Allow the predicate to filter tags.

1 files changed, 10 insertions, 0 deletions

diff --git a/python/subunit/tests/test_test_results.py b/python/subunit/tests/test_test_results.py
index 6beb57a..2bec7e3 100644
--- a/python/subunit/tests/test_test_results.py
+++ b/python/subunit/tests/test_test_results.py
@@ -56,6 +56,16 @@ class AssertBeforeTestResult(LoggingDecorator):
         super(AssertBeforeTestResult, self)._before_event()
 
 
+class TestTestResultDecorator(unittest.TestCase):
+
+    def test_current_tags(self):
+        result = ExtendedTestResult()
+        decorator = subunit.test_results.TestResultDecorator(result)
+        decorator.tags(set('foo'), set())
+        self.assertEqual(set('foo'), decorator.current_tags)
+        self.assertEqual(decorator.current_tags, result.current_tags)
+
+
 class TimeCapturingResult(unittest.TestResult):
 
     def __init__(self):