Files
tubestation/testing/firefox-ui/mach_commands.py
Alex Hochheiden 0bbbbc094d Bug 1959849 - [lint] Remove six usages via auto fix on ruff==0.0.237 via pupgrade r=linter-reviewers,webdriver-reviewers,ahal,whimboo
This was not done through `./mach lint`, `ruff` version `0.0.237` was
called directly on the root directory with a modified `pyproject.toml`
that only enabled the `UP016` rule with `--fix`.

Some manual corrections were made after the auto fix ran to resolve
incorrect behavior.

No manual removals of `six` usages was done in this revision.

Differential Revision: https://phabricator.services.mozilla.com/D245270
2025-04-13 17:48:23 +00:00

91 lines
2.7 KiB
Python

# 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
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 kwargs.items():
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)