Upstream version 10.39.225.0
[platform/framework/web/crosswalk.git] / src / tools / telemetry / telemetry / unittest / gtest_progress_reporter_unittest.py
index 8769770..9d01239 100644 (file)
@@ -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()