| Index: telemetry/telemetry/internal/browser/browser_unittest.py
|
| diff --git a/telemetry/telemetry/internal/browser/browser_unittest.py b/telemetry/telemetry/internal/browser/browser_unittest.py
|
| index 923d48257afdbff116cb999093eae55df68a8ffb..9c9b3edbb13a2ecfae8c88069522c496ae4a9fd9 100644
|
| --- a/telemetry/telemetry/internal/browser/browser_unittest.py
|
| +++ b/telemetry/telemetry/internal/browser/browser_unittest.py
|
| @@ -8,7 +8,6 @@ import shutil
|
| import tempfile
|
| import unittest
|
|
|
| -from telemetry.core import util
|
| from telemetry.core import exceptions
|
| from telemetry import decorators
|
| from telemetry.internal.browser import browser as browser_module
|
| @@ -22,7 +21,7 @@ from telemetry.testing import options_for_unittests
|
| from telemetry.timeline import tracing_config
|
|
|
| import mock
|
| -
|
| +import py_utils
|
|
|
| class IntentionalException(Exception):
|
| pass
|
| @@ -248,7 +247,8 @@ class BrowserRestoreSessionTest(unittest.TestCase):
|
| # old tabs and a new blank tab.
|
| expected_number_of_tabs = self._number_of_tabs + 1
|
| try:
|
| - util.WaitFor(lambda: len(browser.tabs) == expected_number_of_tabs, 10)
|
| + py_utils.WaitFor(
|
| + lambda: len(browser.tabs) == expected_number_of_tabs, 10)
|
| except:
|
| logging.error('Number of tabs is %s' % len(browser.tabs))
|
| raise
|
|
|