| 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 |