Support the time protocol on ExtendedToOriginalDecorator.

2 files changed, 21 insertions, 0 deletions

diff --git a/python/subunit/test_results.py b/python/subunit/test_results.py
index f73d747..fad4760 100644
--- a/python/subunit/test_results.py
+++ b/python/subunit/test_results.py
@@ -362,3 +362,8 @@ class ExtendedToOriginalDecorator(object):
             return
         return method(new_tags, gone_tags)
 
+    def time(self, a_datetime):
+        method = getattr(self.decorated, 'time', None)
+        if method is None:
+            return
+        return method(a_datetime)
diff --git a/python/subunit/tests/test_test_results.py b/python/subunit/tests/test_test_results.py
index b7a26d5..d333c10 100644
--- a/python/subunit/tests/test_test_results.py
+++ b/python/subunit/tests/test_test_results.py
@@ -137,6 +137,9 @@ class ExtendedTestResult(Python27TestResult):
     def tags(self, new_tags, gone_tags):
         self._calls.append(('tags', new_tags, gone_tags))
 
+    def time(self, time):
+        self._calls.append(('time', time))
+
 
 class TestExtendedToOriginalResultDecoratorBase(unittest.TestCase):
 
@@ -329,6 +332,19 @@ class TestExtendedToOriginalResultDecorator(
         self.converter.tags(1, 2)
         self.assertEqual([('tags', 1, 2)], self.result._calls)
 
+    def test_time_py26(self):
+        self.make_26_result()
+        self.converter.time(1)
+
+    def test_time_py27(self):
+        self.make_27_result()
+        self.converter.time(1)
+
+    def test_time_pyextended(self):
+        self.make_extended_result()
+        self.converter.time(1)
+        self.assertEqual([('time', 1)], self.result._calls)
+
 
 class TestExtendedToOriginalAddError(TestExtendedToOriginalResultDecoratorBase):