testing/web-platform/mach_commands_base.py

   1 # This Source Code Form is subject to the terms of the Mozilla Public
   2 # License, v. 2.0. If a copy of the MPL was not distributed with this
   3 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
   4
   5 import sys
   6
   7
   8 def create_parser_wpt():
   9     from wptrunner import wptcommandline
  10
  11     result = wptcommandline.create_parser(
  12         ["firefox", "firefox_android", "chrome", "edge", "servo"]
  13     )
  14     result.add_argument(
  15         "--no-install",
  16         action="store_true",
  17         default=False,
  18         help="Do not install test runner application",
  19     )
  20     return result
  21
  22
  23 class WebPlatformTestsRunner(object):
  24     """Run web platform tests."""
  25
  26     def __init__(self, setup):
  27         self.setup = setup
  28
  29     def setup_logging(self, **kwargs):
  30         from tools.wpt import run
  31
  32         return run.setup_logging(
  33             kwargs,
  34             {self.setup.default_log_type: sys.stdout},
  35             formatter_defaults={"screenshot": True},
  36         )
  37
  38     def run(self, logger, **kwargs):
  39         from wptrunner import wptrunner
  40         from mozbuild.base import BinaryNotFoundException
  41
  42         if kwargs["manifest_update"] is not False:
  43             self.update_manifest(logger)
  44         kwargs["manifest_update"] = False
  45
  46         if kwargs["product"] == "firefox":
  47             try:
  48                 kwargs = self.setup.kwargs_firefox(kwargs)
  49             except BinaryNotFoundException as e:
  50                 logger.error(e)
  51                 logger.info(e.help())
  52                 return 1
  53         elif kwargs["product"] == "firefox_android":
  54             from wptrunner import wptcommandline
  55
  56             kwargs = wptcommandline.check_args(self.setup.kwargs_common(kwargs))
  57         elif kwargs["product"] in ("chrome", "edge", "servo"):
  58             kwargs = self.setup.kwargs_wptrun(kwargs)
  59         else:
  60             raise ValueError("Unknown product %s" % kwargs["product"])
  61         result = wptrunner.start(**kwargs)
  62         return int(result)
  63
  64     def update_manifest(self, logger, **kwargs):
  65         import manifestupdate
  66
  67         return manifestupdate.run(
  68             logger=logger,
  69             src_root=self.setup.topsrcdir,
  70             obj_root=self.setup.topobjdir,
  71             **kwargs
  72         )