Index: tools/telemetry/telemetry/benchmark_unittest.py |
diff --git a/tools/telemetry/telemetry/benchmark_unittest.py b/tools/telemetry/telemetry/benchmark_unittest.py |
index 70212a2ff213b5252515ffdc9240343fcf68e8f8..7c815618901ae47ce0ad3cfe8b19b09159a43353 100644 |
--- a/tools/telemetry/telemetry/benchmark_unittest.py |
+++ b/tools/telemetry/telemetry/benchmark_unittest.py |
@@ -7,13 +7,13 @@ import unittest |
from telemetry import benchmark |
from telemetry.core import browser_options |
+from telemetry.internal import story_runner |
from telemetry import page |
from telemetry.page import page_test |
from telemetry.page import shared_page_state |
from telemetry.story import shared_state |
from telemetry import user_story |
from telemetry.user_story import android |
-from telemetry.user_story import user_story_runner |
from telemetry.user_story import user_story_set as user_story_set_module |
from telemetry.web_perf import timeline_based_measurement |
@@ -58,11 +58,11 @@ class BenchmarkTest(unittest.TestCase): |
b.Run(browser_options.BrowserFinderOptions()) |
def testPageTestWithCompatibleUserStory(self): |
- original_run_fn = user_story_runner.Run |
+ original_run_fn = story_runner.Run |
was_run = [False] |
def RunStub(*_arg, **_kwargs): |
was_run[0] = True |
- user_story_runner.Run = RunStub |
+ story_runner.Run = RunStub |
try: |
options = browser_options.BrowserFinderOptions() |
@@ -75,7 +75,7 @@ class BenchmarkTest(unittest.TestCase): |
b = TestBenchmark(page.Page(url='about:blank')) |
b.Run(options) |
finally: |
- user_story_runner.Run = original_run_fn |
+ story_runner.Run = original_run_fn |
self.assertTrue(was_run[0]) |
@@ -135,7 +135,7 @@ class BenchmarkTest(unittest.TestCase): |
def ValueCanBeAddedPredicate(cls, value, is_first_result): |
return False |
- original_run_fn = user_story_runner.Run |
+ original_run_fn = story_runner.Run |
validPredicate = [False] |
def RunStub(test, user_story_set, expectations, finder_options, results, |
@@ -144,7 +144,7 @@ class BenchmarkTest(unittest.TestCase): |
valid = predicate == PredicateBenchmark.ValueCanBeAddedPredicate |
validPredicate[0] = valid |
- user_story_runner.Run = RunStub |
+ story_runner.Run = RunStub |
try: |
options = browser_options.BrowserFinderOptions() |
@@ -157,6 +157,6 @@ class BenchmarkTest(unittest.TestCase): |
b = PredicateBenchmark(page.Page(url='about:blank')) |
b.Run(options) |
finally: |
- user_story_runner.Run = original_run_fn |
+ story_runner.Run = original_run_fn |
self.assertTrue(validPredicate[0]) |