Chromium Code Reviews| Index: tools/telemetry/telemetry/page/shared_page_state_unittest.py |
| diff --git a/tools/telemetry/telemetry/page/shared_page_state_unittest.py b/tools/telemetry/telemetry/page/shared_page_state_unittest.py |
| index b65eaa6b4fa419ff5664df323c306a43109780b1..8b6e61166d97623d1fb517d0ad7be19b5d2e51ff 100644 |
| --- a/tools/telemetry/telemetry/page/shared_page_state_unittest.py |
| +++ b/tools/telemetry/telemetry/page/shared_page_state_unittest.py |
| @@ -11,8 +11,8 @@ from telemetry.page import page |
| from telemetry.page import page_set |
| from telemetry.page import page_test |
| from telemetry.page import shared_page_state |
| +from telemetry.story import story_set_module |
|
nednguyen
2015/05/04 23:30:24
Same here
aiolos (Not reviewing)
2015/05/09 03:53:59
Done.
|
| from telemetry.unittest_util import options_for_unittests |
| -from telemetry.user_story import user_story_set |
| def SetUpPageRunnerArguments(options): |
| @@ -74,9 +74,9 @@ class SharedPageStateTests(unittest.TestCase): |
| 'http://www.google.com', |
| shared_page_state_class=shared_page_state_class) |
| test = DummyTest() |
| - uss = user_story_set.UserStorySet() |
| - uss.AddUserStory(us) |
| - us.shared_state_class(test, self.options, uss) |
| + story_set = story_set_module.StorySet() |
| + story_set.AddUserStory(us) |
| + us.shared_state_class(test, self.options, story_set) |
| browser_options = self.options.browser_options |
| actual_user_agent = browser_options.browser_user_agent_type |
| self.assertEqual(expected_user_agent, actual_user_agent) |