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 import fnmatch | 5 import fnmatch |
6 import imp | 6 import imp |
7 import logging | 7 import logging |
8 import modulefinder | 8 import modulefinder |
9 import optparse | 9 import optparse |
10 import os | 10 import os |
11 import sys | 11 import sys |
12 import zipfile | 12 import zipfile |
13 | 13 |
14 from catapult_base import cloud_storage | 14 from catapult_base import cloud_storage |
15 from telemetry import benchmark | 15 from telemetry import benchmark |
| 16 from telemetry.core import discover |
16 from telemetry.internal.util import bootstrap | 17 from telemetry.internal.util import bootstrap |
17 from telemetry.internal.util import command_line | 18 from telemetry.internal.util import command_line |
18 from telemetry.internal.util import path | 19 from telemetry.internal.util import path |
19 from telemetry.internal.util import path_set | 20 from telemetry.internal.util import path_set |
20 from telemetry.util import classes_util | |
21 | |
22 | 21 |
23 DEPS_FILE = 'bootstrap_deps' | 22 DEPS_FILE = 'bootstrap_deps' |
24 | 23 |
25 | 24 |
26 def FindBootstrapDependencies(base_dir): | 25 def FindBootstrapDependencies(base_dir): |
27 deps_file = os.path.join(base_dir, DEPS_FILE) | 26 deps_file = os.path.join(base_dir, DEPS_FILE) |
28 if not os.path.exists(deps_file): | 27 if not os.path.exists(deps_file): |
29 return [] | 28 return [] |
30 deps_paths = bootstrap.ListAllDepsPaths(deps_file) | 29 deps_paths = bootstrap.ListAllDepsPaths(deps_file) |
31 return set(os.path.realpath(os.path.join( | 30 return set(os.path.realpath(os.path.join( |
(...skipping 24 matching lines...) Expand all Loading... |
56 continue | 55 continue |
57 | 56 |
58 yield module_path | 57 yield module_path |
59 | 58 |
60 | 59 |
61 def FindPageSetDependencies(base_dir): | 60 def FindPageSetDependencies(base_dir): |
62 logging.info('Finding page sets in %s' % base_dir) | 61 logging.info('Finding page sets in %s' % base_dir) |
63 | 62 |
64 # Add base_dir to path so our imports relative to base_dir will work. | 63 # Add base_dir to path so our imports relative to base_dir will work. |
65 sys.path.append(base_dir) | 64 sys.path.append(base_dir) |
66 tests = classes_util.DiscoverClasses(base_dir, base_dir, benchmark.Benchmark) | 65 tests = discover.DiscoverClasses(base_dir, base_dir, benchmark.Benchmark, |
| 66 index_by_class_name=True) |
67 | 67 |
68 for test_class in tests: | 68 for test_class in tests.itervalues(): |
69 test_obj = test_class() | 69 test_obj = test_class() |
70 | 70 |
71 # Ensure the test's default options are set if needed. | 71 # Ensure the test's default options are set if needed. |
72 parser = optparse.OptionParser() | 72 parser = optparse.OptionParser() |
73 test_obj.AddCommandLineArgs(parser, None) | 73 test_obj.AddCommandLineArgs(parser, None) |
74 options = optparse.Values() | 74 options = optparse.Values() |
75 for k, v in parser.get_default_values().__dict__.iteritems(): | 75 for k, v in parser.get_default_values().__dict__.iteritems(): |
76 options.ensure_value(k, v) | 76 options.ensure_value(k, v) |
77 | 77 |
78 # Page set paths are relative to their runner script, not relative to us. | 78 # Page set paths are relative to their runner script, not relative to us. |
(...skipping 164 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
243 | 243 |
244 def Run(self, args): | 244 def Run(self, args): |
245 target_paths = args.positional_args | 245 target_paths = args.positional_args |
246 dependencies = FindDependencies(target_paths, args) | 246 dependencies = FindDependencies(target_paths, args) |
247 if args.zip: | 247 if args.zip: |
248 ZipDependencies(target_paths, dependencies, args) | 248 ZipDependencies(target_paths, dependencies, args) |
249 print 'Zip archive written to %s.' % args.zip | 249 print 'Zip archive written to %s.' % args.zip |
250 else: | 250 else: |
251 print '\n'.join(sorted(dependencies)) | 251 print '\n'.join(sorted(dependencies)) |
252 return 0 | 252 return 0 |
OLD | NEW |