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 910392815e8f3507de471c5f45712f36b82b0c75..9f24d4d4af7cebf3a0b12b1e76ec198a02a4ceee 100644 |
--- a/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/__init__.py |
+++ b/tools/chrome_proxy/integration_tests/chrome_proxy_pagesets/__init__.py |
@@ -8,6 +8,7 @@ import sys |
from telemetry.core import discover |
from telemetry.page import page_set |
+import video |
# Import all submodules' PageSet classes. |
@@ -17,3 +18,10 @@ base_class = page_set.PageSet |
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): |
+ setattr(sys.modules[__name__], cls.__name__, cls) |