| Index: tools/telemetry/unittest_data/test_page_set.py
|
| diff --git a/tools/telemetry/unittest_data/test_page_set.py b/tools/telemetry/unittest_data/test_page_set.py
|
| index 0dd8a221f5a8ffa99696e2b285adf094fa1c1745..5f8bb4f039de7f8d5a70cc3da0a9d1025ae99836 100644
|
| --- a/tools/telemetry/unittest_data/test_page_set.py
|
| +++ b/tools/telemetry/unittest_data/test_page_set.py
|
| @@ -2,26 +2,27 @@
|
| # Use of this source code is governed by a BSD-style license that can be
|
| # found in the LICENSE file.
|
|
|
| -from telemetry.page.page_set import PageSet
|
| -from telemetry.page.page import Page
|
| +from telemetry.page import page_set
|
| +from telemetry.page import page
|
|
|
| from unittest_data.pages.external_page import ExternalPage
|
|
|
| -class InternalPage(Page):
|
| +class InternalPage(page.Page):
|
| def __init__(self, ps):
|
| super(InternalPage, self).__init__('file://bar.html', page_set=ps)
|
|
|
| -class TestPageSet(PageSet):
|
| +class TestPageSet(page_set.PageSet):
|
| """A pageset for testing purpose"""
|
|
|
| def __init__(self):
|
| super(TestPageSet, self).__init__(
|
| archive_data_file='data/test.json',
|
| credentials_path='data/credential',
|
| - user_agent_type='desktop')
|
| + user_agent_type='desktop',
|
| + bucket=page_set.PUBLIC_BUCKET)
|
|
|
| #top google property; a google tab is often open
|
| - class Google(Page):
|
| + class Google(page.Page):
|
| def __init__(self, ps):
|
| super(Google, self).__init__('https://www.google.com', page_set=ps)
|
|
|
|
|