Index: tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/safebrowsing.py |
diff --git a/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/safebrowsing.py b/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/safebrowsing.py |
index 9047ae6e4bf0fe1865fae0c008105ebf80481256..5c47806c12d0867b02132eea37d8b796b23fc410 100644 |
--- a/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/safebrowsing.py |
+++ b/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/safebrowsing.py |
@@ -5,7 +5,7 @@ import logging |
from telemetry.core import exceptions |
from telemetry.page import page as page_module |
-from telemetry.page import page_set as page_set_module |
+from telemetry import story |
class SafebrowsingPage(page_module.Page): |
@@ -29,12 +29,12 @@ class SafebrowsingPage(page_module.Page): |
raise e |
-class SafebrowsingPageSet(page_set_module.PageSet): |
+class SafebrowsingStorySet(story.StorySet): |
""" Chrome proxy test sites """ |
def __init__(self, expect_timeout=False): |
- super(SafebrowsingPageSet, self).__init__() |
+ super(SafebrowsingStorySet, self).__init__() |
- self.AddUserStory( |
+ self.AddStory( |
SafebrowsingPage('http://www.ianfette.org/', self, expect_timeout)) |