OLD | NEW |
1 # Copyright 2016 The Chromium Authors. All rights reserved. | 1 # Copyright 2016 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 """Run all system health stories used by system health benchmarks. | 5 """Run all system health stories used by system health benchmarks. |
6 | 6 |
7 Only memory benchmarks are used when running these stories to make the total | 7 Only memory benchmarks are used when running these stories to make the total |
8 cycle time manageable. Other system health benchmarks should be using the same | 8 cycle time manageable. Other system health benchmarks should be using the same |
9 stories as memory ones, only with fewer actions (no memory dumping). | 9 stories as memory ones, only with fewer actions (no memory dumping). |
10 """ | 10 """ |
11 | 11 |
12 import unittest | 12 import unittest |
13 | 13 |
14 from core import perf_benchmark | 14 from core import perf_benchmark |
15 | 15 |
16 from telemetry import benchmark as benchmark_module | 16 from telemetry import benchmark as benchmark_module |
17 from telemetry import decorators | 17 from telemetry import decorators |
18 from telemetry.core import discover | |
19 from telemetry.internal.browser import browser_finder | 18 from telemetry.internal.browser import browser_finder |
20 from telemetry.testing import options_for_unittests | 19 from telemetry.testing import options_for_unittests |
21 from telemetry.testing import progress_reporter | 20 from telemetry.testing import progress_reporter |
22 | 21 |
| 22 from py_utils import discover |
| 23 |
23 from benchmarks import system_health | 24 from benchmarks import system_health |
24 | 25 |
25 | 26 |
26 def GetSystemHealthBenchmarksToSmokeTest(): | 27 def GetSystemHealthBenchmarksToSmokeTest(): |
27 sh_benchmark_classes = discover.DiscoverClassesInModule( | 28 sh_benchmark_classes = discover.DiscoverClassesInModule( |
28 system_health, perf_benchmark.PerfBenchmark, | 29 system_health, perf_benchmark.PerfBenchmark, |
29 index_by_class_name=True).values() | 30 index_by_class_name=True).values() |
30 return list(b for b in sh_benchmark_classes if | 31 return list(b for b in sh_benchmark_classes if |
31 b.Name().startswith('system_health.memory')) | 32 b.Name().startswith('system_health.memory')) |
32 | 33 |
(...skipping 138 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
171 # Prefetch WPR archive needed by the stories set to avoid race condition | 172 # Prefetch WPR archive needed by the stories set to avoid race condition |
172 # when feching them when tests are run in parallel. | 173 # when feching them when tests are run in parallel. |
173 # See crbug.com/700426 for more details. | 174 # See crbug.com/700426 for more details. |
174 stories_set.wpr_archive_info.DownloadArchivesIfNeeded() | 175 stories_set.wpr_archive_info.DownloadArchivesIfNeeded() |
175 | 176 |
176 for story_to_smoke_test in stories_set.stories: | 177 for story_to_smoke_test in stories_set.stories: |
177 suite.addTest( | 178 suite.addTest( |
178 _GenerateSmokeTestCase(benchmark_class, story_to_smoke_test)) | 179 _GenerateSmokeTestCase(benchmark_class, story_to_smoke_test)) |
179 | 180 |
180 return suite | 181 return suite |
OLD | NEW |