[Telemetry] Fix exception handling for blink_styles and rasterize_and_record use of WaitFor

tools/perf/measurements/blink_style.py

 # Copyright 2015 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.
 
 from collections import defaultdict
 from itertools import starmap
-from telemetry.core import exceptions
 from telemetry.core import util
 from telemetry.page import legacy_page_test
 from telemetry.value import scalar
 
 from measurements import timeline_controller
+import py_utils
 
 
 class BlinkStyle(legacy_page_test.LegacyPageTest):
 
   def __init__(self):
     super(BlinkStyle, self).__init__()
     self._controller = None
 
   def WillNavigateToPage(self, page, tab):
     self._controller = timeline_controller.TimelineController()
     self._controller.trace_categories = 'blink_style,blink.console'
     self._controller.SetUp(page, tab)
     self._controller.Start(tab)
 
   def DidRunPage(self, platform):
     if self._controller:
       self._controller.CleanUp(platform)
 
   def ValidateAndMeasurePage(self, page, tab, results):
     with tab.action_runner.CreateInteraction('wait-for-quiescence'):
       tab.ExecuteJavaScript('console.time("");')
       try:
         util.WaitFor(tab.HasReachedQuiescence, 15)
-      except exceptions.TimeoutException:
+      except py_utils.TimeoutException:
         # Some sites never reach quiesence. As this benchmark normalizes/
         # categories results, it shouldn't be necessary to reach the same
         # state on every run.
         pass
 
     tab.ExecuteJavaScript('''
         for (var i = 0; i < 11; i++) {
           var cold = i % 2 == 0;
           var name = "update_style";
           if (cold) name += "_cold";
@@ skipping 88 matching lines @@
       tokens = sum(event.tokens for event in events)
       length = sum(event.length for event in events)
 
       results.AddValue(
           scalar.ScalarValue(page, ('parse_css_%s' % category),
                              'tokens/s', 1000 / (parse_duration / tokens)))
 
       results.AddValue(
           scalar.ScalarValue(page, ('tokenize_css_%s' % category),
                              'char/s',  1000 / (tokenize_duration / length)))