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