Add cleanup for measurements that use tracing

tools/telemetry/telemetry/page/page_measurement_unittest_base.py

 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import unittest
 
 from telemetry.core import util
 from telemetry.page import page_runner
 from telemetry.page import page as page_module
 from telemetry.page import page_set
@@ skipping 33 matching lines @@
         continue
       setattr(options, k, v)
 
     measurement.CustomizeBrowserOptions(options)
     options.output_file = None
     options.output_format = 'none'
     options.output_trace_tag = None
     page_runner.ProcessCommandLineArgs(temp_parser, options)
     measurement.ProcessCommandLineArgs(temp_parser, options)
     return page_runner.Run(measurement, ps, expectations, options)
+
+  def TestTracingCleanedUp(self, measurement_class, options=None):
+    ps = self.CreatePageSetFromFileInUnittestDataDir('blank.html')
+    start_tracing_called = [False]
+    tracing_closed_called = [False]
+
+    class BuggyMeasurement(measurement_class):
+      def __init__(self, *args, **kwargs):
+        measurement_class.__init__(self, *args, **kwargs)
+
+      # Inject fake tracing methods to browser
+      def TabForPage(self, page, browser):
+        ActualStartTracing = browser.StartTracing
+        def FakeStartTracing(*args, **kwargs):
+          ActualStartTracing(*args, **kwargs)
+          start_tracing_called[0] = True
+          raise Exception('Intentional exception')
+        browser.StartTracing = FakeStartTracing
+
+        ActualStopTracing = browser.StopTracing
+        def FakeStopTracing(*args, **kwargs):
+          ActualStopTracing(*args, **kwargs)
+          tracing_closed_called[0] = True
+        browser.StopTracing = FakeStopTracing
+
+        return measurement_class.TabForPage(self, page, browser)
+
+    measurement = BuggyMeasurement()
+    self.RunMeasurement(measurement, ps, options=options)
+    assert start_tracing_called[0], ("%s measurement doesn't start tracing" %
+                                     measurement_class)
+    self.assertTrue(tracing_closed_called[0])