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