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..556a5fdef5aae363a4e9ebc54eca102d5d84c499 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.story import story_set as story_set_module |
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_set_module.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)) |