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