Support tags on the ExtendedToOriginal decorator.

2 files changed, 26 insertions, 0 deletions

diff --git a/python/subunit/test_results.py b/python/subunit/test_results.py
index 49fc731..f73d747 100644
--- a/python/subunit/test_results.py
+++ b/python/subunit/test_results.py
@@ -109,6 +109,9 @@ class TestResultDecorator(object):
     def stop(self):
         return self.decorated.stop()
 
+    def tags(self, gone_tags, new_tags):
+        return self._call_maybe("tags", None, gone_tags, new_tags)
+
     def time(self, a_datetime):
         return self._call_maybe("time", None, a_datetime)
 
@@ -352,3 +355,10 @@ class ExtendedToOriginalDecorator(object):
             return self.decorated.stopTestRun()
         except AttributeError:
             return
+
+    def tags(self, new_tags, gone_tags):
+        method = getattr(self.decorated, 'tags', None)
+        if method is None:
+            return
+        return method(new_tags, gone_tags)
+
diff --git a/python/subunit/tests/test_test_results.py b/python/subunit/tests/test_test_results.py
index 7400f0b..b7a26d5 100644
--- a/python/subunit/tests/test_test_results.py
+++ b/python/subunit/tests/test_test_results.py
@@ -134,6 +134,9 @@ class ExtendedTestResult(Python27TestResult):
     def progress(self, offset, whence):
         self._calls.append(('progress', offset, whence))
 
+    def tags(self, new_tags, gone_tags):
+        self._calls.append(('tags', new_tags, gone_tags))
+
 
 class TestExtendedToOriginalResultDecoratorBase(unittest.TestCase):
 
@@ -313,6 +316,19 @@ class TestExtendedToOriginalResultDecorator(
         self.converter.stopTestRun()
         self.assertEqual([('stopTestRun',)], self.result._calls)
 
+    def test_tags_py26(self):
+        self.make_26_result()
+        self.converter.tags(1, 2)
+
+    def test_tags_py27(self):
+        self.make_27_result()
+        self.converter.tags(1, 2)
+
+    def test_tags_pyextended(self):
+        self.make_extended_result()
+        self.converter.tags(1, 2)
+        self.assertEqual([('tags', 1, 2)], self.result._calls)
+
 
 class TestExtendedToOriginalAddError(TestExtendedToOriginalResultDecoratorBase):