chrome_tests.py revision 010d83a9304c5a91596085d917d248abff47903a 
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 TestAppList(self):
258    return self.SimpleTest("app_list", "app_list_unittests")
259
260  def TestAsh(self):
261    return self.SimpleTest("ash", "ash_unittests")
262
263  def TestAura(self):
264    return self.SimpleTest("aura", "aura_unittests")
265
266  def TestBase(self):
267    return self.SimpleTest("base", "base_unittests")
268
269  def TestCast(self):
270    return self.SimpleTest("chrome", "cast_unittests")
271
272  def TestChromeOS(self):
273    return self.SimpleTest("chromeos", "chromeos_unittests")
274
275  def TestComponents(self):
276    return self.SimpleTest("components", "components_unittests")
277
278  def TestCompositor(self):
279    return self.SimpleTest("compositor", "compositor_unittests")
280
281  def TestContent(self):
282    return self.SimpleTest("content", "content_unittests")
283
284  def TestCourgette(self):
285    return self.SimpleTest("courgette", "courgette_unittests")
286
287  def TestCrypto(self):
288    return self.SimpleTest("crypto", "crypto_unittests")
289
290  def TestDevice(self):
291    return self.SimpleTest("device", "device_unittests")
292
293  def TestEvents(self):
294    return self.SimpleTest("events", "events_unittests")
295
296  def TestFFmpeg(self):
297    return self.SimpleTest("chrome", "ffmpeg_unittests")
298
299  def TestFFmpegRegressions(self):
300    return self.SimpleTest("chrome", "ffmpeg_regression_tests")
301
302  def TestGCM(self):
303    return self.SimpleTest("gcm", "gcm_unit_tests")
304
305  def TestGPU(self):
306    return self.SimpleTest("gpu", "gpu_unittests")
307
308  def TestIpc(self):
309    return self.SimpleTest("ipc", "ipc_tests",
310                           valgrind_test_args=["--trace_children"])
311
312  def TestJingle(self):
313    return self.SimpleTest("chrome", "jingle_unittests")
314
315  def TestMedia(self):
316    return self.SimpleTest("chrome", "media_unittests")
317
318  def TestMessageCenter(self):
319    return self.SimpleTest("message_center", "message_center_unittests")
320
321  def TestNet(self):
322    return self.SimpleTest("net", "net_unittests")
323
324  def TestNetPerf(self):
325    return self.SimpleTest("net", "net_perftests")
326
327  def TestPPAPI(self):
328    return self.SimpleTest("chrome", "ppapi_unittests")
329
330  def TestPrinting(self):
331    return self.SimpleTest("chrome", "printing_unittests")
332
333  def TestRemoting(self):
334    return self.SimpleTest("chrome", "remoting_unittests",
335                           cmd_args=[
336                               "--ui-test-action-timeout=60000",
337                               "--ui-test-action-max-timeout=150000"])
338
339  def TestSql(self):
340    return self.SimpleTest("chrome", "sql_unittests")
341
342  def TestSync(self):
343    return self.SimpleTest("chrome", "sync_unit_tests")
344
345  def TestLinuxSandbox(self):
346    return self.SimpleTest("sandbox", "sandbox_linux_unittests")
347
348  def TestUnit(self):
349    # http://crbug.com/51716
350    # Disabling all unit tests
351    # Problems reappeared after r119922
352    if common.IsMac() and (self._options.valgrind_tool == "memcheck"):
353      logging.warning("unit_tests are disabled for memcheck on MacOS.")
354      return 0;
355    return self.SimpleTest("chrome", "unit_tests")
356
357  def TestUIUnit(self):
358    return self.SimpleTest("chrome", "ui_unittests")
359
360  def TestURL(self):
361    return self.SimpleTest("chrome", "url_unittests")
362
363  def TestViews(self):
364    return self.SimpleTest("views", "views_unittests")
365
366  def TestCloudPrint(self):
367    return self.SimpleTest("cloud_print", "cloud_print_unittests")
368
369  def TestCacheInvalidation(self):
370    return self.SimpleTest("cacheinvalidation", "cacheinvalidation_unittests")
371
372  def TestAddressInput(self):
373    return self.SimpleTest("addressinput", "libaddressinput_unittests")
374
375  def TestPhoneNumber(self):
376    return self.SimpleTest("phonenumber", "libphonenumber_unittests")
377
378  def TestMojoSystem(self):
379    return self.SimpleTest("mojo_system", "mojo_system_unittests")
380
381  def TestMojoPublicSystem(self):
382    return self.SimpleTest("mojo_public_system",
383                           "mojo_public_system_unittests")
384
385  def TestMojoPublicUtility(self):
386    return self.SimpleTest("mojo_public_utility",
387                           "mojo_public_unittests_unittests")
388
389  def TestMojoPublicBindings(self):
390    return self.SimpleTest("mojo_public_bindings",
391                           "mojo_public_bindings_unittests")
392
393  def TestMojoPublicEnv(self):
394    return self.SimpleTest("mojo_public_env",
395                           "mojo_public_environment_unittests")
396
397  def TestMojoPublicSysPerf(self):
398    return self.SimpleTest("mojo_public_sysperf",
399                           "mojo_public_system_perftests")
400
401  def TestMojoCommon(self):
402    return self.SimpleTest("mojo_common", "mojo_common_unittests")
403
404  def TestMojoAppsJS(self):
405    return self.SimpleTest("mojo_apps_js", "mojo_apps_js_unittests")
406
407  def TestMojoJS(self):
408    return self.SimpleTest("mojo_js", "mojo_js_unittests")
409
410  def TestMojoServiceManager(self):
411    return self.SimpleTest("mojo_service_manager",
412                           "mojo_service_manager_unittests")
413
414  def TestMojoViewManager(self):
415    return self.SimpleTest("mojo_view_manager", "mojo_view_manager_unittests")
416
417  # Valgrind timeouts are in seconds.
418  UI_VALGRIND_ARGS = ["--timeout=14400", "--trace_children", "--indirect"]
419  # UI test timeouts are in milliseconds.
420  UI_TEST_ARGS = ["--ui-test-action-timeout=60000",
421                  "--ui-test-action-max-timeout=150000",
422                  "--no-sandbox"]
423
424  # TODO(thestig) fine-tune these values.
425  # Valgrind timeouts are in seconds.
426  BROWSER_VALGRIND_ARGS = ["--timeout=50000", "--trace_children", "--indirect"]
427  # Browser test timeouts are in milliseconds.
428  BROWSER_TEST_ARGS = ["--ui-test-action-timeout=400000",
429                       "--ui-test-action-max-timeout=800000",
430                       "--no-sandbox"]
431
432  def TestAutomatedUI(self):
433    return self.SimpleTest("chrome", "automated_ui_tests",
434                           valgrind_test_args=self.UI_VALGRIND_ARGS,
435                           cmd_args=self.UI_TEST_ARGS)
436
437  def TestBrowser(self):
438    return self.SimpleTest("chrome", "browser_tests",
439                           valgrind_test_args=self.BROWSER_VALGRIND_ARGS,
440                           cmd_args=self.BROWSER_TEST_ARGS)
441
442  def TestContentBrowser(self):
443    return self.SimpleTest("content", "content_browsertests",
444                           valgrind_test_args=self.BROWSER_VALGRIND_ARGS,
445                           cmd_args=self.BROWSER_TEST_ARGS)
446
447  def TestInteractiveUI(self):
448    return self.SimpleTest("chrome", "interactive_ui_tests",
449                           valgrind_test_args=self.UI_VALGRIND_ARGS,
450                           cmd_args=self.UI_TEST_ARGS)
451
452  def TestReliability(self):
453    script_dir = path_utils.ScriptDir()
454    url_list_file = os.path.join(script_dir, "reliability", "url_list.txt")
455    return self.SimpleTest("chrome", "reliability_tests",
456                           valgrind_test_args=self.UI_VALGRIND_ARGS,
457                           cmd_args=(self.UI_TEST_ARGS +
458                                     ["--list=%s" % url_list_file]))
459
460  def TestSafeBrowsing(self):
461    return self.SimpleTest("chrome", "safe_browsing_tests",
462                           valgrind_test_args=self.UI_VALGRIND_ARGS,
463                           cmd_args=(["--ui-test-action-max-timeout=450000"]))
464
465  def TestSyncIntegration(self):
466    return self.SimpleTest("chrome", "sync_integration_tests",
467                           valgrind_test_args=self.UI_VALGRIND_ARGS,
468                           cmd_args=(["--ui-test-action-max-timeout=450000"]))
469
470  def TestLayoutChunk(self, chunk_num, chunk_size):
471    # Run tests [chunk_num*chunk_size .. (chunk_num+1)*chunk_size) from the
472    # list of tests.  Wrap around to beginning of list at end.
473    # If chunk_size is zero, run all tests in the list once.
474    # If a text file is given as argument, it is used as the list of tests.
475    #
476    # Build the ginormous commandline in 'cmd'.
477    # It's going to be roughly
478    #  python valgrind_test.py ... python run_webkit_tests.py ...
479    # but we'll use the --indirect flag to valgrind_test.py
480    # to avoid valgrinding python.
481    # Start by building the valgrind_test.py commandline.
482    tool = valgrind_test.CreateTool(self._options.valgrind_tool)
483    cmd = self._DefaultCommand(tool)
484    cmd.append("--trace_children")
485    cmd.append("--indirect_webkit_layout")
486    cmd.append("--ignore_exit_code")
487    # Now build script_cmd, the run_webkits_tests.py commandline
488    # Store each chunk in its own directory so that we can find the data later
489    chunk_dir = os.path.join("layout", "chunk_%05d" % chunk_num)
490    out_dir = os.path.join(path_utils.ScriptDir(), "latest")
491    out_dir = os.path.join(out_dir, chunk_dir)
492    if os.path.exists(out_dir):
493      old_files = glob.glob(os.path.join(out_dir, "*.txt"))
494      for f in old_files:
495        os.remove(f)
496    else:
497      os.makedirs(out_dir)
498    script = os.path.join(self._source_dir, "webkit", "tools", "layout_tests",
499                          "run_webkit_tests.py")
500    # http://crbug.com/260627: After the switch to content_shell from DRT, each
501    # test now brings up 3 processes.  Under Valgrind, they become memory bound
502    # and can eventually OOM if we don't reduce the total count.
503    # It'd be nice if content_shell automatically throttled the startup of new
504    # tests if we're low on memory.
505    jobs = max(1, int(multiprocessing.cpu_count() * 0.3))
506    script_cmd = ["python", script, "-v",
507                  # run a separate DumpRenderTree for each test
508                  "--batch-size=1",
509                  "--fully-parallel",
510                  "--child-processes=%d" % jobs,
511                  "--time-out-ms=800000",
512                  "--no-retry-failures",  # retrying takes too much time
513                  # http://crbug.com/176908: Don't launch a browser when done.
514                  "--no-show-results",
515                  "--nocheck-sys-deps"]
516    # Pass build mode to run_webkit_tests.py.  We aren't passed it directly,
517    # so parse it out of build_dir.  run_webkit_tests.py can only handle
518    # the two values "Release" and "Debug".
519    # TODO(Hercules): unify how all our scripts pass around build mode
520    # (--mode / --target / --build-dir / --debug)
521    if self._options.build_dir:
522      build_root, mode = os.path.split(self._options.build_dir)
523      script_cmd.extend(["--build-directory", build_root, "--target", mode])
524    if (chunk_size > 0):
525      script_cmd.append("--run-chunk=%d:%d" % (chunk_num, chunk_size))
526    if len(self._args):
527      # if the arg is a txt file, then treat it as a list of tests
528      if os.path.isfile(self._args[0]) and self._args[0][-4:] == ".txt":
529        script_cmd.append("--test-list=%s" % self._args[0])
530      else:
531        script_cmd.extend(self._args)
532    self._AppendGtestFilter(tool, "layout", script_cmd)
533    # Now run script_cmd with the wrapper in cmd
534    cmd.extend(["--"])
535    cmd.extend(script_cmd)
536
537    # Layout tests often times fail quickly, but the buildbot remains green.
538    # Detect this situation when running with the default chunk size.
539    if chunk_size == self.LAYOUT_TESTS_DEFAULT_CHUNK_SIZE:
540      min_runtime_in_seconds=120
541    else:
542      min_runtime_in_seconds=0
543    ret = tool.Run(cmd, "layout", min_runtime_in_seconds=min_runtime_in_seconds)
544    return ret
545
546
547  def TestLayout(self):
548    # A "chunk file" is maintained in the local directory so that each test
549    # runs a slice of the layout tests of size chunk_size that increments with
550    # each run.  Since tests can be added and removed from the layout tests at
551    # any time, this is not going to give exact coverage, but it will allow us
552    # to continuously run small slices of the layout tests under valgrind rather
553    # than having to run all of them in one shot.
554    chunk_size = self._options.num_tests
555    if (chunk_size == 0):
556      return self.TestLayoutChunk(0, 0)
557    chunk_num = 0
558    chunk_file = os.path.join("valgrind_layout_chunk.txt")
559    logging.info("Reading state from " + chunk_file)
560    try:
561      f = open(chunk_file)
562      if f:
563        chunk_str = f.read()
564        if len(chunk_str):
565          chunk_num = int(chunk_str)
566        # This should be enough so that we have a couple of complete runs
567        # of test data stored in the archive (although note that when we loop
568        # that we almost guaranteed won't be at the end of the test list)
569        if chunk_num > 10000:
570          chunk_num = 0
571        f.close()
572    except IOError, (errno, strerror):
573      logging.error("error reading from file %s (%d, %s)" % (chunk_file,
574                    errno, strerror))
575    # Save the new chunk size before running the tests. Otherwise if a
576    # particular chunk hangs the bot, the chunk number will never get
577    # incremented and the bot will be wedged.
578    logging.info("Saving state to " + chunk_file)
579    try:
580      f = open(chunk_file, "w")
581      chunk_num += 1
582      f.write("%d" % chunk_num)
583      f.close()
584    except IOError, (errno, strerror):
585      logging.error("error writing to file %s (%d, %s)" % (chunk_file, errno,
586                    strerror))
587    # Since we're running small chunks of the layout tests, it's important to
588    # mark the ones that have errors in them.  These won't be visible in the
589    # summary list for long, but will be useful for someone reviewing this bot.
590    return self.TestLayoutChunk(chunk_num, chunk_size)
591
592  # The known list of tests.
593  # Recognise the original abbreviations as well as full executable names.
594  _test_list = {
595    "cmdline" : RunCmdLine,
596    "app_list": TestAppList,     "app_list_unittests": TestAppList,
597    "ash": TestAsh,              "ash_unittests": TestAsh,
598    "aura": TestAura,            "aura_unittests": TestAura,
599    "automated_ui" : TestAutomatedUI,
600    "base": TestBase,            "base_unittests": TestBase,
601    "browser": TestBrowser,      "browser_tests": TestBrowser,
602    "cast": TestCast,            "cast_unittests": TestCast,
603    "chromeos": TestChromeOS,    "chromeos_unittests": TestChromeOS,
604    "components": TestComponents,"components_unittests": TestComponents,
605    "compositor": TestCompositor,"compositor_unittests": TestCompositor,
606    "content": TestContent,      "content_unittests": TestContent,
607    "content_browsertests": TestContentBrowser,
608    "courgette": TestCourgette,  "courgette_unittests": TestCourgette,
609    "crypto": TestCrypto,        "crypto_unittests": TestCrypto,
610    "device": TestDevice,        "device_unittests": TestDevice,
611    "events": TestEvents,        "events_unittests": TestEvents,
612    "ffmpeg": TestFFmpeg,        "ffmpeg_unittests": TestFFmpeg,
613    "ffmpeg_regression_tests": TestFFmpegRegressions,
614    "gcm": TestGCM,              "gcm_unit_tests": TestGCM,
615    "gpu": TestGPU,              "gpu_unittests": TestGPU,
616    "ipc": TestIpc,              "ipc_tests": TestIpc,
617    "interactive_ui": TestInteractiveUI,
618    "jingle": TestJingle,        "jingle_unittests": TestJingle,
619    "layout": TestLayout,        "layout_tests": TestLayout,
620    "webkit": TestLayout,
621    "media": TestMedia,          "media_unittests": TestMedia,
622    "message_center": TestMessageCenter,
623    "message_center_unittests" : TestMessageCenter,
624    "net": TestNet,              "net_unittests": TestNet,
625    "net_perf": TestNetPerf,     "net_perftests": TestNetPerf,
626    "ppapi": TestPPAPI,          "ppapi_unittests": TestPPAPI,
627    "printing": TestPrinting,    "printing_unittests": TestPrinting,
628    "reliability": TestReliability, "reliability_tests": TestReliability,
629    "remoting": TestRemoting,    "remoting_unittests": TestRemoting,
630    "safe_browsing": TestSafeBrowsing, "safe_browsing_tests": TestSafeBrowsing,
631    "sandbox": TestLinuxSandbox, "sandbox_linux_unittests": TestLinuxSandbox,
632    "sql": TestSql,              "sql_unittests": TestSql,
633    "sync": TestSync,            "sync_unit_tests": TestSync,
634    "sync_integration_tests": TestSyncIntegration,
635    "sync_integration": TestSyncIntegration,
636    "ui_unit": TestUIUnit,       "ui_unittests": TestUIUnit,
637    "unit": TestUnit,            "unit_tests": TestUnit,
638    "url": TestURL,              "url_unittests": TestURL,
639    "views": TestViews,          "views_unittests": TestViews,
640    "cloud_print": TestCloudPrint,
641    "cloud_print_unittests": TestCloudPrint,
642    "cacheinvalidation": TestCacheInvalidation,
643    "cacheinvalidation_unittests": TestCacheInvalidation,
644    "addressinput": TestAddressInput,
645    "libaddressinput_unittests": TestAddressInput,
646    "phonenumber": TestPhoneNumber,
647    "libphonenumber_unittests": TestPhoneNumber,
648    "mojo_system": TestMojoSystem,
649    "mojo_public_system": TestMojoPublicSystem,
650    "mojo_public_utility": TestMojoPublicUtility,
651    "mojo_public_bindings": TestMojoPublicBindings,
652    "mojo_public_env": TestMojoPublicEnv,
653    "mojo_public_sysperf": TestMojoPublicSysPerf,
654    "mojo_common": TestMojoCommon,
655    "mojo_apps_js": TestMojoAppsJS,
656    "mojo_js": TestMojoJS,
657    "mojo_service_manager": TestMojoServiceManager,
658    "mojo_view_manager": TestMojoViewManager,
659  }
660
661
662def _main():
663  parser = optparse.OptionParser("usage: %prog -b <dir> -t <test> "
664                                 "[-t <test> ...]")
665
666  parser.add_option("--help-tests", dest="help_tests", action="store_true",
667                    default=False, help="List all available tests")
668  parser.add_option("-b", "--build-dir",
669                    help="the location of the compiler output")
670  parser.add_option("--target", help="Debug or Release")
671  parser.add_option("-t", "--test", action="append", default=[],
672                    help="which test to run, supports test:gtest_filter format "
673                         "as well.")
674  parser.add_option("--baseline", action="store_true", default=False,
675                    help="generate baseline data instead of validating")
676  parser.add_option("--gtest_filter",
677                    help="additional arguments to --gtest_filter")
678  parser.add_option("--gtest_repeat", help="argument for --gtest_repeat")
679  parser.add_option("--gtest_shuffle", action="store_true", default=False,
680                    help="Randomize tests' orders on every iteration.")
681  parser.add_option("-v", "--verbose", action="store_true", default=False,
682                    help="verbose output - enable debug log messages")
683  parser.add_option("--tool", dest="valgrind_tool", default="memcheck",
684                    help="specify a valgrind tool to run the tests under")
685  parser.add_option("--tool_flags", dest="valgrind_tool_flags", default="",
686                    help="specify custom flags for the selected valgrind tool")
687  parser.add_option("--keep_logs", action="store_true", default=False,
688                    help="store memory tool logs in the <tool>.logs directory "
689                         "instead of /tmp.\nThis can be useful for tool "
690                         "developers/maintainers.\nPlease note that the <tool>"
691                         ".logs directory will be clobbered on tool startup.")
692  parser.add_option("-n", "--num_tests", type="int",
693                    default=ChromeTests.LAYOUT_TESTS_DEFAULT_CHUNK_SIZE,
694                    help="for layout tests: # of subtests per run.  0 for all.")
695  # TODO(thestig) Remove this if we can.
696  parser.add_option("--gtest_color", dest="gtest_color", default="no",
697                    help="dummy compatibility flag for sharding_supervisor.")
698  parser.add_option("--brave-new-test-launcher", action="store_true",
699                    help="run the tests with --brave-new-test-launcher")
700  parser.add_option("--test-launcher-bot-mode", action="store_true",
701                    help="run the tests with --test-launcher-bot-mode")
702
703  options, args = parser.parse_args()
704
705  # Bake target into build_dir.
706  if options.target and options.build_dir:
707    assert (options.target !=
708            os.path.basename(os.path.dirname(options.build_dir)))
709    options.build_dir = os.path.join(os.path.abspath(options.build_dir),
710                                     options.target)
711
712  if options.verbose:
713    logging_utils.config_root(logging.DEBUG)
714  else:
715    logging_utils.config_root()
716
717  if options.help_tests:
718    ChromeTests.ShowTests()
719    return 0
720
721  if not options.test:
722    parser.error("--test not specified")
723
724  if len(options.test) != 1 and options.gtest_filter:
725    parser.error("--gtest_filter and multiple tests don't make sense together")
726
727  for t in options.test:
728    tests = ChromeTests(options, args, t)
729    ret = tests.Run()
730    if ret: return ret
731  return 0
732
733
734if __name__ == "__main__":
735  sys.exit(_main())
736