[Telemetry] Move from telemetry WaitFor to py_utils WaitFor

telemetry/telemetry/internal/actions/action_runner_unittest.py

 # Copyright 2014 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 mock
 import unittest
 
 from telemetry.core import exceptions
 from telemetry import decorators
 from telemetry.internal.actions import action_runner as action_runner_module
 from telemetry.internal.actions import page_action
 from telemetry.testing import tab_test_case
 from telemetry.timeline import chrome_trace_category_filter
 from telemetry.timeline import model
 from telemetry.timeline import tracing_config
 from telemetry.web_perf import timeline_interaction_record as tir_module
 
+import py_utils
+
 
 class ActionRunnerInteractionTest(tab_test_case.TabTestCase):
 
   def GetInteractionRecords(self, trace_data):
     timeline_model = model.TimelineModel(trace_data)
     renderer_thread = timeline_model.GetRendererThreadFromTabId(self._tab.id)
     return [
         tir_module.TimelineInteractionRecord.FromAsyncEvent(e)
         for e in renderer_thread.async_slices
         if tir_module.IsTimelineInteractionRecord(e.name)
@@ skipping 177 matching lines @@
     action_runner.ExecuteJavaScript(
         '(function() {'
         '  var el = document.createElement("div");'
         '  el.id = "test1";'
         '  el.textContent = "foo";'
         '  document.body.appendChild(el);'
         '})()')
     action_runner.WaitForElement('#test1', timeout_in_seconds=0.2)
     def WaitForElement():
       action_runner.WaitForElement(text='oo', timeout_in_seconds=0.2)
-    self.assertRaises(exceptions.TimeoutException, WaitForElement)
+    self.assertRaises(py_utils.TimeoutException, WaitForElement)
 
   def testClickElement(self):
     self.Navigate('page_with_clickables.html')
     action_runner = action_runner_module.ActionRunner(self._tab,
                                                       skip_waits=True)
 
     action_runner.ExecuteJavaScript('valueSettableByTest = 1;')
     action_runner.ClickElement('#test')
     self.assertEqual(1, action_runner.EvaluateJavaScript('valueToTest'))
 
@@ skipping 179 matching lines @@
     with self.assertRaises(FooException):
       with action_runner_module.Interaction(
           self.mock_action_runner, label='ABC', flags=[]):
         raise FooException()
 
     # Test that the end console.timeEnd(...) isn't called because exception was
     # raised.
     expected_calls = [
         mock.call.ExecuteJavaScript('console.time("Interaction.ABC");')]
     self.assertEqual(expected_calls, self.mock_action_runner.mock_calls)