summaryrefslogtreecommitdiffstats
path: root/testing/firefox-ui/mach_commands.py
diff options
context:
space:
mode:
Diffstat (limited to 'testing/firefox-ui/mach_commands.py')
-rw-r--r--testing/firefox-ui/mach_commands.py91
1 files changed, 91 insertions, 0 deletions
diff --git a/testing/firefox-ui/mach_commands.py b/testing/firefox-ui/mach_commands.py
new file mode 100644
index 0000000000..4c0d4a126f
--- /dev/null
+++ b/testing/firefox-ui/mach_commands.py
@@ -0,0 +1,91 @@
+# 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 logging
+import os
+import sys
+
+import six
+from mach.decorators import Command
+from mozbuild.base import BinaryNotFoundException
+from mozbuild.base import MachCommandConditions as conditions
+
+
+def setup_argument_parser_functional():
+ from firefox_ui_harness.arguments.base import FirefoxUIArguments
+ from mozlog.structured import commandline
+
+ parser = FirefoxUIArguments()
+ commandline.add_logging_group(parser)
+ return parser
+
+
+def run_firefox_ui_test(topsrcdir=None, **kwargs):
+ from argparse import Namespace
+
+ import firefox_ui_harness
+ from mozlog.structured import commandline
+
+ parser = setup_argument_parser_functional()
+
+ fxui_dir = os.path.join(topsrcdir, "testing", "firefox-ui")
+
+ # Set the resources path which is used to serve test data via wptserve
+ if not kwargs["server_root"]:
+ kwargs["server_root"] = os.path.join(fxui_dir, "resources")
+
+ # If called via "mach test" a dictionary of tests is passed in
+ if "test_objects" in kwargs:
+ tests = []
+ for obj in kwargs["test_objects"]:
+ tests.append(obj["file_relpath"])
+ kwargs["tests"] = tests
+ elif not kwargs.get("tests"):
+ # If no tests have been selected, set default ones
+ kwargs["tests"] = os.path.join(fxui_dir, "tests", "functional", "manifest.ini")
+
+ kwargs["logger"] = kwargs.pop("log", None)
+ if not kwargs["logger"]:
+ kwargs["logger"] = commandline.setup_logging(
+ "Firefox UI - Functional Tests", {"mach": sys.stdout}
+ )
+
+ args = Namespace()
+
+ for k, v in six.iteritems(kwargs):
+ setattr(args, k, v)
+
+ parser.verify_usage(args)
+
+ failed = firefox_ui_harness.cli_functional.cli(args=vars(args))
+
+ if failed > 0:
+ return 1
+ else:
+ return 0
+
+
+@Command(
+ "firefox-ui-functional",
+ category="testing",
+ conditions=[conditions.is_firefox],
+ description="Run the functional test suite of Firefox UI tests.",
+ parser=setup_argument_parser_functional,
+)
+def run_firefox_ui_functional(command_context, **kwargs):
+ try:
+ kwargs["binary"] = kwargs["binary"] or command_context.get_binary_path("app")
+ except BinaryNotFoundException as e:
+ command_context.log(
+ logging.ERROR,
+ "firefox-ui-functional",
+ {"error": str(e)},
+ "ERROR: {error}",
+ )
+ command_context.log(
+ logging.INFO, "firefox-ui-functional", {"help": e.help()}, "{help}"
+ )
+ return 1
+
+ return run_firefox_ui_test(topsrcdir=command_context.topsrcdir, **kwargs)