Index: tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/__init__.py |
diff --git a/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/__init__.py b/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/__init__.py |
index 9f24d4d4af7cebf3a0b12b1e76ec198a02a4ceee..799c2462114f876d084458af5fea9785446251c1 100644 |
--- a/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/__init__.py |
+++ b/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/__init__.py |
@@ -7,21 +7,21 @@ import os |
import sys |
from telemetry.core import discover |
-from telemetry.page import page_set |
+from telemetry import story |
import video |
-# Import all submodules' PageSet classes. |
+# Import all submodules' StorySet classes. |
start_dir = os.path.dirname(os.path.abspath(__file__)) |
top_level_dir = os.path.abspath(os.path.join(start_dir, os.pardir, os.pardir)) |
-base_class = page_set.PageSet |
+base_class = story.StorySet |
for cls in discover.DiscoverClasses( |
start_dir, top_level_dir, base_class).values(): |
setattr(sys.modules[__name__], cls.__name__, cls) |
# DiscoverClasses makes the assumption that there is exactly one matching |
# class per file, however the following are declared in the same file. |
-for cls in (video.VideoDirectPageSet, |
- video.VideoProxiedPageSet, |
- video.VideoComparePageSet): |
+for cls in (video.VideoDirectStorySet, |
+ video.VideoProxiedStorySet, |
+ video.VideoCompareStorySet): |
setattr(sys.modules[__name__], cls.__name__, cls) |