Adding upstream version 110.0.1.upstream/110.0.1upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 219 insertions, 0 deletions

diff --git a/testing/mochitest/mach_test_package_commands.py b/testing/mochitest/mach_test_package_commands.py
new file mode 100644
index 0000000000..b5c4d5cdbe
--- /dev/null
+++ b/testing/mochitest/mach_test_package_commands.py
@@ -0,0 +1,219 @@
+# 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/.
+
+import json
+import os
+import sys
+from argparse import Namespace
+from functools import partial
+
+import six
+from mach.decorators import Command
+
+here = os.path.abspath(os.path.dirname(__file__))
+parser = None
+logger = None
+
+
+def run_test(context, is_junit, **kwargs):
+    from mochitest_options import ALL_FLAVORS
+    from mozlog.commandline import setup_logging
+
+    if not kwargs.get("log"):
+        kwargs["log"] = setup_logging("mochitest", kwargs, {"mach": sys.stdout})
+    global logger
+    logger = kwargs["log"]
+
+    flavor = kwargs.get("flavor") or "mochitest"
+    if flavor not in ALL_FLAVORS:
+        for fname, fobj in six.iteritems(ALL_FLAVORS):
+            if flavor in fobj["aliases"]:
+                flavor = fname
+                break
+    fobj = ALL_FLAVORS[flavor]
+    kwargs.update(fobj.get("extra_args", {}))
+
+    if kwargs.get("allow_software_gl_layers"):
+        os.environ["MOZ_LAYERS_ALLOW_SOFTWARE_GL"] = "1"
+        del kwargs["allow_software_gl_layers"]
+    if kwargs.get("mochitest_suite"):
+        suite = kwargs["mochitest_suite"]
+        del kwargs["mochitest_suite"]
+    elif kwargs.get("test_suite"):
+        suite = kwargs["test_suite"]
+        del kwargs["test_suite"]
+    if kwargs.get("no_run_tests"):
+        del kwargs["no_run_tests"]
+
+    args = Namespace(**kwargs)
+    args.e10s = context.mozharness_config.get("e10s", args.e10s)
+    args.certPath = context.certs_dir
+
+    if is_junit:
+        return run_geckoview_junit(context, args)
+
+    # subsuite mapping from mozharness configs
+    subsuites = {
+        # --mochitest-suite -> --subsuite
+        "mochitest-chrome-gpu": "gpu",
+        "mochitest-plain-gpu": "gpu",
+        "mochitest-media": "media",
+        "mochitest-browser-chrome-screenshots": "screenshots",
+        "mochitest-webgl1-core": "webgl1-core",
+        "mochitest-webgl1-ext": "webgl1-ext",
+        "mochitest-webgl2-core": "webgl2-core",
+        "mochitest-webgl2-ext": "webgl2-ext",
+        "mochitest-webgl2-deqp": "webgl2-deqp",
+        "mochitest-webgpu": "webgpu",
+        "mochitest-devtools-chrome": "devtools",
+        "mochitest-browser-a11y": "a11y",
+        "mochitest-remote": "remote",
+        "mochitest-browser-media": "media-bc",
+    }
+    args.subsuite = subsuites.get(suite)
+    if args.subsuite == "devtools":
+        args.flavor = "browser"
+    if args.subsuite == "a11y":
+        args.flavor = "browser"
+    if args.subsuite == "media-bc":
+        args.flavor = "browser"
+
+    if not args.test_paths:
+        mh_test_paths = json.loads(os.environ.get("MOZHARNESS_TEST_PATHS", '""'))
+        if mh_test_paths:
+            logger.info("Found MOZHARNESS_TEST_PATHS:")
+            logger.info(str(mh_test_paths))
+            args.test_paths = []
+            for k in mh_test_paths:
+                args.test_paths.extend(mh_test_paths[k])
+    if args.test_paths:
+        install_subdir = fobj.get("install_subdir", fobj["suite"])
+        test_root = os.path.join(context.package_root, "mochitest", install_subdir)
+        normalize = partial(context.normalize_test_path, test_root)
+        # pylint --py3k: W1636
+        args.test_paths = list(map(normalize, args.test_paths))
+
+    import mozinfo
+
+    if mozinfo.info.get("buildapp") == "mobile/android":
+        return run_mochitest_android(context, args)
+    return run_mochitest_desktop(context, args)
+
+
+def run_mochitest_desktop(context, args):
+    args.app = args.app or context.firefox_bin
+    args.utilityPath = context.bin_dir
+    args.extraProfileFiles.append(os.path.join(context.bin_dir, "plugins"))
+    args.extraPrefs.append("webgl.force-enabled=true")
+    args.quiet = True
+    args.useTestMediaDevices = True
+    args.screenshotOnFail = True
+    args.cleanupCrashes = True
+    args.marionette_startup_timeout = "180"
+    args.sandboxReadWhitelist.append(context.mozharness_workdir)
+    if args.flavor == "browser":
+        args.chunkByRuntime = True
+    else:
+        args.chunkByDir = 4
+
+    from runtests import run_test_harness
+
+    logger.info("mach calling runtests with args: " + str(args))
+    return run_test_harness(parser, args)
+
+
+def set_android_args(context, args):
+    args.app = args.app or "org.mozilla.geckoview.test_runner"
+    args.utilityPath = context.hostutils
+    args.xrePath = context.hostutils
+    config = context.mozharness_config
+    if config:
+        host = os.environ.get("HOST_IP", "10.0.2.2")
+        args.remoteWebServer = config.get("remote_webserver", host)
+        args.httpPort = config.get("http_port", 8854)
+        args.sslPort = config.get("ssl_port", 4454)
+        args.adbPath = config["exes"]["adb"] % {
+            "abs_work_dir": context.mozharness_workdir
+        }
+        args.deviceSerial = os.environ.get("DEVICE_SERIAL", "emulator-5554")
+    return args
+
+
+def run_mochitest_android(context, args):
+    args = set_android_args(context, args)
+    args.extraProfileFiles.append(
+        os.path.join(context.package_root, "mochitest", "fonts")
+    )
+
+    from runtestsremote import run_test_harness
+
+    logger.info("mach calling runtestsremote with args: " + str(args))
+    return run_test_harness(parser, args)
+
+
+def run_geckoview_junit(context, args):
+    args = set_android_args(context, args)
+
+    from runjunit import run_test_harness
+
+    # Force fission disabled by default for android
+    args["disable_fission"] = True
+
+    logger.info("mach calling runjunit with args: " + str(args))
+    return run_test_harness(parser, args)
+
+
+def add_global_arguments(parser):
+    parser.add_argument("--test-suite")
+    parser.add_argument("--mochitest-suite")
+    parser.add_argument("--download-symbols")
+    parser.add_argument("--allow-software-gl-layers", action="store_true")
+    parser.add_argument("--no-run-tests", action="store_true")
+
+
+def setup_mochitest_argument_parser():
+    import mozinfo
+
+    mozinfo.find_and_update_from_json(here)
+    app = "generic"
+    if mozinfo.info.get("buildapp") == "mobile/android":
+        app = "android"
+
+    from mochitest_options import MochitestArgumentParser
+
+    global parser
+    parser = MochitestArgumentParser(app=app)
+    add_global_arguments(parser)
+    return parser
+
+
+def setup_junit_argument_parser():
+    from runjunit import JunitArgumentParser
+
+    global parser
+    parser = JunitArgumentParser()
+    add_global_arguments(parser)
+    return parser
+
+
+@Command(
+    "mochitest",
+    category="testing",
+    description="Run the mochitest harness.",
+    parser=setup_mochitest_argument_parser,
+)
+def mochitest(command_context, **kwargs):
+    command_context._mach_context.activate_mozharness_venv()
+    return run_test(command_context._mach_context, False, **kwargs)
+
+
+@Command(
+    "geckoview-junit",
+    category="testing",
+    description="Run the geckoview-junit harness.",
+    parser=setup_junit_argument_parser,
+)
+def geckoview_junit(command_context, **kwargs):
+    command_context._mach_context.activate_mozharness_venv()
+    return run_test(command_context._mach_context, True, **kwargs)