summaryrefslogtreecommitdiffstats
path: root/testing/marionette/mach_test_package_commands.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
commit36d22d82aa202bb199967e9512281e9a53db42c9 (patch)
tree105e8c98ddea1c1e4784a60a5a6410fa416be2de /testing/marionette/mach_test_package_commands.py
parentInitial commit. (diff)
downloadfirefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz
firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip
Adding upstream version 115.7.0esr.upstream/115.7.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'testing/marionette/mach_test_package_commands.py')
-rw-r--r--testing/marionette/mach_test_package_commands.py66
1 files changed, 66 insertions, 0 deletions
diff --git a/testing/marionette/mach_test_package_commands.py b/testing/marionette/mach_test_package_commands.py
new file mode 100644
index 0000000000..22c298a88a
--- /dev/null
+++ b/testing/marionette/mach_test_package_commands.py
@@ -0,0 +1,66 @@
+# 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 argparse
+import os
+import sys
+from functools import partial
+
+from mach.decorators import Command
+
+parser = None
+
+
+def run_marionette(context, **kwargs):
+ from marionette.runtests import MarionetteHarness, MarionetteTestRunner
+ from mozlog.structured import commandline
+
+ args = argparse.Namespace(**kwargs)
+ args.binary = args.binary or context.firefox_bin
+
+ test_root = os.path.join(context.package_root, "marionette", "tests")
+ if not args.tests:
+ args.tests = [
+ os.path.join(
+ test_root,
+ "testing",
+ "marionette",
+ "harness",
+ "marionette_harness",
+ "tests",
+ "unit-tests.ini",
+ )
+ ]
+
+ normalize = partial(context.normalize_test_path, test_root)
+ args.tests = list(map(normalize, args.tests))
+
+ commandline.add_logging_group(parser)
+ parser.verify_usage(args)
+
+ args.logger = commandline.setup_logging(
+ "Marionette Unit Tests", args, {"mach": sys.stdout}
+ )
+ status = MarionetteHarness(MarionetteTestRunner, args=vars(args)).run()
+ return 1 if status else 0
+
+
+def setup_marionette_argument_parser():
+ from marionette.runner.base import BaseMarionetteArguments
+
+ global parser
+ parser = BaseMarionetteArguments()
+ return parser
+
+
+@Command(
+ "marionette-test",
+ category="testing",
+ description="Run a Marionette test (Check UI or the internal JavaScript "
+ "using marionette).",
+ parser=setup_marionette_argument_parser,
+)
+def run_marionette_test(command_context, **kwargs):
+ command_context.context.activate_mozharness_venv()
+ return run_marionette(command_context.context, **kwargs)