Index: tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/block_once.py |
diff --git a/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/block_once.py b/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/block_once.py |
index ce3fd0da43208f5431251a5c07cd0381d63ea01a..07f0e0821f44c00ee60b9731f08168a3ada78d1d 100644 |
--- a/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/block_once.py |
+++ b/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/block_once.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 BlockOncePage(page_module.Page): |
@@ -31,12 +31,12 @@ class BlockOncePage(page_module.Page): |
''') |
action_runner.Wait(1) |
-class BlockOncePageSet(page_set_module.PageSet): |
+class BlockOnceStorySet(story.StorySet): |
""" Chrome proxy test sites """ |
def __init__(self): |
- super(BlockOncePageSet, self).__init__() |
+ super(BlockOnceStorySet, self).__init__() |
# Test block-once for a GET request. |
urls_list = [ |
@@ -44,4 +44,4 @@ class BlockOncePageSet(page_set_module.PageSet): |
] |
for url in urls_list: |
- self.AddUserStory(BlockOncePage(url, self)) |
+ self.AddStory(BlockOncePage(url, self)) |