From ecf9af693a3e5f89a8854f53f0a5a1f14fed081d Mon Sep 17 00:00:00 2001 From: Robert Collins Date: Mon, 19 Oct 2009 18:54:57 +1100 Subject: Support the time protocol on ExtendedToOriginalDecorator. --- python/subunit/test_results.py | 5 +++++ python/subunit/tests/test_test_results.py | 16 ++++++++++++++++ 2 files changed, 21 insertions(+) (limited to 'python') 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): -- cgit v1.2.1