Index: tools/telemetry/telemetry/testing/story_set_smoke_test.py |
diff --git a/tools/telemetry/telemetry/testing/story_set_smoke_test.py b/tools/telemetry/telemetry/testing/story_set_smoke_test.py |
index d0a8d575bf6eb2dba1bba0d4ed26bf68420ad43d..008773f5f7eaf8c4b0848cb0651ecded627abdcb 100644 |
--- a/tools/telemetry/telemetry/testing/story_set_smoke_test.py |
+++ b/tools/telemetry/telemetry/testing/story_set_smoke_test.py |
@@ -6,10 +6,10 @@ |
import os |
import unittest |
+from telemetry.core import discover |
from telemetry.internal.browser import browser_credentials |
from telemetry import page |
from telemetry import story as story_module |
-from telemetry.util import classes_util |
from telemetry.wpr import archive_info |
@@ -23,9 +23,9 @@ |
def GetAllStorySetClasses(self, story_sets_dir, top_level_dir): |
# We can't test page sets that aren't directly constructable since we |
# don't know what arguments to put for the constructor. |
- return classes_util.DiscoverClasses(story_sets_dir, top_level_dir, |
- story_module.StorySet, |
- directly_constructable=True) |
+ return discover.DiscoverClasses(story_sets_dir, top_level_dir, |
+ story_module.StorySet, |
+ directly_constructable=True).values() |
def CheckArchive(self, story_set): |
"""Verify that all URLs of pages in story_set have an associated archive.""" |