OLD | NEW |
1 # Copyright 2015 The Chromium Authors. All rights reserved. | 1 # Copyright 2015 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 from page_sets import pregenerated_large_profile_shared_state | 4 from page_sets import pregenerated_large_profile_shared_state |
| 5 from telemetry import story |
5 from telemetry.page import page as page_module | 6 from telemetry.page import page as page_module |
6 from telemetry.page import page_set as page_set_module | |
7 | 7 |
8 | 8 |
9 class BlankPageWithLargeProfile(page_module.Page): | 9 class BlankPageWithLargeProfile(page_module.Page): |
10 def __init__(self, url, page_set): | 10 def __init__(self, url, page_set): |
11 super(BlankPageWithLargeProfile, self).__init__( | 11 super(BlankPageWithLargeProfile, self).__init__( |
12 url=url, page_set=page_set, | 12 url=url, page_set=page_set, |
13 shared_page_state_class=pregenerated_large_profile_shared_state. | 13 shared_page_state_class=pregenerated_large_profile_shared_state. |
14 PregeneratedLargeProfileSharedState) | 14 PregeneratedLargeProfileSharedState) |
15 | 15 |
16 | 16 |
17 class BlankPageSetWithLargeProfile(page_set_module.PageSet): | 17 class BlankPageSetWithLargeProfile(story.StorySet): |
18 """A single blank page loaded with a large profile.""" | 18 """A single blank page loaded with a large profile.""" |
19 | 19 |
20 def __init__(self): | 20 def __init__(self): |
21 super(BlankPageSetWithLargeProfile, self).__init__() | 21 super(BlankPageSetWithLargeProfile, self).__init__() |
22 self.AddUserStory(BlankPageWithLargeProfile( | 22 self.AddUserStory(BlankPageWithLargeProfile( |
23 'file://blank_page/blank_page.html', self)) | 23 'file://blank_page/blank_page.html', self)) |
OLD | NEW |