| 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 import sys | 5 import sys |
| 6 import unittest | 6 import unittest |
| 7 | 7 |
| 8 from benchmarks import system_health as system_health_benchmark | 8 from benchmarks import system_health as system_health_benchmark |
| 9 from core import path_util | 9 from core import path_util |
| 10 from page_sets.system_health import system_health_stories | 10 from page_sets.system_health import system_health_stories |
| 11 from page_sets.system_health import system_health_story | 11 from page_sets.system_health import system_health_story |
| 12 | 12 |
| 13 from telemetry import benchmark as benchmark_module | 13 from telemetry import benchmark as benchmark_module |
| 14 from telemetry.core import discover | 14 |
| 15 from py_utils import discover |
| 15 | 16 |
| 16 | 17 |
| 17 def _GetAllSystemHealthBenchmarks(): | 18 def _GetAllSystemHealthBenchmarks(): |
| 18 all_perf_benchmarks = discover.DiscoverClasses( | 19 all_perf_benchmarks = discover.DiscoverClasses( |
| 19 path_util.GetPerfBenchmarksDir(), path_util.GetPerfDir(), | 20 path_util.GetPerfBenchmarksDir(), path_util.GetPerfDir(), |
| 20 benchmark_module.Benchmark, | 21 benchmark_module.Benchmark, |
| 21 index_by_class_name=True).values() | 22 index_by_class_name=True).values() |
| 22 return [b for b in all_perf_benchmarks if | 23 return [b for b in all_perf_benchmarks if |
| 23 sys.modules[b.__module__] == system_health_benchmark] | 24 sys.modules[b.__module__] == system_health_benchmark] |
| 24 | 25 |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 56 system_health_stories.MobileSystemHealthStorySet().stories) | 57 system_health_stories.MobileSystemHealthStorySet().stories) |
| 57 for s in desktop_stories + mobile_stories: | 58 for s in desktop_stories + mobile_stories: |
| 58 # Long running stories has their own way of collecting memory dumps, | 59 # Long running stories has their own way of collecting memory dumps, |
| 59 # so they explicitly override RunPageInteractions method. | 60 # so they explicitly override RunPageInteractions method. |
| 60 if s.name.startswith('long_running:'): | 61 if s.name.startswith('long_running:'): |
| 61 continue | 62 continue |
| 62 self.assertEquals(s.__class__.RunPageInteractions, | 63 self.assertEquals(s.__class__.RunPageInteractions, |
| 63 system_health_story.SystemHealthStory.RunPageInteractions, | 64 system_health_story.SystemHealthStory.RunPageInteractions, |
| 64 'Story %s overrides RunPageInteractions. Override _DidLoadDocument ' | 65 'Story %s overrides RunPageInteractions. Override _DidLoadDocument ' |
| 65 'instead' % s.name) | 66 'instead' % s.name) |
| OLD | NEW |