| 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()
|
|
|