1#!/usr/bin/env python 2# Copyright (c) 2012 The Chromium Authors. All rights reserved. 3# Use of this source code is governed by a BSD-style license that can be 4# found in the LICENSE file. 5 6''' Runs various chrome tests through valgrind_test.py.''' 7 8import glob 9import logging 10import multiprocessing 11import optparse 12import os 13import stat 14import sys 15 16import logging_utils 17import path_utils 18 19import common 20import valgrind_test 21 22class TestNotFound(Exception): pass 23 24class MultipleGTestFiltersSpecified(Exception): pass 25 26class BuildDirNotFound(Exception): pass 27 28class BuildDirAmbiguous(Exception): pass 29 30class ChromeTests: 31 SLOW_TOOLS = ["memcheck", "tsan", "tsan_rv", "drmemory"] 32 LAYOUT_TESTS_DEFAULT_CHUNK_SIZE = 300 33 34 def __init__(self, options, args, test): 35 if ':' in test: 36 (self._test, self._gtest_filter) = test.split(':', 1) 37 else: 38 self._test = test 39 self._gtest_filter = options.gtest_filter 40 41 if self._test not in self._test_list: 42 raise TestNotFound("Unknown test: %s" % test) 43 44 if options.gtest_filter and options.gtest_filter != self._gtest_filter: 45 raise MultipleGTestFiltersSpecified("Can not specify both --gtest_filter " 46 "and --test %s" % test) 47 48 self._options = options 49 self._args = args 50 51 script_dir = path_utils.ScriptDir() 52 # Compute the top of the tree (the "source dir") from the script dir (where 53 # this script lives). We assume that the script dir is in tools/valgrind/ 54 # relative to the top of the tree. 55 self._source_dir = os.path.dirname(os.path.dirname(script_dir)) 56 # since this path is used for string matching, make sure it's always 57 # an absolute Unix-style path 58 self._source_dir = os.path.abspath(self._source_dir).replace('\\', '/') 59 valgrind_test_script = os.path.join(script_dir, "valgrind_test.py") 60 self._command_preamble = ["--source-dir=%s" % (self._source_dir)] 61 62 if not self._options.build_dir: 63 dirs = [ 64 os.path.join(self._source_dir, "xcodebuild", "Debug"), 65 os.path.join(self._source_dir, "out", "Debug"), 66 os.path.join(self._source_dir, "build", "Debug"), 67 ] 68 build_dir = [d for d in dirs if os.path.isdir(d)] 69 if len(build_dir) > 1: 70 raise BuildDirAmbiguous("Found more than one suitable build dir:\n" 71 "%s\nPlease specify just one " 72 "using --build-dir" % ", ".join(build_dir)) 73 elif build_dir: 74 self._options.build_dir = build_dir[0] 75 else: 76 self._options.build_dir = None 77 78 if self._options.build_dir: 79 build_dir = os.path.abspath(self._options.build_dir) 80 self._command_preamble += ["--build-dir=%s" % (self._options.build_dir)] 81 82 def _EnsureBuildDirFound(self): 83 if not self._options.build_dir: 84 raise BuildDirNotFound("Oops, couldn't find a build dir, please " 85 "specify it manually using --build-dir") 86 87 def _DefaultCommand(self, tool, exe=None, valgrind_test_args=None): 88 '''Generates the default command array that most tests will use.''' 89 if exe and common.IsWindows(): 90 exe += '.exe' 91 92 cmd = list(self._command_preamble) 93 94 # Find all suppressions matching the following pattern: 95 # tools/valgrind/TOOL/suppressions[_PLATFORM].txt 96 # and list them with --suppressions= prefix. 97 script_dir = path_utils.ScriptDir() 98 tool_name = tool.ToolName(); 99 suppression_file = os.path.join(script_dir, tool_name, "suppressions.txt") 100 if os.path.exists(suppression_file): 101 cmd.append("--suppressions=%s" % suppression_file) 102 # Platform-specific suppression 103 for platform in common.PlatformNames(): 104 platform_suppression_file = \ 105 os.path.join(script_dir, tool_name, 'suppressions_%s.txt' % platform) 106 if os.path.exists(platform_suppression_file): 107 cmd.append("--suppressions=%s" % platform_suppression_file) 108 109 if self._options.valgrind_tool_flags: 110 cmd += self._options.valgrind_tool_flags.split(" ") 111 if self._options.keep_logs: 112 cmd += ["--keep_logs"] 113 if valgrind_test_args != None: 114 for arg in valgrind_test_args: 115 cmd.append(arg) 116 if exe: 117 self._EnsureBuildDirFound() 118 cmd.append(os.path.join(self._options.build_dir, exe)) 119 # Valgrind runs tests slowly, so slow tests hurt more; show elapased time 120 # so we can find the slowpokes. 121 cmd.append("--gtest_print_time") 122 # Built-in test launcher for gtest-based executables runs tests using 123 # multiple process by default. Force the single-process mode back. 124 cmd.append("--single-process-tests") 125 if self._options.gtest_repeat: 126 cmd.append("--gtest_repeat=%s" % self._options.gtest_repeat) 127 if self._options.gtest_shuffle: 128 cmd.append("--gtest_shuffle") 129 if self._options.brave_new_test_launcher: 130 cmd.append("--brave-new-test-launcher") 131 if self._options.test_launcher_bot_mode: 132 cmd.append("--test-launcher-bot-mode") 133 return cmd 134 135 def Run(self): 136 ''' Runs the test specified by command-line argument --test ''' 137 logging.info("running test %s" % (self._test)) 138 return self._test_list[self._test](self) 139 140 def _AppendGtestFilter(self, tool, name, cmd): 141 '''Append an appropriate --gtest_filter flag to the googletest binary 142 invocation. 143 If the user passed his own filter mentioning only one test, just use it. 144 Othewise, filter out tests listed in the appropriate gtest_exclude files. 145 ''' 146 if (self._gtest_filter and 147 ":" not in self._gtest_filter and 148 "?" not in self._gtest_filter and 149 "*" not in self._gtest_filter): 150 cmd.append("--gtest_filter=%s" % self._gtest_filter) 151 return 152 153 filters = [] 154 gtest_files_dir = os.path.join(path_utils.ScriptDir(), "gtest_exclude") 155 156 gtest_filter_files = [ 157 os.path.join(gtest_files_dir, name + ".gtest-%s.txt" % tool.ToolName())] 158 # Use ".gtest.txt" files only for slow tools, as they now contain 159 # Valgrind- and Dr.Memory-specific filters. 160 # TODO(glider): rename the files to ".gtest_slow.txt" 161 if tool.ToolName() in ChromeTests.SLOW_TOOLS: 162 gtest_filter_files += [os.path.join(gtest_files_dir, name + ".gtest.txt")] 163 for platform_suffix in common.PlatformNames(): 164 gtest_filter_files += [ 165 os.path.join(gtest_files_dir, name + ".gtest_%s.txt" % platform_suffix), 166 os.path.join(gtest_files_dir, name + ".gtest-%s_%s.txt" % \ 167 (tool.ToolName(), platform_suffix))] 168 logging.info("Reading gtest exclude filter files:") 169 for filename in gtest_filter_files: 170 # strip the leading absolute path (may be very long on the bot) 171 # and the following / or \. 172 readable_filename = filename.replace("\\", "/") # '\' on Windows 173 readable_filename = readable_filename.replace(self._source_dir, "")[1:] 174 if not os.path.exists(filename): 175 logging.info(" \"%s\" - not found" % readable_filename) 176 continue 177 logging.info(" \"%s\" - OK" % readable_filename) 178 f = open(filename, 'r') 179 for line in f.readlines(): 180 if line.startswith("#") or line.startswith("//") or line.isspace(): 181 continue 182 line = line.rstrip() 183 test_prefixes = ["FLAKY", "FAILS"] 184 for p in test_prefixes: 185 # Strip prefixes from the test names. 186 line = line.replace(".%s_" % p, ".") 187 # Exclude the original test name. 188 filters.append(line) 189 if line[-2:] != ".*": 190 # List all possible prefixes if line doesn't end with ".*". 191 for p in test_prefixes: 192 filters.append(line.replace(".", ".%s_" % p)) 193 # Get rid of duplicates. 194 filters = set(filters) 195 gtest_filter = self._gtest_filter 196 if len(filters): 197 if gtest_filter: 198 gtest_filter += ":" 199 if gtest_filter.find("-") < 0: 200 gtest_filter += "-" 201 else: 202 gtest_filter = "-" 203 gtest_filter += ":".join(filters) 204 if gtest_filter: 205 cmd.append("--gtest_filter=%s" % gtest_filter) 206 207 @staticmethod 208 def ShowTests(): 209 test_to_names = {} 210 for name, test_function in ChromeTests._test_list.iteritems(): 211 test_to_names.setdefault(test_function, []).append(name) 212 213 name_to_aliases = {} 214 for names in test_to_names.itervalues(): 215 names.sort(key=lambda name: len(name)) 216 name_to_aliases[names[0]] = names[1:] 217 218 print 219 print "Available tests:" 220 print "----------------" 221 for name, aliases in sorted(name_to_aliases.iteritems()): 222 if aliases: 223 print " {} (aka {})".format(name, ', '.join(aliases)) 224 else: 225 print " {}".format(name) 226 227 def SetupLdPath(self, requires_build_dir): 228 if requires_build_dir: 229 self._EnsureBuildDirFound() 230 elif not self._options.build_dir: 231 return 232 233 # Append build_dir to LD_LIBRARY_PATH so external libraries can be loaded. 234 if (os.getenv("LD_LIBRARY_PATH")): 235 os.putenv("LD_LIBRARY_PATH", "%s:%s" % (os.getenv("LD_LIBRARY_PATH"), 236 self._options.build_dir)) 237 else: 238 os.putenv("LD_LIBRARY_PATH", self._options.build_dir) 239 240 def SimpleTest(self, module, name, valgrind_test_args=None, cmd_args=None): 241 tool = valgrind_test.CreateTool(self._options.valgrind_tool) 242 cmd = self._DefaultCommand(tool, name, valgrind_test_args) 243 self._AppendGtestFilter(tool, name, cmd) 244 cmd.extend(['--test-tiny-timeout=1000']) 245 if cmd_args: 246 cmd.extend(cmd_args) 247 248 self.SetupLdPath(True) 249 return tool.Run(cmd, module) 250 251 def RunCmdLine(self): 252 tool = valgrind_test.CreateTool(self._options.valgrind_tool) 253 cmd = self._DefaultCommand(tool, None, self._args) 254 self.SetupLdPath(False) 255 return tool.Run(cmd, None) 256 257 def TestAccessibility(self): 258 return self.SimpleTest("accessibility", "accessibility_unittests") 259 260 def TestAddressInput(self): 261 return self.SimpleTest("addressinput", "libaddressinput_unittests") 262 263 def TestAngle(self): 264 return self.SimpleTest("angle", "angle_unittests") 265 266 def TestAppList(self): 267 return self.SimpleTest("app_list", "app_list_unittests") 268 269 def TestAsh(self): 270 return self.SimpleTest("ash", "ash_unittests") 271 272 def TestAshShell(self): 273 return self.SimpleTest("ash_shelf", "ash_shell_unittests") 274 275 def TestAura(self): 276 return self.SimpleTest("aura", "aura_unittests") 277 278 def TestBase(self): 279 return self.SimpleTest("base", "base_unittests") 280 281 def TestBlinkHeap(self): 282 return self.SimpleTest("blink_heap", "blink_heap_unittests") 283 284 def TestBlinkPlatform(self): 285 return self.SimpleTest("blink_platform", "blink_platform_unittests") 286 287 def TestCacheInvalidation(self): 288 return self.SimpleTest("cacheinvalidation", "cacheinvalidation_unittests") 289 290 def TestCast(self): 291 return self.SimpleTest("chrome", "cast_unittests") 292 293 def TestCC(self): 294 return self.SimpleTest("cc", "cc_unittests") 295 296 def TestChromeApp(self): 297 return self.SimpleTest("chrome_app", "chrome_app_unittests") 298 299 def TestChromeElf(self): 300 return self.SimpleTest("chrome_elf", "chrome_elf_unittests") 301 302 def TestChromeDriver(self): 303 return self.SimpleTest("chromedriver", "chromedriver_unittests") 304 305 def TestChromeOS(self): 306 return self.SimpleTest("chromeos", "chromeos_unittests") 307 308 def TestCloudPrint(self): 309 return self.SimpleTest("cloud_print", "cloud_print_unittests") 310 311 def TestComponents(self): 312 return self.SimpleTest("components", "components_unittests") 313 314 def TestCompositor(self): 315 return self.SimpleTest("compositor", "compositor_unittests") 316 317 def TestContent(self): 318 return self.SimpleTest("content", "content_unittests") 319 320 def TestCourgette(self): 321 return self.SimpleTest("courgette", "courgette_unittests") 322 323 def TestCrypto(self): 324 return self.SimpleTest("crypto", "crypto_unittests") 325 326 def TestDevice(self): 327 return self.SimpleTest("device", "device_unittests") 328 329 def TestDisplay(self): 330 return self.SimpleTest("display", "display_unittests") 331 332 def TestEvents(self): 333 return self.SimpleTest("events", "events_unittests") 334 335 def TestExtensions(self): 336 return self.SimpleTest("extensions", "extensions_unittests") 337 338 def TestFFmpeg(self): 339 return self.SimpleTest("chrome", "ffmpeg_unittests") 340 341 def TestFFmpegRegressions(self): 342 return self.SimpleTest("chrome", "ffmpeg_regression_tests") 343 344 def TestGCM(self): 345 return self.SimpleTest("gcm", "gcm_unit_tests") 346 347 def TestGfx(self): 348 return self.SimpleTest("gfx", "gfx_unittests") 349 350 def TestGin(self): 351 return self.SimpleTest("gin", "gin_unittests") 352 353 def TestGoogleApis(self): 354 return self.SimpleTest("google_apis", "google_apis_unittests") 355 356 def TestGPU(self): 357 return self.SimpleTest("gpu", "gpu_unittests") 358 359 def TestIpc(self): 360 return self.SimpleTest("ipc", "ipc_tests", 361 valgrind_test_args=["--trace_children"]) 362 363 def TestInstallerUtil(self): 364 return self.SimpleTest("installer_util", "installer_util_unittests") 365 366 def TestJingle(self): 367 return self.SimpleTest("chrome", "jingle_unittests") 368 369 def TestKeyboard(self): 370 return self.SimpleTest("keyboard", "keyboard_unittests") 371 372 def TestMedia(self): 373 return self.SimpleTest("chrome", "media_unittests") 374 375 def TestMessageCenter(self): 376 return self.SimpleTest("message_center", "message_center_unittests") 377 378 def TestMojoAppsJS(self): 379 return self.SimpleTest("mojo_apps_js", "mojo_apps_js_unittests") 380 381 def TestMojoCommon(self): 382 return self.SimpleTest("mojo_common", "mojo_common_unittests") 383 384 def TestMojoJS(self): 385 return self.SimpleTest("mojo_js", "mojo_js_unittests") 386 387 def TestMojoPublicBindings(self): 388 return self.SimpleTest("mojo_public_bindings", 389 "mojo_public_bindings_unittests") 390 391 def TestMojoPublicEnv(self): 392 return self.SimpleTest("mojo_public_env", 393 "mojo_public_environment_unittests") 394 395 def TestMojoPublicSystem(self): 396 return self.SimpleTest("mojo_public_system", 397 "mojo_public_system_unittests") 398 399 def TestMojoPublicSysPerf(self): 400 return self.SimpleTest("mojo_public_sysperf", 401 "mojo_public_system_perftests") 402 403 def TestMojoPublicUtility(self): 404 return self.SimpleTest("mojo_public_utility", 405 "mojo_public_utility_unittests") 406 407 def TestMojoApplicationManager(self): 408 return self.SimpleTest("mojo_application_manager", 409 "mojo_application_manager_unittests") 410 411 def TestMojoSystem(self): 412 return self.SimpleTest("mojo_system", "mojo_system_unittests") 413 414 def TestMojoViewManager(self): 415 return self.SimpleTest("mojo_view_manager", "mojo_view_manager_unittests") 416 417 def TestNet(self): 418 return self.SimpleTest("net", "net_unittests") 419 420 def TestNetPerf(self): 421 return self.SimpleTest("net", "net_perftests") 422 423 def TestPhoneNumber(self): 424 return self.SimpleTest("phonenumber", "libphonenumber_unittests") 425 426 def TestPPAPI(self): 427 return self.SimpleTest("chrome", "ppapi_unittests") 428 429 def TestPrinting(self): 430 return self.SimpleTest("chrome", "printing_unittests") 431 432 def TestRemoting(self): 433 return self.SimpleTest("chrome", "remoting_unittests", 434 cmd_args=[ 435 "--ui-test-action-timeout=60000", 436 "--ui-test-action-max-timeout=150000"]) 437 438 def TestSql(self): 439 return self.SimpleTest("chrome", "sql_unittests") 440 441 def TestSync(self): 442 return self.SimpleTest("chrome", "sync_unit_tests") 443 444 def TestLinuxSandbox(self): 445 return self.SimpleTest("sandbox", "sandbox_linux_unittests") 446 447 def TestUnit(self): 448 # http://crbug.com/51716 449 # Disabling all unit tests 450 # Problems reappeared after r119922 451 if common.IsMac() and (self._options.valgrind_tool == "memcheck"): 452 logging.warning("unit_tests are disabled for memcheck on MacOS.") 453 return 0; 454 return self.SimpleTest("chrome", "unit_tests") 455 456 def TestUIUnit(self): 457 return self.SimpleTest("chrome", "ui_unittests") 458 459 def TestURL(self): 460 return self.SimpleTest("chrome", "url_unittests") 461 462 def TestViews(self): 463 return self.SimpleTest("views", "views_unittests") 464 465 466 # Valgrind timeouts are in seconds. 467 UI_VALGRIND_ARGS = ["--timeout=14400", "--trace_children", "--indirect"] 468 # UI test timeouts are in milliseconds. 469 UI_TEST_ARGS = ["--ui-test-action-timeout=60000", 470 "--ui-test-action-max-timeout=150000", 471 "--no-sandbox"] 472 473 # TODO(thestig) fine-tune these values. 474 # Valgrind timeouts are in seconds. 475 BROWSER_VALGRIND_ARGS = ["--timeout=50000", "--trace_children", "--indirect"] 476 # Browser test timeouts are in milliseconds. 477 BROWSER_TEST_ARGS = ["--ui-test-action-timeout=400000", 478 "--ui-test-action-max-timeout=800000", 479 "--no-sandbox"] 480 481 def TestBrowser(self): 482 return self.SimpleTest("chrome", "browser_tests", 483 valgrind_test_args=self.BROWSER_VALGRIND_ARGS, 484 cmd_args=self.BROWSER_TEST_ARGS) 485 486 def TestContentBrowser(self): 487 return self.SimpleTest("content", "content_browsertests", 488 valgrind_test_args=self.BROWSER_VALGRIND_ARGS, 489 cmd_args=self.BROWSER_TEST_ARGS) 490 491 def TestInteractiveUI(self): 492 return self.SimpleTest("chrome", "interactive_ui_tests", 493 valgrind_test_args=self.UI_VALGRIND_ARGS, 494 cmd_args=self.UI_TEST_ARGS) 495 496 def TestSafeBrowsing(self): 497 return self.SimpleTest("chrome", "safe_browsing_tests", 498 valgrind_test_args=self.UI_VALGRIND_ARGS, 499 cmd_args=(["--ui-test-action-max-timeout=450000"])) 500 501 def TestSyncIntegration(self): 502 return self.SimpleTest("chrome", "sync_integration_tests", 503 valgrind_test_args=self.UI_VALGRIND_ARGS, 504 cmd_args=(["--ui-test-action-max-timeout=450000"])) 505 506 def TestLayoutChunk(self, chunk_num, chunk_size): 507 # Run tests [chunk_num*chunk_size .. (chunk_num+1)*chunk_size) from the 508 # list of tests. Wrap around to beginning of list at end. 509 # If chunk_size is zero, run all tests in the list once. 510 # If a text file is given as argument, it is used as the list of tests. 511 assert((chunk_size == 0) != (len(self._args) == 0)) 512 # Build the ginormous commandline in 'cmd'. 513 # It's going to be roughly 514 # python valgrind_test.py ... python run_webkit_tests.py ... 515 # but we'll use the --indirect flag to valgrind_test.py 516 # to avoid valgrinding python. 517 # Start by building the valgrind_test.py commandline. 518 tool = valgrind_test.CreateTool(self._options.valgrind_tool) 519 cmd = self._DefaultCommand(tool) 520 cmd.append("--trace_children") 521 cmd.append("--indirect_webkit_layout") 522 cmd.append("--ignore_exit_code") 523 # Now build script_cmd, the run_webkits_tests.py commandline 524 # Store each chunk in its own directory so that we can find the data later 525 chunk_dir = os.path.join("layout", "chunk_%05d" % chunk_num) 526 out_dir = os.path.join(path_utils.ScriptDir(), "latest") 527 out_dir = os.path.join(out_dir, chunk_dir) 528 if os.path.exists(out_dir): 529 old_files = glob.glob(os.path.join(out_dir, "*.txt")) 530 for f in old_files: 531 os.remove(f) 532 else: 533 os.makedirs(out_dir) 534 script = os.path.join(self._source_dir, "webkit", "tools", "layout_tests", 535 "run_webkit_tests.py") 536 # http://crbug.com/260627: After the switch to content_shell from DRT, each 537 # test now brings up 3 processes. Under Valgrind, they become memory bound 538 # and can eventually OOM if we don't reduce the total count. 539 # It'd be nice if content_shell automatically throttled the startup of new 540 # tests if we're low on memory. 541 jobs = max(1, int(multiprocessing.cpu_count() * 0.3)) 542 script_cmd = ["python", script, "-v", 543 # run a separate DumpRenderTree for each test 544 "--batch-size=1", 545 "--fully-parallel", 546 "--child-processes=%d" % jobs, 547 "--time-out-ms=800000", 548 "--no-retry-failures", # retrying takes too much time 549 # http://crbug.com/176908: Don't launch a browser when done. 550 "--no-show-results", 551 "--nocheck-sys-deps"] 552 # Pass build mode to run_webkit_tests.py. We aren't passed it directly, 553 # so parse it out of build_dir. run_webkit_tests.py can only handle 554 # the two values "Release" and "Debug". 555 # TODO(Hercules): unify how all our scripts pass around build mode 556 # (--mode / --target / --build-dir / --debug) 557 if self._options.build_dir: 558 build_root, mode = os.path.split(self._options.build_dir) 559 script_cmd.extend(["--build-directory", build_root, "--target", mode]) 560 if (chunk_size > 0): 561 script_cmd.append("--run-chunk=%d:%d" % (chunk_num, chunk_size)) 562 if len(self._args): 563 # if the arg is a txt file, then treat it as a list of tests 564 if os.path.isfile(self._args[0]) and self._args[0][-4:] == ".txt": 565 script_cmd.append("--test-list=%s" % self._args[0]) 566 else: 567 script_cmd.extend(self._args) 568 self._AppendGtestFilter(tool, "layout", script_cmd) 569 # Now run script_cmd with the wrapper in cmd 570 cmd.extend(["--"]) 571 cmd.extend(script_cmd) 572 573 # Layout tests often times fail quickly, but the buildbot remains green. 574 # Detect this situation when running with the default chunk size. 575 if chunk_size == self.LAYOUT_TESTS_DEFAULT_CHUNK_SIZE: 576 min_runtime_in_seconds=120 577 else: 578 min_runtime_in_seconds=0 579 ret = tool.Run(cmd, "layout", min_runtime_in_seconds=min_runtime_in_seconds) 580 return ret 581 582 583 def TestLayout(self): 584 # A "chunk file" is maintained in the local directory so that each test 585 # runs a slice of the layout tests of size chunk_size that increments with 586 # each run. Since tests can be added and removed from the layout tests at 587 # any time, this is not going to give exact coverage, but it will allow us 588 # to continuously run small slices of the layout tests under valgrind rather 589 # than having to run all of them in one shot. 590 chunk_size = self._options.num_tests 591 if chunk_size == 0 or len(self._args): 592 return self.TestLayoutChunk(0, 0) 593 chunk_num = 0 594 chunk_file = os.path.join("valgrind_layout_chunk.txt") 595 logging.info("Reading state from " + chunk_file) 596 try: 597 f = open(chunk_file) 598 if f: 599 chunk_str = f.read() 600 if len(chunk_str): 601 chunk_num = int(chunk_str) 602 # This should be enough so that we have a couple of complete runs 603 # of test data stored in the archive (although note that when we loop 604 # that we almost guaranteed won't be at the end of the test list) 605 if chunk_num > 10000: 606 chunk_num = 0 607 f.close() 608 except IOError, (errno, strerror): 609 logging.error("error reading from file %s (%d, %s)" % (chunk_file, 610 errno, strerror)) 611 # Save the new chunk size before running the tests. Otherwise if a 612 # particular chunk hangs the bot, the chunk number will never get 613 # incremented and the bot will be wedged. 614 logging.info("Saving state to " + chunk_file) 615 try: 616 f = open(chunk_file, "w") 617 chunk_num += 1 618 f.write("%d" % chunk_num) 619 f.close() 620 except IOError, (errno, strerror): 621 logging.error("error writing to file %s (%d, %s)" % (chunk_file, errno, 622 strerror)) 623 # Since we're running small chunks of the layout tests, it's important to 624 # mark the ones that have errors in them. These won't be visible in the 625 # summary list for long, but will be useful for someone reviewing this bot. 626 return self.TestLayoutChunk(chunk_num, chunk_size) 627 628 # The known list of tests. 629 # Recognise the original abbreviations as well as full executable names. 630 _test_list = { 631 "cmdline" : RunCmdLine, 632 "addressinput": TestAddressInput, 633 "libaddressinput_unittests": TestAddressInput, 634 "accessibility": TestAccessibility, 635 "angle": TestAngle, "angle_unittests": TestAngle, 636 "app_list": TestAppList, "app_list_unittests": TestAppList, 637 "ash": TestAsh, "ash_unittests": TestAsh, 638 "ash_shell": TestAshShell, "ash_shell_unittests": TestAshShell, 639 "aura": TestAura, "aura_unittests": TestAura, 640 "base": TestBase, "base_unittests": TestBase, 641 "blink_heap": TestBlinkHeap, 642 "blink_platform": TestBlinkPlatform, 643 "browser": TestBrowser, "browser_tests": TestBrowser, 644 "cacheinvalidation": TestCacheInvalidation, 645 "cacheinvalidation_unittests": TestCacheInvalidation, 646 "cast": TestCast, "cast_unittests": TestCast, 647 "cc": TestCC, "cc_unittests": TestCC, 648 "chrome_app": TestChromeApp, 649 "chrome_elf": TestChromeElf, 650 "chromedriver": TestChromeDriver, 651 "chromeos": TestChromeOS, "chromeos_unittests": TestChromeOS, 652 "cloud_print": TestCloudPrint, 653 "cloud_print_unittests": TestCloudPrint, 654 "components": TestComponents,"components_unittests": TestComponents, 655 "compositor": TestCompositor,"compositor_unittests": TestCompositor, 656 "content": TestContent, "content_unittests": TestContent, 657 "content_browsertests": TestContentBrowser, 658 "courgette": TestCourgette, "courgette_unittests": TestCourgette, 659 "crypto": TestCrypto, "crypto_unittests": TestCrypto, 660 "device": TestDevice, "device_unittests": TestDevice, 661 "display": TestDisplay, "display_unittests": TestDisplay, 662 "events": TestEvents, "events_unittests": TestEvents, 663 "extensions": TestExtensions, "extensions_unittests": TestExtensions, 664 "ffmpeg": TestFFmpeg, "ffmpeg_unittests": TestFFmpeg, 665 "ffmpeg_regression_tests": TestFFmpegRegressions, 666 "gcm": TestGCM, "gcm_unit_tests": TestGCM, 667 "gin": TestGin, "gin_unittests": TestGin, 668 "gfx": TestGfx, "gfx_unittests": TestGfx, 669 "google_apis": TestGoogleApis, 670 "gpu": TestGPU, "gpu_unittests": TestGPU, 671 "ipc": TestIpc, "ipc_tests": TestIpc, 672 "installer_util": TestInstallerUtil, 673 "interactive_ui": TestInteractiveUI, 674 "jingle": TestJingle, "jingle_unittests": TestJingle, 675 "keyboard": TestKeyboard, "keyboard_unittests": TestKeyboard, 676 "layout": TestLayout, "layout_tests": TestLayout, 677 "media": TestMedia, "media_unittests": TestMedia, 678 "message_center": TestMessageCenter, 679 "message_center_unittests" : TestMessageCenter, 680 "mojo_apps_js": TestMojoAppsJS, 681 "mojo_common": TestMojoCommon, 682 "mojo_js": TestMojoJS, 683 "mojo_system": TestMojoSystem, 684 "mojo_public_system": TestMojoPublicSystem, 685 "mojo_public_utility": TestMojoPublicUtility, 686 "mojo_public_bindings": TestMojoPublicBindings, 687 "mojo_public_env": TestMojoPublicEnv, 688 "mojo_public_sysperf": TestMojoPublicSysPerf, 689 "mojo_application_manager": TestMojoApplicationManager, 690 "mojo_view_manager": TestMojoViewManager, 691 "net": TestNet, "net_unittests": TestNet, 692 "net_perf": TestNetPerf, "net_perftests": TestNetPerf, 693 "phonenumber": TestPhoneNumber, 694 "libphonenumber_unittests": TestPhoneNumber, 695 "ppapi": TestPPAPI, "ppapi_unittests": TestPPAPI, 696 "printing": TestPrinting, "printing_unittests": TestPrinting, 697 "remoting": TestRemoting, "remoting_unittests": TestRemoting, 698 "safe_browsing": TestSafeBrowsing, "safe_browsing_tests": TestSafeBrowsing, 699 "sandbox": TestLinuxSandbox, "sandbox_linux_unittests": TestLinuxSandbox, 700 "sql": TestSql, "sql_unittests": TestSql, 701 "sync": TestSync, "sync_unit_tests": TestSync, 702 "sync_integration_tests": TestSyncIntegration, 703 "sync_integration": TestSyncIntegration, 704 "ui_unit": TestUIUnit, "ui_unittests": TestUIUnit, 705 "unit": TestUnit, "unit_tests": TestUnit, 706 "url": TestURL, "url_unittests": TestURL, 707 "views": TestViews, "views_unittests": TestViews, 708 "webkit": TestLayout, 709 } 710 711 712def _main(): 713 parser = optparse.OptionParser("usage: %prog -b <dir> -t <test> " 714 "[-t <test> ...]") 715 716 parser.add_option("--help-tests", dest="help_tests", action="store_true", 717 default=False, help="List all available tests") 718 parser.add_option("-b", "--build-dir", 719 help="the location of the compiler output") 720 parser.add_option("--target", help="Debug or Release") 721 parser.add_option("-t", "--test", action="append", default=[], 722 help="which test to run, supports test:gtest_filter format " 723 "as well.") 724 parser.add_option("--baseline", action="store_true", default=False, 725 help="generate baseline data instead of validating") 726 parser.add_option("--gtest_filter", 727 help="additional arguments to --gtest_filter") 728 parser.add_option("--gtest_repeat", help="argument for --gtest_repeat") 729 parser.add_option("--gtest_shuffle", action="store_true", default=False, 730 help="Randomize tests' orders on every iteration.") 731 parser.add_option("-v", "--verbose", action="store_true", default=False, 732 help="verbose output - enable debug log messages") 733 parser.add_option("--tool", dest="valgrind_tool", default="memcheck", 734 help="specify a valgrind tool to run the tests under") 735 parser.add_option("--tool_flags", dest="valgrind_tool_flags", default="", 736 help="specify custom flags for the selected valgrind tool") 737 parser.add_option("--keep_logs", action="store_true", default=False, 738 help="store memory tool logs in the <tool>.logs directory " 739 "instead of /tmp.\nThis can be useful for tool " 740 "developers/maintainers.\nPlease note that the <tool>" 741 ".logs directory will be clobbered on tool startup.") 742 parser.add_option("-n", "--num_tests", type="int", 743 default=ChromeTests.LAYOUT_TESTS_DEFAULT_CHUNK_SIZE, 744 help="for layout tests: # of subtests per run. 0 for all.") 745 # TODO(thestig) Remove this if we can. 746 parser.add_option("--gtest_color", dest="gtest_color", default="no", 747 help="dummy compatibility flag for sharding_supervisor.") 748 parser.add_option("--brave-new-test-launcher", action="store_true", 749 help="run the tests with --brave-new-test-launcher") 750 parser.add_option("--test-launcher-bot-mode", action="store_true", 751 help="run the tests with --test-launcher-bot-mode") 752 753 options, args = parser.parse_args() 754 755 # Bake target into build_dir. 756 if options.target and options.build_dir: 757 assert (options.target != 758 os.path.basename(os.path.dirname(options.build_dir))) 759 options.build_dir = os.path.join(os.path.abspath(options.build_dir), 760 options.target) 761 762 if options.verbose: 763 logging_utils.config_root(logging.DEBUG) 764 else: 765 logging_utils.config_root() 766 767 if options.help_tests: 768 ChromeTests.ShowTests() 769 return 0 770 771 if not options.test: 772 parser.error("--test not specified") 773 774 if len(options.test) != 1 and options.gtest_filter: 775 parser.error("--gtest_filter and multiple tests don't make sense together") 776 777 for t in options.test: 778 tests = ChromeTests(options, args, t) 779 ret = tests.Run() 780 if ret: return ret 781 return 0 782 783 784if __name__ == "__main__": 785 sys.exit(_main()) 786