| Index: tools/telemetry/telemetry/benchmark_unittest.py
|
| diff --git a/tools/telemetry/telemetry/benchmark_unittest.py b/tools/telemetry/telemetry/benchmark_unittest.py
|
| index 488706ff3a05aeb33e155b4a4f418ac7e502b635..70212a2ff213b5252515ffdc9240343fcf68e8f8 100644
|
| --- a/tools/telemetry/telemetry/benchmark_unittest.py
|
| +++ b/tools/telemetry/telemetry/benchmark_unittest.py
|
| @@ -10,9 +10,9 @@ from telemetry.core import browser_options
|
| 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 shared_user_story_state
|
| 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
|
| @@ -40,14 +40,14 @@ class BenchmarkTest(unittest.TestCase):
|
|
|
| def testPageTestWithIncompatibleUserStory(self):
|
| b = TestBenchmark(user_story.UserStory(
|
| - shared_user_story_state_class=shared_page_state.SharedPageState))
|
| + shared_state_class=shared_page_state.SharedPageState))
|
| with self.assertRaisesRegexp(
|
| Exception, 'containing only telemetry.page.Page user stories'):
|
| b.Run(browser_options.BrowserFinderOptions())
|
|
|
| - state_class = shared_user_story_state.SharedUserStoryState
|
| + state_class = shared_state.SharedState
|
| b = TestBenchmark(user_story.UserStory(
|
| - shared_user_story_state_class=state_class))
|
| + shared_state_class=state_class))
|
| with self.assertRaisesRegexp(
|
| Exception, 'containing only telemetry.page.Page user stories'):
|
| b.Run(browser_options.BrowserFinderOptions())
|
|
|