Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(158)

Side by Side Diff: build/android/pylib/gtest/dispatch.py

Issue 19220004: [Android] Move modules_unittest to isolate. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Addressed all comments Created 7 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « build/android/pylib/constants.py ('k') | build/android/pylib/gtest/test_runner.py » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 # Copyright (c) 2013 The Chromium Authors. All rights reserved. 1 # Copyright (c) 2013 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 """Dispatches GTests.""" 5 """Dispatches GTests."""
6 6
7 import copy 7 import copy
8 import fnmatch 8 import fnmatch
9 import glob 9 import glob
10 import logging 10 import logging
(...skipping 17 matching lines...) Expand all
28 # TODO(frankf): Add more test targets here after making sure we don't 28 # TODO(frankf): Add more test targets here after making sure we don't
29 # blow up the dependency size (and the world). 29 # blow up the dependency size (and the world).
30 _ISOLATE_FILE_PATHS = { 30 _ISOLATE_FILE_PATHS = {
31 'base_unittests': 'base/base_unittests.isolate', 31 'base_unittests': 'base/base_unittests.isolate',
32 'breakpad_unittests': 'breakpad/breakpad_unittests.isolate', 32 'breakpad_unittests': 'breakpad/breakpad_unittests.isolate',
33 'cc_perftests': 'cc/cc_perftests.isolate', 33 'cc_perftests': 'cc/cc_perftests.isolate',
34 'components_unittests': 'components/components_unittests.isolate', 34 'components_unittests': 'components/components_unittests.isolate',
35 'content_browsertests': 'content/content_browsertests.isolate', 35 'content_browsertests': 'content/content_browsertests.isolate',
36 'content_unittests': 'content/content_unittests.isolate', 36 'content_unittests': 'content/content_unittests.isolate',
37 'media_unittests': 'media/media_unittests.isolate', 37 'media_unittests': 'media/media_unittests.isolate',
38 'modules_unittests': 'third_party/webrtc/modules/modules_unittests.isolate',
38 'net_unittests': 'net/net_unittests.isolate', 39 'net_unittests': 'net/net_unittests.isolate',
39 'ui_unittests': 'ui/ui_unittests.isolate', 40 'ui_unittests': 'ui/ui_unittests.isolate',
40 'unit_tests': 'chrome/unit_tests.isolate', 41 'unit_tests': 'chrome/unit_tests.isolate',
41 } 42 }
42 43
43 # Used for filtering large data deps at a finer grain than what's allowed in 44 # Used for filtering large data deps at a finer grain than what's allowed in
44 # isolate files since pushing deps to devices is expensive. 45 # isolate files since pushing deps to devices is expensive.
45 # Wildcards are allowed. 46 # Wildcards are allowed.
46 _DEPS_EXCLUSION_LIST = [ 47 _DEPS_EXCLUSION_LIST = [
47 'chrome/test/data/extensions/api_test', 48 'chrome/test/data/extensions/api_test',
(...skipping 17 matching lines...) Expand all
65 _ISOLATE_SCRIPT = os.path.join( 66 _ISOLATE_SCRIPT = os.path.join(
66 constants.DIR_SOURCE_ROOT, 'tools', 'swarm_client', 'isolate.py') 67 constants.DIR_SOURCE_ROOT, 'tools', 'swarm_client', 'isolate.py')
67 68
68 69
69 def _GenerateDepsDirUsingIsolate(test_suite, build_type): 70 def _GenerateDepsDirUsingIsolate(test_suite, build_type):
70 """Generate the dependency dir for the test suite using isolate. 71 """Generate the dependency dir for the test suite using isolate.
71 72
72 Args: 73 Args:
73 test_suite: The test suite basename (e.g. base_unittests). 74 test_suite: The test suite basename (e.g. base_unittests).
74 build_type: Release/Debug 75 build_type: Release/Debug
75
76 Returns:
77 If an isolate file exists, returns path to dependency dir on the host.
78 Otherwise, returns False.
79 """ 76 """
80 product_dir = os.path.join(cmd_helper.OutDirectory.get(), build_type) 77 product_dir = os.path.join(cmd_helper.OutDirectory.get(), build_type)
81 assert os.path.isabs(product_dir) 78 assert os.path.isabs(product_dir)
79
80 if os.path.isdir(constants.ISOLATE_DEPS_DIR):
81 shutil.rmtree(constants.ISOLATE_DEPS_DIR)
82
82 isolate_rel_path = _ISOLATE_FILE_PATHS.get(test_suite) 83 isolate_rel_path = _ISOLATE_FILE_PATHS.get(test_suite)
83 if not isolate_rel_path: 84 if not isolate_rel_path:
84 return False 85 logging.info('Did not find an isolate file for the test suite.')
86 return
85 87
86 isolate_abs_path = os.path.join(constants.DIR_SOURCE_ROOT, isolate_rel_path) 88 isolate_abs_path = os.path.join(constants.DIR_SOURCE_ROOT, isolate_rel_path)
87 isolated_abs_path = os.path.join( 89 isolated_abs_path = os.path.join(
88 product_dir, '%s.isolated' % test_suite) 90 product_dir, '%s.isolated' % test_suite)
89 assert os.path.exists(isolate_abs_path) 91 assert os.path.exists(isolate_abs_path)
90 deps_dir = os.path.join(product_dir, 'isolate_deps_dir')
91 if os.path.isdir(deps_dir):
92 shutil.rmtree(deps_dir)
93 isolate_cmd = [ 92 isolate_cmd = [
94 'python', _ISOLATE_SCRIPT, 93 'python', _ISOLATE_SCRIPT,
95 'remap', 94 'remap',
96 '--isolate', isolate_abs_path, 95 '--isolate', isolate_abs_path,
97 '--isolated', isolated_abs_path, 96 '--isolated', isolated_abs_path,
98 '-V', 'PRODUCT_DIR=%s' % product_dir, 97 '-V', 'PRODUCT_DIR=%s' % product_dir,
99 '-V', 'OS=android', 98 '-V', 'OS=android',
100 '--outdir', deps_dir, 99 '--outdir', constants.ISOLATE_DEPS_DIR,
101 ] 100 ]
102 assert not cmd_helper.RunCmd(isolate_cmd) 101 assert not cmd_helper.RunCmd(isolate_cmd)
103 102
104 # We're relying on the fact that timestamps are preserved 103 # We're relying on the fact that timestamps are preserved
105 # by the remap command (hardlinked). Otherwise, all the data 104 # by the remap command (hardlinked). Otherwise, all the data
106 # will be pushed to the device once we move to using time diff 105 # will be pushed to the device once we move to using time diff
107 # instead of md5sum. Perform a sanity check here. 106 # instead of md5sum. Perform a sanity check here.
108 for root, _, filenames in os.walk(deps_dir): 107 for root, _, filenames in os.walk(constants.ISOLATE_DEPS_DIR):
109 if filenames: 108 if filenames:
110 linked_file = os.path.join(root, filenames[0]) 109 linked_file = os.path.join(root, filenames[0])
111 orig_file = os.path.join( 110 orig_file = os.path.join(
112 constants.DIR_SOURCE_ROOT, 111 constants.DIR_SOURCE_ROOT,
113 os.path.relpath(linked_file, deps_dir)) 112 os.path.relpath(linked_file, constants.ISOLATE_DEPS_DIR))
114 if os.stat(linked_file).st_ino == os.stat(orig_file).st_ino: 113 if os.stat(linked_file).st_ino == os.stat(orig_file).st_ino:
115 break 114 break
116 else: 115 else:
117 raise Exception('isolate remap command did not use hardlinks.') 116 raise Exception('isolate remap command did not use hardlinks.')
118 117
119 # Delete excluded files as defined by _DEPS_EXCLUSION_LIST. 118 # Delete excluded files as defined by _DEPS_EXCLUSION_LIST.
120 old_cwd = os.getcwd() 119 old_cwd = os.getcwd()
121 try: 120 try:
122 os.chdir(deps_dir) 121 os.chdir(constants.ISOLATE_DEPS_DIR)
123 excluded_paths = [x for y in _DEPS_EXCLUSION_LIST for x in glob.glob(y)] 122 excluded_paths = [x for y in _DEPS_EXCLUSION_LIST for x in glob.glob(y)]
124 if excluded_paths: 123 if excluded_paths:
125 logging.info('Excluding the following from dependency list: %s', 124 logging.info('Excluding the following from dependency list: %s',
126 excluded_paths) 125 excluded_paths)
127 for p in excluded_paths: 126 for p in excluded_paths:
128 if os.path.isdir(p): 127 if os.path.isdir(p):
129 shutil.rmtree(p) 128 shutil.rmtree(p)
130 else: 129 else:
131 os.remove(p) 130 os.remove(p)
132 finally: 131 finally:
133 os.chdir(old_cwd) 132 os.chdir(old_cwd)
134 133
135 # On Android, all pak files need to be in the top-level 'paks' directory. 134 # On Android, all pak files need to be in the top-level 'paks' directory.
136 paks_dir = os.path.join(deps_dir, 'paks') 135 paks_dir = os.path.join(constants.ISOLATE_DEPS_DIR, 'paks')
137 os.mkdir(paks_dir) 136 os.mkdir(paks_dir)
138 for root, _, filenames in os.walk(os.path.join(deps_dir, 'out')): 137 for root, _, filenames in os.walk(os.path.join(constants.ISOLATE_DEPS_DIR,
138 'out')):
139 for filename in fnmatch.filter(filenames, '*.pak'): 139 for filename in fnmatch.filter(filenames, '*.pak'):
140 shutil.move(os.path.join(root, filename), paks_dir) 140 shutil.move(os.path.join(root, filename), paks_dir)
141 141
142 # Move everything in PRODUCT_DIR to top level. 142 # Move everything in PRODUCT_DIR to top level.
143 deps_product_dir = os.path.join(deps_dir, 'out', build_type) 143 deps_product_dir = os.path.join(constants.ISOLATE_DEPS_DIR, 'out', build_type)
144 if os.path.isdir(deps_product_dir): 144 if os.path.isdir(deps_product_dir):
145 for p in os.listdir(deps_product_dir): 145 for p in os.listdir(deps_product_dir):
146 shutil.move(os.path.join(deps_product_dir, p), deps_dir) 146 shutil.move(os.path.join(deps_product_dir, p), constants.ISOLATE_DEPS_DIR)
147 os.rmdir(deps_product_dir) 147 os.rmdir(deps_product_dir)
148 os.rmdir(os.path.join(deps_dir, 'out')) 148 os.rmdir(os.path.join(constants.ISOLATE_DEPS_DIR, 'out'))
149
150 return deps_dir
151 149
152 150
153 def _FullyQualifiedTestSuites(exe, option_test_suite, build_type): 151 def _FullyQualifiedTestSuites(exe, option_test_suite, build_type):
154 """Get a list of absolute paths to test suite targets. 152 """Get a list of absolute paths to test suite targets.
155 153
156 Args: 154 Args:
157 exe: if True, use the executable-based test runner. 155 exe: if True, use the executable-based test runner.
158 option_test_suite: the test_suite specified as an option. 156 option_test_suite: the test_suite specified as an option.
159 build_type: 'Release' or 'Debug'. 157 build_type: 'Release' or 'Debug'.
160 158
(...skipping 109 matching lines...) Expand 10 before | Expand all | Expand 10 after
270 attached_devices = android_commands.GetAttachedDevices() 268 attached_devices = android_commands.GetAttachedDevices()
271 269
272 if not attached_devices: 270 if not attached_devices:
273 raise Exception('A device must be attached and online.') 271 raise Exception('A device must be attached and online.')
274 272
275 # Reset the test port allocation. It's important to do it before starting 273 # Reset the test port allocation. It's important to do it before starting
276 # to dispatch any tests. 274 # to dispatch any tests.
277 if not ports.ResetTestServerPortAllocation(): 275 if not ports.ResetTestServerPortAllocation():
278 raise Exception('Failed to reset test server port.') 276 raise Exception('Failed to reset test server port.')
279 277
280 deps_dir = _GenerateDepsDirUsingIsolate(suite_name, options.build_type) 278 _GenerateDepsDirUsingIsolate(suite_name, options.build_type)
281 279
282 # Constructs a new TestRunner with the current options. 280 # Constructs a new TestRunner with the current options.
283 def RunnerFactory(device, shard_index): 281 def RunnerFactory(device, shard_index):
284 return test_runner.TestRunner( 282 return test_runner.TestRunner(
285 device, 283 device,
286 options.test_suite, 284 options.test_suite,
287 options.test_arguments, 285 options.test_arguments,
288 options.timeout, 286 options.timeout,
289 options.cleanup_test_files, 287 options.cleanup_test_files,
290 options.tool, 288 options.tool,
291 options.build_type, 289 options.build_type,
292 options.webkit, 290 options.webkit,
293 options.push_deps, 291 options.push_deps,
294 constants.GTEST_TEST_PACKAGE_NAME, 292 constants.GTEST_TEST_PACKAGE_NAME,
295 constants.GTEST_TEST_ACTIVITY_NAME, 293 constants.GTEST_TEST_ACTIVITY_NAME,
296 constants.GTEST_COMMAND_LINE_FILE, 294 constants.GTEST_COMMAND_LINE_FILE)
297 deps_dir=deps_dir)
298 295
299 # Get tests and split them up based on the number of devices. 296 # Get tests and split them up based on the number of devices.
300 if options.test_filter: 297 if options.test_filter:
301 all_tests = [t for t in options.test_filter.split(':') if t] 298 all_tests = [t for t in options.test_filter.split(':') if t]
302 else: 299 else:
303 all_tests = GetAllEnabledTests(RunnerFactory, attached_devices) 300 all_tests = GetAllEnabledTests(RunnerFactory, attached_devices)
304 num_devices = len(attached_devices) 301 num_devices = len(attached_devices)
305 tests = [':'.join(all_tests[i::num_devices]) for i in xrange(num_devices)] 302 tests = [':'.join(all_tests[i::num_devices]) for i in xrange(num_devices)]
306 tests = [t for t in tests if t] 303 tests = [t for t in tests if t]
307 304
308 # Run tests. 305 # Run tests.
309 test_results, exit_code = shard.ShardAndRunTests( 306 test_results, exit_code = shard.ShardAndRunTests(
310 RunnerFactory, attached_devices, tests, options.build_type, 307 RunnerFactory, attached_devices, tests, options.build_type,
311 test_timeout=None, num_retries=options.num_retries) 308 test_timeout=None, num_retries=options.num_retries)
312 309
313 report_results.LogFull( 310 report_results.LogFull(
314 results=test_results, 311 results=test_results,
315 test_type='Unit test', 312 test_type='Unit test',
316 test_package=suite_name, 313 test_package=suite_name,
317 build_type=options.build_type, 314 build_type=options.build_type,
318 flakiness_server=options.flakiness_dashboard_server) 315 flakiness_server=options.flakiness_dashboard_server)
319 316
317 if os.path.isdir(constants.ISOLATE_DEPS_DIR):
318 shutil.rmtree(constants.ISOLATE_DEPS_DIR)
319
320 for buildbot_emulator in buildbot_emulators: 320 for buildbot_emulator in buildbot_emulators:
321 buildbot_emulator.Shutdown() 321 buildbot_emulator.Shutdown()
322 322
323 return (test_results, exit_code) 323 return (test_results, exit_code)
324 324
325 325
326 def _ListTestSuites(): 326 def _ListTestSuites():
327 """Display a list of available test suites.""" 327 """Display a list of available test suites."""
328 print 'Available test suites are:' 328 print 'Available test suites are:'
329 for test_suite in gtest_config.STABLE_TEST_SUITES: 329 for test_suite in gtest_config.STABLE_TEST_SUITES:
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after
361 test_options.test_suite = suite_path 361 test_options.test_suite = suite_path
362 test_results, test_exit_code = _RunATestSuite(test_options, suite_name) 362 test_results, test_exit_code = _RunATestSuite(test_options, suite_name)
363 results.AddTestRunResults(test_results) 363 results.AddTestRunResults(test_results)
364 if test_exit_code and exit_code != constants.ERROR_EXIT_CODE: 364 if test_exit_code and exit_code != constants.ERROR_EXIT_CODE:
365 exit_code = test_exit_code 365 exit_code = test_exit_code
366 366
367 if options.use_xvfb: 367 if options.use_xvfb:
368 framebuffer.Stop() 368 framebuffer.Stop()
369 369
370 return (results, exit_code) 370 return (results, exit_code)
OLDNEW
« no previous file with comments | « build/android/pylib/constants.py ('k') | build/android/pylib/gtest/test_runner.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698