OLD | NEW |
1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 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 json | 5 import json |
6 import os | 6 import os |
7 | 7 |
8 from telemetry import test | 8 from telemetry import test |
| 9 from telemetry.core import exceptions |
9 from telemetry.core import wpr_modes | 10 from telemetry.core import wpr_modes |
10 from telemetry.page import page as page_module | 11 from telemetry.page import page as page_module |
11 from telemetry.page import page_measurement | 12 from telemetry.page import page_measurement |
12 from telemetry.page import page_measurement_unittest_base | 13 from telemetry.page import page_measurement_unittest_base |
13 from telemetry.page import page_set | 14 from telemetry.page import page_set |
14 from telemetry.page import page_set_archive_info | 15 from telemetry.page import page_set_archive_info |
15 from telemetry.unittest import options_for_unittests | 16 from telemetry.unittest import options_for_unittests |
16 | 17 |
17 | 18 |
18 class MeasurementThatFails(page_measurement.PageMeasurement): | 19 class MeasurementThatFails(page_measurement.PageMeasurement): |
19 def MeasurePage(self, page, tab, results): | 20 def MeasurePage(self, page, tab, results): |
20 raise page_measurement.MeasurementFailure('Intentional failure.') | 21 raise exceptions.IntentionalException |
21 | 22 |
22 class MeasurementThatHasDefaults(page_measurement.PageMeasurement): | 23 class MeasurementThatHasDefaults(page_measurement.PageMeasurement): |
23 def AddCommandLineArgs(self, parser): | 24 def AddCommandLineArgs(self, parser): |
24 parser.add_option('-x', dest='x', default=3) | 25 parser.add_option('-x', dest='x', default=3) |
25 | 26 |
26 def MeasurePage(self, page, tab, results): | 27 def MeasurePage(self, page, tab, results): |
27 if not hasattr(self.options, 'x'): | 28 if not hasattr(self.options, 'x'): |
28 raise page_measurement.MeasurementFailure('Default option was not set.') | 29 raise page_measurement.MeasurementFailure('Default option was not set.') |
29 if self.options.x != 3: | 30 if self.options.x != 3: |
30 raise page_measurement.MeasurementFailure( | 31 raise page_measurement.MeasurementFailure( |
(...skipping 120 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
151 if os.path.isfile(test_archive): | 152 if os.path.isfile(test_archive): |
152 os.remove(test_archive) | 153 os.remove(test_archive) |
153 | 154 |
154 def testActions(self): | 155 def testActions(self): |
155 ps = self.CreateEmptyPageSet() | 156 ps = self.CreateEmptyPageSet() |
156 page = PageWithAction('file://blank.html', ps) | 157 page = PageWithAction('file://blank.html', ps) |
157 ps.AddPage(page) | 158 ps.AddPage(page) |
158 measurement = MeasurementWithAction() | 159 measurement = MeasurementWithAction() |
159 self.RunMeasurement(measurement, ps, options=self._options) | 160 self.RunMeasurement(measurement, ps, options=self._options) |
160 self.assertTrue(page.run_test_action_called) | 161 self.assertTrue(page.run_test_action_called) |
OLD | NEW |