| Index: tools/telemetry/telemetry/page/page_test_unittest.py
|
| diff --git a/tools/telemetry/telemetry/page/page_test_unittest.py b/tools/telemetry/telemetry/page/page_test_unittest.py
|
| index 0654d791e6d0b79f7aabb88fea03e4aa85b8edf2..5d2ec7d9ad3225e2b7f1a759912f1a50e026e265 100644
|
| --- a/tools/telemetry/telemetry/page/page_test_unittest.py
|
| +++ b/tools/telemetry/telemetry/page/page_test_unittest.py
|
| @@ -10,10 +10,10 @@ from telemetry.core import exceptions
|
| from telemetry.core import wpr_modes
|
| from telemetry.page import page as page_module
|
| from telemetry.page import page_set
|
| -from telemetry.page import page_set_archive_info
|
| from telemetry.page import page_test
|
| from telemetry.unittest_util import options_for_unittests
|
| from telemetry.unittest_util import page_test_test_case
|
| +from telemetry.wpr import archive_info
|
|
|
|
|
| class PageTestThatFails(page_test.PageTest):
|
| @@ -109,7 +109,7 @@ class PageTestUnitTest(page_test_test_case.PageTestTestCase):
|
| self._options.browser_options.wpr_mode = wpr_modes.WPR_RECORD
|
|
|
| # pylint: disable=protected-access
|
| - ps._wpr_archive_info = page_set_archive_info.PageSetArchiveInfo(
|
| + ps._wpr_archive_info = archive_info.WprArchiveInfo(
|
| '', '', ps.bucket, json.loads(archive_info_template %
|
| (test_archive, google_url)))
|
| ps.pages = [page_module.Page(google_url, ps)]
|
| @@ -120,7 +120,7 @@ class PageTestUnitTest(page_test_test_case.PageTestTestCase):
|
| self._options.browser_options.wpr_mode = wpr_modes.WPR_REPLAY
|
|
|
| # pylint: disable=protected-access
|
| - ps._wpr_archive_info = page_set_archive_info.PageSetArchiveInfo(
|
| + ps._wpr_archive_info = archive_info.WprArchiveInfo(
|
| '', '', ps.bucket, json.loads(archive_info_template %
|
| (test_archive, foo_url)))
|
| ps.pages = [page_module.Page(foo_url, ps)]
|
| @@ -128,7 +128,7 @@ class PageTestUnitTest(page_test_test_case.PageTestTestCase):
|
| self.assertEquals(1, len(all_results.failures))
|
|
|
| # pylint: disable=protected-access
|
| - ps._wpr_archive_info = page_set_archive_info.PageSetArchiveInfo(
|
| + ps._wpr_archive_info = archive_info.WprArchiveInfo(
|
| '', '', ps.bucket, json.loads(archive_info_template %
|
| (test_archive, google_url)))
|
| ps.pages = [page_module.Page(google_url, ps)]
|
|
|