OLD | NEW |
1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 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 from measurements import task_execution_time | 5 from measurements import task_execution_time |
6 from telemetry.core import wpr_modes | 6 from telemetry.core import wpr_modes |
7 from telemetry.page import page as page_module | 7 from telemetry.page import page as page_module |
8 from telemetry.unittest import options_for_unittests | 8 from telemetry.unittest import options_for_unittests |
9 from telemetry.unittest import page_test_test_case | 9 from telemetry.unittest import page_test_test_case |
10 | 10 |
(...skipping 12 matching lines...) Expand all Loading... |
23 | 23 |
24 class TaskExecutionTimeUnitTest(page_test_test_case.PageTestTestCase): | 24 class TaskExecutionTimeUnitTest(page_test_test_case.PageTestTestCase): |
25 def setUp(self): | 25 def setUp(self): |
26 self._options = options_for_unittests.GetCopy() | 26 self._options = options_for_unittests.GetCopy() |
27 self._options.browser_options.wpr_mode = wpr_modes.WPR_OFF | 27 self._options.browser_options.wpr_mode = wpr_modes.WPR_OFF |
28 | 28 |
29 def testCorrectNumberOfResultsReturned(self): | 29 def testCorrectNumberOfResultsReturned(self): |
30 ps = self.CreateEmptyPageSet() | 30 ps = self.CreateEmptyPageSet() |
31 ps.AddPage(TestTaskExecutionTimePage(ps, ps.base_dir)) | 31 ps.AddPage(TestTaskExecutionTimePage(ps, ps.base_dir)) |
32 measurement = task_execution_time.TaskExecutionTime() | 32 measurement = task_execution_time.TaskExecutionTime() |
| 33 measurement.SetInUnitTest() |
33 | 34 |
34 results = self.RunMeasurement(measurement, ps, options=self._options) | 35 results = self.RunMeasurement(measurement, ps, options=self._options) |
35 | 36 |
36 self.assertEquals( | 37 self.assertEquals( |
37 task_execution_time.TaskExecutionTime.GetExpectedResultCount(), | 38 task_execution_time.TaskExecutionTime.GetExpectedResultCount(), |
38 len(results.all_page_specific_values)) | 39 len(results.all_page_specific_values)) |
39 | 40 |
40 def testResultsAreDecreasing(self): | 41 def testResultsAreDecreasing(self): |
41 ps = self.CreateEmptyPageSet() | 42 ps = self.CreateEmptyPageSet() |
42 ps.AddPage(TestTaskExecutionTimePage(ps, ps.base_dir)) | 43 ps.AddPage(TestTaskExecutionTimePage(ps, ps.base_dir)) |
43 measurement = task_execution_time.TaskExecutionTime() | 44 measurement = task_execution_time.TaskExecutionTime() |
| 45 measurement.SetInUnitTest() |
44 | 46 |
45 results = self.RunMeasurement(measurement, ps, options=self._options) | 47 results = self.RunMeasurement(measurement, ps, options=self._options) |
46 | 48 |
47 for first, second in zip( | 49 for first, second in zip( |
48 results.all_page_specific_values, results.all_page_specific_values[1:]): | 50 results.all_page_specific_values, results.all_page_specific_values[1:]): |
49 self.assertGreaterEqual(first.value, second.value) | 51 self.assertGreaterEqual(first.value, second.value) |
OLD | NEW |