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) |