Index: tools/telemetry/telemetry/unittest/gtest_progress_reporter_unittest.py |
diff --git a/tools/telemetry/telemetry/unittest/gtest_progress_reporter_unittest.py b/tools/telemetry/telemetry/unittest/gtest_progress_reporter_unittest.py |
index 8769770ef83407fc3c538e1058e91191bb513e0f..9d012398ef4327f94815b7ab3ac97ac3b70177b2 100644 |
--- a/tools/telemetry/telemetry/unittest/gtest_progress_reporter_unittest.py |
+++ b/tools/telemetry/telemetry/unittest/gtest_progress_reporter_unittest.py |
@@ -55,12 +55,10 @@ class GTestProgressReporterTest(unittest.TestCase): |
self._formatter = gtest_progress_reporter.GTestProgressReporter( |
self._stream) |
- self._mock_timer = simple_mock.MockTimer() |
- self._real_time_time = gtest_progress_reporter.time.time |
- gtest_progress_reporter.time.time = self._mock_timer.GetTime |
+ self._mock_timer = simple_mock.MockTimer(gtest_progress_reporter) |
def tearDown(self): |
- gtest_progress_reporter.time.time = self._real_time_time |
+ self._mock_timer.Restore() |
def testTestSuiteWithWrapperSuite(self): |
suite = unittest.TestSuite() |