Index: tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/fallback_viaheader.py |
diff --git a/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/fallback_viaheader.py b/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/fallback_viaheader.py |
index a1ec2ca2aab4c963debaefc8ca741ff05b893012..8b3dc56acf824120d586f4e3641b37fe91d44db1 100644 |
--- a/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/fallback_viaheader.py |
+++ b/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/fallback_viaheader.py |
@@ -3,7 +3,7 @@ |
# found in the LICENSE file. |
from telemetry.page import page as page_module |
-from telemetry.page import page_set as page_set_module |
+from telemetry import story |
class FallbackViaHeaderPage(page_module.Page): |
@@ -12,11 +12,11 @@ class FallbackViaHeaderPage(page_module.Page): |
super(FallbackViaHeaderPage, self).__init__(url=url, page_set=page_set) |
-class FallbackViaHeaderPageSet(page_set_module.PageSet): |
+class FallbackViaHeaderStorySet(story.StorySet): |
""" Chrome proxy test sites """ |
def __init__(self): |
- super(FallbackViaHeaderPageSet, self).__init__() |
+ super(FallbackViaHeaderStorySet, self).__init__() |
urls_list = [ |
'http://chromeproxy-test.appspot.com/default?respStatus=200', |
@@ -24,4 +24,4 @@ class FallbackViaHeaderPageSet(page_set_module.PageSet): |
] |
for url in urls_list: |
- self.AddUserStory(FallbackViaHeaderPage(url, self)) |
+ self.AddStory(FallbackViaHeaderPage(url, self)) |