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