OLD | NEW |
1 # Copyright 2013 The Chromium Authors. All rights reserved. | 1 # Copyright 2013 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 | 4 |
5 import sys | 5 import sys |
6 import unittest | 6 import unittest |
7 | 7 |
8 from telemetry.core import browser_options | 8 from telemetry.core import browser_options |
| 9 from telemetry.internal import story_runner |
9 from telemetry.results import page_test_results | 10 from telemetry.results import page_test_results |
10 from telemetry.unittest_util import simple_mock | 11 from telemetry.unittest_util import simple_mock |
11 from telemetry.user_story import user_story_runner | |
12 | 12 |
13 from measurements import page_cycler | 13 from measurements import page_cycler |
14 from metrics import keychain_metric | 14 from metrics import keychain_metric |
15 | 15 |
16 | 16 |
17 # Allow testing protected members in the unit test. | 17 # Allow testing protected members in the unit test. |
18 # pylint: disable=W0212 | 18 # pylint: disable=W0212 |
19 | 19 |
20 class MockMemoryMetric(object): | 20 class MockMemoryMetric(object): |
21 """Used instead of simple_mock.MockObject so that the precise order and | 21 """Used instead of simple_mock.MockObject so that the precise order and |
(...skipping 85 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
107 def SetUpCycler(self, page_repeat=1, pageset_repeat=10, cold_load_percent=50, | 107 def SetUpCycler(self, page_repeat=1, pageset_repeat=10, cold_load_percent=50, |
108 report_speed_index=False, setup_memory_module=False): | 108 report_speed_index=False, setup_memory_module=False): |
109 cycler = page_cycler.PageCycler( | 109 cycler = page_cycler.PageCycler( |
110 page_repeat = page_repeat, | 110 page_repeat = page_repeat, |
111 pageset_repeat = pageset_repeat, | 111 pageset_repeat = pageset_repeat, |
112 cold_load_percent = cold_load_percent, | 112 cold_load_percent = cold_load_percent, |
113 report_speed_index = report_speed_index) | 113 report_speed_index = report_speed_index) |
114 options = browser_options.BrowserFinderOptions() | 114 options = browser_options.BrowserFinderOptions() |
115 options.browser_options.platform = FakePlatform() | 115 options.browser_options.platform = FakePlatform() |
116 parser = options.CreateParser() | 116 parser = options.CreateParser() |
117 user_story_runner.AddCommandLineArgs(parser) | 117 story_runner.AddCommandLineArgs(parser) |
118 args = ['--page-repeat=%i' % page_repeat, | 118 args = ['--page-repeat=%i' % page_repeat, |
119 '--pageset-repeat=%i' % pageset_repeat] | 119 '--pageset-repeat=%i' % pageset_repeat] |
120 parser.parse_args(args) | 120 parser.parse_args(args) |
121 user_story_runner.ProcessCommandLineArgs(parser, options) | 121 story_runner.ProcessCommandLineArgs(parser, options) |
122 cycler.CustomizeBrowserOptions(options.browser_options) | 122 cycler.CustomizeBrowserOptions(options.browser_options) |
123 | 123 |
124 if setup_memory_module: | 124 if setup_memory_module: |
125 # Mock out memory metrics; the real ones require a real browser. | 125 # Mock out memory metrics; the real ones require a real browser. |
126 mock_memory_metric = MockMemoryMetric() | 126 mock_memory_metric = MockMemoryMetric() |
127 | 127 |
128 mock_memory_module = simple_mock.MockObject() | 128 mock_memory_module = simple_mock.MockObject() |
129 mock_memory_module.ExpectCall( | 129 mock_memory_module.ExpectCall( |
130 'MemoryMetric').WithArgs(simple_mock.DONT_CARE).WillReturn( | 130 'MemoryMetric').WithArgs(simple_mock.DONT_CARE).WillReturn( |
131 mock_memory_metric) | 131 mock_memory_metric) |
(...skipping 120 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
252 # does an initial navigate to avoid paying for a cross-renderer navigation. | 252 # does an initial navigate to avoid paying for a cross-renderer navigation. |
253 cycler = self.SetUpCycler(setup_memory_module=True) | 253 cycler = self.SetUpCycler(setup_memory_module=True) |
254 pages = [FakePage('file://fakepage1.com'), FakePage('file://fakepage2.com')] | 254 pages = [FakePage('file://fakepage1.com'), FakePage('file://fakepage2.com')] |
255 tab = FakeTab() | 255 tab = FakeTab() |
256 | 256 |
257 self.assertEqual([], tab.navigated_urls) | 257 self.assertEqual([], tab.navigated_urls) |
258 for page in pages * 2: | 258 for page in pages * 2: |
259 cycler.WillNavigateToPage(page, tab) | 259 cycler.WillNavigateToPage(page, tab) |
260 self.assertEqual( | 260 self.assertEqual( |
261 ['http://fakeserver:99999/nonexistent.html'], tab.navigated_urls) | 261 ['http://fakeserver:99999/nonexistent.html'], tab.navigated_urls) |
OLD | NEW |