| Index: telemetry/telemetry/internal/actions/media_action.py
|
| diff --git a/telemetry/telemetry/internal/actions/media_action.py b/telemetry/telemetry/internal/actions/media_action.py
|
| index 438f5259a186772c2d9f77092219b73c3d456f52..ee07c56220201e49ddcafdeab1c598a3314ad4f4 100644
|
| --- a/telemetry/telemetry/internal/actions/media_action.py
|
| +++ b/telemetry/telemetry/internal/actions/media_action.py
|
| @@ -6,10 +6,11 @@
|
|
|
| import logging
|
|
|
| -from telemetry.core import util
|
| from telemetry.internal.actions import page_action
|
| from telemetry.internal.actions import utils
|
|
|
| +import py_utils
|
| +
|
|
|
| class MediaAction(page_action.PageAction):
|
| def WillRunAction(self, tab):
|
| @@ -29,9 +30,9 @@ class MediaAction(page_action.PageAction):
|
| timeout_in_seconds: Timeout to check for event, throws an exception if
|
| not fired.
|
| """
|
| - util.WaitFor(lambda:
|
| - self.HasEventCompletedOrError(tab, selector, event_name),
|
| - timeout=timeout_in_seconds)
|
| + py_utils.WaitFor(
|
| + lambda: self.HasEventCompletedOrError(tab, selector, event_name),
|
| + timeout=timeout_in_seconds)
|
|
|
| def HasEventCompletedOrError(self, tab, selector, event_name):
|
| if tab.EvaluateJavaScript(
|
|
|