diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 09:22:09 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 09:22:09 +0000 |
commit | 43a97878ce14b72f0981164f87f2e35e14151312 (patch) | |
tree | 620249daf56c0258faa40cbdcf9cfba06de2a846 /testing/web-platform/mach_commands.py | |
parent | Initial commit. (diff) | |
download | firefox-43a97878ce14b72f0981164f87f2e35e14151312.tar.xz firefox-43a97878ce14b72f0981164f87f2e35e14151312.zip |
Adding upstream version 110.0.1.upstream/110.0.1upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'testing/web-platform/mach_commands.py')
-rw-r--r-- | testing/web-platform/mach_commands.py | 649 |
1 files changed, 649 insertions, 0 deletions
diff --git a/testing/web-platform/mach_commands.py b/testing/web-platform/mach_commands.py new file mode 100644 index 0000000000..4d0333d08d --- /dev/null +++ b/testing/web-platform/mach_commands.py @@ -0,0 +1,649 @@ +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this +# file, You can obtain one at http://mozilla.org/MPL/2.0/. + +# Integrates the web-platform-tests test runner with mach. + +import os +import sys + +from mach.decorators import Command +from mach_commands_base import WebPlatformTestsRunner, create_parser_wpt +from mozbuild.base import MachCommandConditions as conditions +from mozbuild.base import MozbuildObject +from six import iteritems + + +class WebPlatformTestsRunnerSetup(MozbuildObject): + default_log_type = "mach" + + def __init__(self, *args, **kwargs): + super(WebPlatformTestsRunnerSetup, self).__init__(*args, **kwargs) + self._here = os.path.join(self.topsrcdir, "testing", "web-platform") + kwargs["tests_root"] = os.path.join(self._here, "tests") + sys.path.insert(0, kwargs["tests_root"]) + build_path = os.path.join(self.topobjdir, "build") + if build_path not in sys.path: + sys.path.append(build_path) + + def kwargs_common(self, kwargs): + """Setup kwargs relevant for all browser products""" + + tests_src_path = os.path.join(self._here, "tests") + + if ( + kwargs["product"] in {"firefox", "firefox_android"} + and kwargs["specialpowers_path"] is None + ): + kwargs["specialpowers_path"] = os.path.join( + self.distdir, "xpi-stage", "specialpowers@mozilla.org.xpi" + ) + + if kwargs["product"] == "firefox_android": + # package_name may be different in the future + package_name = kwargs["package_name"] + if not package_name: + kwargs[ + "package_name" + ] = package_name = "org.mozilla.geckoview.test_runner" + + # Note that this import may fail in non-firefox-for-android trees + from mozrunner.devices.android_device import ( + InstallIntent, + get_adb_path, + verify_android_device, + ) + + kwargs["adb_binary"] = get_adb_path(self) + install = ( + InstallIntent.NO if kwargs.pop("no_install") else InstallIntent.YES + ) + verify_android_device( + self, install=install, verbose=False, xre=True, app=package_name + ) + + if kwargs["certutil_binary"] is None: + kwargs["certutil_binary"] = os.path.join( + os.environ.get("MOZ_HOST_BIN"), "certutil" + ) + + if kwargs["install_fonts"] is None: + kwargs["install_fonts"] = True + + if not kwargs["device_serial"]: + kwargs["device_serial"] = ["emulator-5554"] + + if kwargs["config"] is None: + kwargs["config"] = os.path.join( + self.topobjdir, "_tests", "web-platform", "wptrunner.local.ini" + ) + + if ( + kwargs["exclude"] is None + and kwargs["include"] is None + and not sys.platform.startswith("linux") + ): + kwargs["exclude"] = ["css"] + + if kwargs["ssl_type"] in (None, "pregenerated"): + cert_root = os.path.join(tests_src_path, "tools", "certs") + if kwargs["ca_cert_path"] is None: + kwargs["ca_cert_path"] = os.path.join(cert_root, "cacert.pem") + + if kwargs["host_key_path"] is None: + kwargs["host_key_path"] = os.path.join( + cert_root, "web-platform.test.key" + ) + + if kwargs["host_cert_path"] is None: + kwargs["host_cert_path"] = os.path.join( + cert_root, "web-platform.test.pem" + ) + + if kwargs["reftest_screenshot"] is None: + kwargs["reftest_screenshot"] = "fail" + + kwargs["capture_stdio"] = True + + return kwargs + + def kwargs_firefox(self, kwargs): + """Setup kwargs specific to running Firefox and other gecko browsers""" + import mozinfo + from wptrunner import wptcommandline + + kwargs = self.kwargs_common(kwargs) + + if kwargs["binary"] is None: + kwargs["binary"] = self.get_binary_path() + + if kwargs["certutil_binary"] is None: + kwargs["certutil_binary"] = self.get_binary_path("certutil") + + if kwargs["webdriver_binary"] is None: + kwargs["webdriver_binary"] = self.get_binary_path( + "geckodriver", validate_exists=False + ) + + if kwargs["install_fonts"] is None: + kwargs["install_fonts"] = True + + if ( + kwargs["install_fonts"] + and mozinfo.info["os"] == "win" + and mozinfo.info["os_version"] == "6.1" + ): + # On Windows 7 --install-fonts fails, so fall back to a Firefox-specific codepath + self.setup_fonts_firefox() + kwargs["install_fonts"] = False + + if kwargs["preload_browser"] is None: + kwargs["preload_browser"] = False + + if kwargs["prefs_root"] is None: + kwargs["prefs_root"] = os.path.join(self.topsrcdir, "testing", "profiles") + + if kwargs["stackfix_dir"] is None: + kwargs["stackfix_dir"] = self.bindir + + kwargs = wptcommandline.check_args(kwargs) + + return kwargs + + def kwargs_wptrun(self, kwargs): + """Setup kwargs for wpt-run which is only used for non-gecko browser products""" + from tools.wpt import run, virtualenv + + kwargs = self.kwargs_common(kwargs) + + # Add additional kwargs consumed by the run frontend. Currently we don't + # have a way to set these through mach + kwargs["channel"] = None + kwargs["prompt"] = True + kwargs["install_browser"] = False + kwargs["install_webdriver"] = None + kwargs["affected"] = None + + # Install the deps + # We do this explicitly to avoid calling pip with options that aren't + # supported in the in-tree version + wptrunner_path = os.path.join(self._here, "tests", "tools", "wptrunner") + browser_cls = run.product_setup[kwargs["product"]].browser_cls + requirements = ["requirements.txt"] + if hasattr(browser_cls, "requirements"): + requirements.append(browser_cls.requirements) + + for filename in requirements: + path = os.path.join(wptrunner_path, filename) + if os.path.exists(path): + self.virtualenv_manager.install_pip_requirements( + path, require_hashes=False + ) + + venv = virtualenv.Virtualenv( + self.virtualenv_manager.virtualenv_root, skip_virtualenv_setup=True + ) + try: + kwargs = run.setup_wptrunner(venv, **kwargs) + except run.WptrunError as e: + print(e, file=sys.stderr) + sys.exit(1) + + # This is kind of a hack; override the metadata paths so we don't use + # gecko metadata for non-gecko products + for key, value in list(iteritems(kwargs["test_paths"])): + meta_suffix = key.strip("/") + meta_dir = os.path.join( + self._here, "products", kwargs["product"], meta_suffix + ) + value["metadata_path"] = meta_dir + if not os.path.exists(meta_dir): + os.makedirs(meta_dir) + return kwargs + + def setup_fonts_firefox(self): + # Ensure the Ahem font is available + if not sys.platform.startswith("darwin"): + font_path = os.path.join(os.path.dirname(self.get_binary_path()), "fonts") + else: + font_path = os.path.join( + os.path.dirname(self.get_binary_path()), + os.pardir, + "Resources", + "res", + "fonts", + ) + ahem_src = os.path.join( + self.topsrcdir, "testing", "web-platform", "tests", "fonts", "Ahem.ttf" + ) + ahem_dest = os.path.join(font_path, "Ahem.ttf") + if not os.path.exists(ahem_dest) and os.path.exists(ahem_src): + with open(ahem_src, "rb") as src, open(ahem_dest, "wb") as dest: + dest.write(src.read()) + + +class WebPlatformTestsServeRunner(MozbuildObject): + def run(self, **kwargs): + sys.path.insert( + 0, + os.path.abspath(os.path.join(os.path.dirname(__file__), "tests", "tools")), + ) + import logging + + import manifestupdate + from serve import serve + from wptrunner import wptcommandline + + logger = logging.getLogger("web-platform-tests") + + src_root = self.topsrcdir + obj_root = self.topobjdir + src_wpt_dir = os.path.join(src_root, "testing", "web-platform") + + config_path = manifestupdate.generate_config( + logger, + src_root, + src_wpt_dir, + os.path.join(obj_root, "_tests", "web-platform"), + False, + ) + + test_paths = wptcommandline.get_test_paths( + wptcommandline.config.read(config_path) + ) + + def get_route_builder(*args, **kwargs): + route_builder = serve.get_route_builder(*args, **kwargs) + + for url_base, paths in iteritems(test_paths): + if url_base != "/": + route_builder.add_mount_point(url_base, paths["tests_path"]) + + return route_builder + + return 0 if serve.run(route_builder=get_route_builder, **kwargs) else 1 + + +class WebPlatformTestsUpdater(MozbuildObject): + """Update web platform tests.""" + + def setup_logging(self, **kwargs): + import update + + return update.setup_logging(kwargs, {"mach": sys.stdout}) + + def update_manifest(self, logger, **kwargs): + import manifestupdate + + return manifestupdate.run( + logger=logger, src_root=self.topsrcdir, obj_root=self.topobjdir, **kwargs + ) + + def run_update(self, logger, **kwargs): + import update + from update import updatecommandline + + self.update_manifest(logger, **kwargs) + + if kwargs["config"] is None: + kwargs["config"] = os.path.join( + self.topobjdir, "_tests", "web-platform", "wptrunner.local.ini" + ) + if kwargs["product"] is None: + kwargs["product"] = "firefox" + + kwargs["store_state"] = False + + kwargs = updatecommandline.check_args(kwargs) + + try: + update.run_update(logger, **kwargs) + except Exception: + import traceback + + traceback.print_exc() + + +class WebPlatformTestsUnittestRunner(MozbuildObject): + def run(self, **kwargs): + import unittestrunner + + return unittestrunner.run(self.topsrcdir, **kwargs) + + +class WebPlatformTestsTestPathsRunner(MozbuildObject): + """Update web platform tests.""" + + def run(self, **kwargs): + sys.path.insert( + 0, + os.path.abspath(os.path.join(os.path.dirname(__file__), "tests", "tools")), + ) + import logging + + import manifestupdate + from manifest import testpaths + from wptrunner import wptcommandline + + logger = logging.getLogger("web-platform-tests") + + src_root = self.topsrcdir + obj_root = self.topobjdir + src_wpt_dir = os.path.join(src_root, "testing", "web-platform") + + config_path = manifestupdate.generate_config( + logger, + src_root, + src_wpt_dir, + os.path.join(obj_root, "_tests", "web-platform"), + False, + ) + + test_paths = wptcommandline.get_test_paths( + wptcommandline.config.read(config_path) + ) + results = {} + for url_base, paths in iteritems(test_paths): + if "manifest_path" not in paths: + paths["manifest_path"] = os.path.join( + paths["metadata_path"], "MANIFEST.json" + ) + results.update( + testpaths.get_paths( + path=paths["manifest_path"], + src_root=src_root, + tests_root=paths["tests_path"], + update=kwargs["update"], + rebuild=kwargs["rebuild"], + url_base=url_base, + cache_root=kwargs["cache_root"], + test_ids=kwargs["test_ids"], + ) + ) + testpaths.write_output(results, kwargs["json"]) + return True + + +class WebPlatformTestsFissionRegressionsRunner(MozbuildObject): + def run(self, **kwargs): + import fissionregressions + import mozlog + + src_root = self.topsrcdir + obj_root = self.topobjdir + logger = mozlog.structuredlog.StructuredLogger("web-platform-tests") + + try: + return fissionregressions.run(logger, src_root, obj_root, **kwargs) + except Exception: + import pdb + import traceback + + traceback.print_exc() + pdb.post_mortem() + + +def create_parser_update(): + from update import updatecommandline + + return updatecommandline.create_parser() + + +def create_parser_manifest_update(): + import manifestupdate + + return manifestupdate.create_parser() + + +def create_parser_metadata_summary(): + import metasummary + + return metasummary.create_parser() + + +def create_parser_metadata_merge(): + import metamerge + + return metamerge.get_parser() + + +def create_parser_serve(): + sys.path.insert( + 0, os.path.abspath(os.path.join(os.path.dirname(__file__), "tests", "tools")) + ) + import serve + + return serve.serve.get_parser() + + +def create_parser_unittest(): + import unittestrunner + + return unittestrunner.get_parser() + + +def create_parser_fission_regressions(): + import fissionregressions + + return fissionregressions.get_parser() + + +def create_parser_testpaths(): + import argparse + + from mach.util import get_state_dir + + parser = argparse.ArgumentParser() + parser.add_argument( + "--no-update", + dest="update", + action="store_false", + default=True, + help="Don't update manifest before continuing", + ) + parser.add_argument( + "-r", + "--rebuild", + action="store_true", + default=False, + help="Force a full rebuild of the manifest.", + ) + parser.add_argument( + "--cache-root", + action="store", + default=os.path.join(get_state_dir(), "cache", "wpt"), + help="Path in which to store any caches (default <tests_root>/.wptcache/)", + ) + parser.add_argument( + "test_ids", action="store", nargs="+", help="Test ids for which to get paths" + ) + parser.add_argument( + "--json", action="store_true", default=False, help="Output as JSON" + ) + return parser + + +@Command( + "web-platform-tests", + category="testing", + conditions=[conditions.is_firefox_or_android], + description="Run web-platform-tests.", + parser=create_parser_wpt, + virtualenv_name="wpt", +) +def run_web_platform_tests(command_context, **params): + if params["product"] is None: + if conditions.is_android(command_context): + params["product"] = "firefox_android" + else: + params["product"] = "firefox" + if "test_objects" in params: + include = [] + test_types = set() + for item in params["test_objects"]: + include.append(item["name"]) + test_types.add(item.get("subsuite")) + if None not in test_types: + params["test_types"] = list(test_types) + params["include"] = include + del params["test_objects"] + if params.get("debugger", None): + import mozdebug + + if not mozdebug.get_debugger_info(params.get("debugger")): + sys.exit(1) + + wpt_setup = command_context._spawn(WebPlatformTestsRunnerSetup) + wpt_setup._mach_context = command_context._mach_context + wpt_setup._virtualenv_name = command_context._virtualenv_name + wpt_runner = WebPlatformTestsRunner(wpt_setup) + + logger = wpt_runner.setup_logging(**params) + # wptrunner already handles setting any log parameter from + # mach test to the logger, so it's OK to remove that argument now + if "log" in params: + del params["log"] + + if ( + conditions.is_android(command_context) + and params["product"] != "firefox_android" + ): + logger.warning("Must specify --product=firefox_android in Android environment.") + + return wpt_runner.run(logger, **params) + + +@Command( + "wpt", + category="testing", + conditions=[conditions.is_firefox_or_android], + description="Run web-platform-tests.", + parser=create_parser_wpt, + virtualenv_name="wpt", +) +def run_wpt(command_context, **params): + return run_web_platform_tests(command_context, **params) + + +@Command( + "web-platform-tests-update", + category="testing", + description="Update web-platform-test metadata.", + parser=create_parser_update, + virtualenv_name="wpt", +) +def update_web_platform_tests(command_context, **params): + wpt_updater = command_context._spawn(WebPlatformTestsUpdater) + logger = wpt_updater.setup_logging(**params) + return wpt_updater.run_update(logger, **params) + + +@Command( + "wpt-update", + category="testing", + description="Update web-platform-test metadata.", + parser=create_parser_update, + virtualenv_name="wpt", +) +def update_wpt(command_context, **params): + return update_web_platform_tests(command_context, **params) + + +@Command( + "wpt-manifest-update", + category="testing", + description="Update web-platform-test manifests.", + parser=create_parser_manifest_update, + virtualenv_name="wpt", +) +def wpt_manifest_update(command_context, **params): + wpt_setup = command_context._spawn(WebPlatformTestsRunnerSetup) + wpt_runner = WebPlatformTestsRunner(wpt_setup) + logger = wpt_runner.setup_logging(**params) + logger.warning( + "The wpt manifest is now automatically updated, " + "so running this command is usually unnecessary" + ) + return 0 if wpt_runner.update_manifest(logger, **params) else 1 + + +@Command( + "wpt-serve", + category="testing", + description="Run the wpt server", + parser=create_parser_serve, + virtualenv_name="wpt", +) +def wpt_serve(command_context, **params): + import logging + + logger = logging.getLogger("web-platform-tests") + logger.addHandler(logging.StreamHandler(sys.stdout)) + wpt_serve = command_context._spawn(WebPlatformTestsServeRunner) + return wpt_serve.run(**params) + + +@Command( + "wpt-metadata-summary", + category="testing", + description="Create a json summary of the wpt metadata", + parser=create_parser_metadata_summary, + virtualenv_name="wpt", +) +def wpt_summary(command_context, **params): + import metasummary + + wpt_setup = command_context._spawn(WebPlatformTestsRunnerSetup) + return metasummary.run(wpt_setup.topsrcdir, wpt_setup.topobjdir, **params) + + +@Command( + "wpt-metadata-merge", + category="testing", + parser=create_parser_metadata_merge, + virtualenv_name="wpt", +) +def wpt_meta_merge(command_context, **params): + import metamerge + + if params["dest"] is None: + params["dest"] = params["current"] + return metamerge.run(**params) + + +@Command( + "wpt-unittest", + category="testing", + description="Run the wpt tools and wptrunner unit tests", + parser=create_parser_unittest, + virtualenv_name="wpt", +) +def wpt_unittest(command_context, **params): + runner = command_context._spawn(WebPlatformTestsUnittestRunner) + return 0 if runner.run(**params) else 1 + + +@Command( + "wpt-test-paths", + category="testing", + description="Get a mapping from test ids to files", + parser=create_parser_testpaths, + virtualenv_name="wpt", +) +def wpt_test_paths(command_context, **params): + runner = command_context._spawn(WebPlatformTestsTestPathsRunner) + runner.run(**params) + return 0 + + +@Command( + "wpt-fission-regressions", + category="testing", + description="Dump a list of fission-specific regressions", + parser=create_parser_fission_regressions, + virtualenv_name="wpt", +) +def wpt_fission_regressions(command_context, **params): + runner = command_context._spawn(WebPlatformTestsFissionRegressionsRunner) + runner.run(**params) + return 0 |