Source code
Revision control
Copy as Markdown
Other Tools
# 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
import argparse
import logging
import os
import subprocess
import sys
import requests
from mach.decorators import Command, CommandArgument, SubCommand
from mozbuild.base import BuildEnvironmentNotFoundException
from mozbuild.base import MachCommandConditions as conditions
UNKNOWN_TEST = """
I was unable to find tests from the given argument(s).
You should specify a test directory, filename, test suite name, or
abbreviation.
It's possible my little brain doesn't know about the type of test you are
trying to execute. If you suspect this, please request support by filing
a bug at
""".strip()
UNKNOWN_FLAVOR = """
I know you are trying to run a %s%s test. Unfortunately, I can't run those
tests yet. Sorry!
""".strip()
TEST_HELP = """
Test or tests to run. Tests can be specified by filename, directory, suite
name or suite alias.
The following test suites and aliases are supported: {}
""".strip()
def get_test_parser():
from mozlog.commandline import add_logging_group
from moztest.resolve import TEST_SUITES
parser = argparse.ArgumentParser()
parser.add_argument(
"what",
default=None,
nargs="+",
help=TEST_HELP.format(", ".join(sorted(TEST_SUITES))),
)
parser.add_argument(
"extra_args",
default=None,
nargs=argparse.REMAINDER,
help="Extra arguments to pass to the underlying test command(s). "
"If an underlying command doesn't recognize the argument, it "
"will fail.",
)
parser.add_argument(
"--debugger",
default=None,
action="store",
nargs="?",
help="Specify a debugger to use.",
)
add_logging_group(parser)
return parser
ADD_TEST_SUPPORTED_SUITES = [
"mochitest-chrome",
"mochitest-plain",
"mochitest-browser-chrome",
"web-platform-tests-privatebrowsing",
"web-platform-tests-testharness",
"web-platform-tests-reftest",
"xpcshell",
]
ADD_TEST_SUPPORTED_DOCS = ["js", "html", "xhtml", "xul"]
SUITE_SYNONYMS = {
"wpt": "web-platform-tests-testharness",
"wpt-privatebrowsing": "web-platform-tests-privatebrowsing",
"wpt-testharness": "web-platform-tests-testharness",
"wpt-reftest": "web-platform-tests-reftest",
}
MISSING_ARG = object()
def create_parser_addtest():
import addtest
parser = argparse.ArgumentParser()
parser.add_argument(
"--suite",
choices=sorted(ADD_TEST_SUPPORTED_SUITES + list(SUITE_SYNONYMS.keys())),
help="suite for the test. "
"If you pass a `test` argument this will be determined "
"based on the filename and the folder it is in",
)
parser.add_argument(
"-o",
"--overwrite",
action="store_true",
help="Overwrite an existing file if it exists.",
)
parser.add_argument(
"--doc",
choices=ADD_TEST_SUPPORTED_DOCS,
help="Document type for the test (if applicable)."
"If you pass a `test` argument this will be determined "
"based on the filename.",
)
parser.add_argument(
"-e",
"--editor",
action="store",
nargs="?",
default=MISSING_ARG,
help="Open the created file(s) in an editor; if a "
"binary is supplied it will be used otherwise the default editor for "
"your environment will be opened",
)
for base_suite in addtest.TEST_CREATORS:
cls = addtest.TEST_CREATORS[base_suite]
if hasattr(cls, "get_parser"):
group = parser.add_argument_group(base_suite)
cls.get_parser(group)
parser.add_argument("test", nargs="?", help=("Test to create."))
return parser
@Command(
"addtest",
category="testing",
description="Generate tests based on templates",
parser=create_parser_addtest,
)
def addtest(
command_context,
suite=None,
test=None,
doc=None,
overwrite=False,
editor=MISSING_ARG,
**kwargs,
):
import io
import addtest
from moztest.resolve import TEST_SUITES
if not suite and not test:
return create_parser_addtest().parse_args(["--help"])
if suite in SUITE_SYNONYMS:
suite = SUITE_SYNONYMS[suite]
if test:
if not overwrite and os.path.isfile(os.path.abspath(test)):
print("Error: can't generate a test that already exists:", test)
return 1
abs_test = os.path.abspath(test)
if doc is None:
doc = guess_doc(abs_test)
if suite is None:
guessed_suite, err = guess_suite(abs_test)
if err:
print(err)
return 1
suite = guessed_suite
else:
test = None
if doc is None:
doc = "html"
if not suite:
print(
"We couldn't automatically determine a suite. "
"Please specify `--suite` with one of the following options:\n{}\n"
"If you'd like to add support to a new suite, please file a bug "
ADD_TEST_SUPPORTED_SUITES
)
)
return 1
if doc not in ADD_TEST_SUPPORTED_DOCS:
print(
"Error: invalid `doc`. Either pass in a test with a valid extension"
"({}) or pass in the `doc` argument".format(ADD_TEST_SUPPORTED_DOCS)
)
return 1
creator_cls = addtest.creator_for_suite(suite)
if creator_cls is None:
print("Sorry, `addtest` doesn't currently know how to add {}".format(suite))
return 1
creator = creator_cls(command_context.topsrcdir, test, suite, doc, **kwargs)
creator.check_args()
paths = []
added_tests = False
for path, template in creator:
if not template:
continue
added_tests = True
if path:
paths.append(path)
print("Adding a test file at {} (suite `{}`)".format(path, suite))
try:
os.makedirs(os.path.dirname(path))
except OSError:
pass
with io.open(path, "w", newline="\n") as f:
f.write(template)
else:
# write to stdout if you passed only suite and doc and not a file path
print(template)
if not added_tests:
return 1
if test:
creator.update_manifest()
# Small hack, should really do this better
if suite.startswith("wpt-"):
suite = "web-platform-tests"
mach_command = TEST_SUITES[suite]["mach_command"]
print(
"Please make sure to add the new test to your commit. "
"You can now run the test with:\n ./mach {} {}".format(
mach_command, test
)
)
if editor is not MISSING_ARG:
if editor is not None:
editor = editor
elif "VISUAL" in os.environ:
editor = os.environ["VISUAL"]
elif "EDITOR" in os.environ:
editor = os.environ["EDITOR"]
else:
print("Unable to determine editor; please specify a binary")
editor = None
proc = None
if editor:
import subprocess
proc = subprocess.Popen("%s %s" % (editor, " ".join(paths)), shell=True)
if proc:
proc.wait()
return 0
def guess_doc(abs_test):
filename = os.path.basename(abs_test)
return os.path.splitext(filename)[1].strip(".")
def guess_suite(abs_test):
# If you pass a abs_test, try to detect the type based on the name
# and folder. This detection can be skipped if you pass the `type` arg.
err = None
guessed_suite = None
parent = os.path.dirname(abs_test)
filename = os.path.basename(abs_test)
has_browser_ini = os.path.isfile(os.path.join(parent, "browser.ini"))
has_browser_toml = os.path.isfile(os.path.join(parent, "browser.toml"))
has_chrome_ini = os.path.isfile(os.path.join(parent, "chrome.ini"))
has_chrome_toml = os.path.isfile(os.path.join(parent, "chrome.toml"))
has_plain_ini = os.path.isfile(os.path.join(parent, "mochitest.ini"))
has_plain_toml = os.path.isfile(os.path.join(parent, "mochitest.toml"))
has_xpcshell_ini = os.path.isfile(os.path.join(parent, "xpcshell.ini"))
has_xpcshell_toml = os.path.isfile(os.path.join(parent, "xpcshell.toml"))
in_wpt_folder = abs_test.startswith(
os.path.abspath(os.path.join("testing", "web-platform"))
)
if in_wpt_folder:
guessed_suite = "web-platform-tests-testharness"
if "/css/" in abs_test:
guessed_suite = "web-platform-tests-reftest"
elif (
filename.startswith("test_")
and (has_xpcshell_ini or has_xpcshell_toml)
and guess_doc(abs_test) == "js"
):
guessed_suite = "xpcshell"
else:
if filename.startswith("browser_") and (has_browser_ini or has_browser_toml):
guessed_suite = "mochitest-browser-chrome"
elif filename.startswith("test_"):
if (has_chrome_ini or has_chrome_toml) and (
has_plain_ini or has_plain_toml
):
err = (
"Error: directory contains both a chrome.{ini|toml} and mochitest.{ini|toml}. "
"Please set --suite=mochitest-chrome or --suite=mochitest-plain."
)
elif has_chrome_ini or has_chrome_toml:
guessed_suite = "mochitest-chrome"
elif has_plain_ini or has_plain_toml:
guessed_suite = "mochitest-plain"
return guessed_suite, err
class MachTestRunner:
"""Adapter for mach test to simplify it's import externally."""
def test(command_context, what, extra_args, **log_args):
return test(command_context, what, extra_args, **log_args)
@Command(
"test",
category="testing",
description="Run tests (detects the kind of test and runs it).",
parser=get_test_parser,
)
def test(command_context, what, extra_args, **log_args):
"""Run tests from names or paths.
mach test accepts arguments specifying which tests to run. Each argument
can be:
* The path to a test file
* A directory containing tests
* A test suite name
* An alias to a test suite name (codes used on TreeHerder)
* path to a test manifest
When paths or directories are given, they are first resolved to test
files known to the build system.
If resolved tests belong to more than one test type/flavor/harness,
the harness for each relevant type/flavor will be invoked. e.g. if
you specify a directory with xpcshell and browser chrome mochitests,
both harnesses will be invoked.
Warning: `mach test` does not automatically re-build.
Please remember to run `mach build` when necessary.
EXAMPLES
Run all test files in the devtools/client/shared/redux/middleware/xpcshell/
directory:
`./mach test devtools/client/shared/redux/middleware/xpcshell/`
The below command prints a short summary of results instead of
the default more verbose output.
Do not forget the - (minus sign) after --log-grouped!
`./mach test --log-grouped - devtools/client/shared/redux/middleware/xpcshell/`
To learn more about arguments for each test type/flavor/harness, please run
`./mach <test-harness> --help`. For example, `./mach mochitest --help`.
"""
from mozlog.commandline import setup_logging
from mozlog.handlers import StreamHandler
from moztest.resolve import TEST_SUITES, TestResolver, get_suite_definition
resolver = command_context._spawn(TestResolver)
run_suites, run_tests = resolver.resolve_metadata(what)
if not run_suites and not run_tests:
print(UNKNOWN_TEST)
return 1
if log_args.get("debugger", None):
import mozdebug
if not mozdebug.get_debugger_info(log_args.get("debugger")):
sys.exit(1)
extra_args_debugger_notation = "=".join(
["--debugger", log_args.get("debugger")]
)
if extra_args:
extra_args.append(extra_args_debugger_notation)
else:
extra_args = [extra_args_debugger_notation]
# Create shared logger
format_args = {"level": command_context._mach_context.settings["test"]["level"]}
if not run_suites and len(run_tests) == 1:
format_args["verbose"] = True
format_args["compact"] = False
default_format = command_context._mach_context.settings["test"]["format"]
log = setup_logging(
"mach-test", log_args, {default_format: sys.stdout}, format_args
)
for handler in log.handlers:
if isinstance(handler, StreamHandler):
handler.formatter.inner.summary_on_shutdown = True
if log_args.get("custom_handler", None) is not None:
log.add_handler(log_args.get("custom_handler"))
status = None
for suite_name in run_suites:
suite = TEST_SUITES[suite_name]
kwargs = suite["kwargs"]
kwargs["log"] = log
kwargs.setdefault("subsuite", None)
if "mach_command" in suite:
res = command_context._mach_context.commands.dispatch(
suite["mach_command"],
command_context._mach_context,
argv=extra_args,
**kwargs,
)
if res:
status = res
buckets = {}
for test in run_tests:
key = (test["flavor"], test.get("subsuite", ""))
buckets.setdefault(key, []).append(test)
for (flavor, subsuite), tests in sorted(buckets.items()):
_, m = get_suite_definition(flavor, subsuite)
if "mach_command" not in m:
substr = "-{}".format(subsuite) if subsuite else ""
print(UNKNOWN_FLAVOR % (flavor, substr))
status = 1
continue
kwargs = dict(m["kwargs"])
kwargs["log"] = log
kwargs.setdefault("subsuite", None)
res = command_context._mach_context.commands.dispatch(
m["mach_command"],
command_context._mach_context,
argv=extra_args,
test_objects=tests,
**kwargs,
)
if res:
status = res
if not log.has_shutdown:
log.shutdown()
return status
@Command(
"cppunittest", category="testing", description="Run cpp unit tests (C++ tests)."
)
@CommandArgument(
"test_files",
nargs="*",
metavar="N",
help="Test to run. Can be specified as one or more files or "
"directories, or omitted. If omitted, the entire test suite is "
"executed.",
)
def run_cppunit_test(command_context, **params):
from mozlog import commandline
log = params.get("log")
if not log:
log = commandline.setup_logging("cppunittest", {}, {"tbpl": sys.stdout})
# See if we have crash symbols
symbols_path = os.path.join(command_context.distdir, "crashreporter-symbols")
if not os.path.isdir(symbols_path):
symbols_path = None
# If no tests specified, run all tests in main manifest
tests = params["test_files"]
if not tests:
tests = [os.path.join(command_context.distdir, "cppunittests")]
manifest_path = os.path.join(
command_context.topsrcdir, "testing", "cppunittest.toml"
)
else:
manifest_path = None
utility_path = command_context.bindir
if conditions.is_android(command_context):
from mozrunner.devices.android_device import (
InstallIntent,
verify_android_device,
)
verify_android_device(command_context, install=InstallIntent.NO)
return run_android_test(tests, symbols_path, manifest_path, log)
return run_desktop_test(
command_context, tests, symbols_path, manifest_path, utility_path, log
)
def run_desktop_test(
command_context, tests, symbols_path, manifest_path, utility_path, log
):
import runcppunittests as cppunittests
from mozlog import commandline
parser = cppunittests.CPPUnittestOptions()
commandline.add_logging_group(parser)
options, args = parser.parse_args()
options.symbols_path = symbols_path
options.manifest_path = manifest_path
options.utility_path = utility_path
options.xre_path = command_context.bindir
try:
result = cppunittests.run_test_harness(options, tests)
except Exception as e:
log.error("Caught exception running cpp unit tests: %s" % str(e))
result = False
raise
return 0 if result else 1
def run_android_test(command_context, tests, symbols_path, manifest_path, log):
import remotecppunittests
from mozlog import commandline
parser = remotecppunittests.RemoteCPPUnittestOptions()
commandline.add_logging_group(parser)
options, args = parser.parse_args()
if not options.adb_path:
from mozrunner.devices.android_device import get_adb_path
options.adb_path = get_adb_path(command_context)
options.symbols_path = symbols_path
options.manifest_path = manifest_path
options.xre_path = command_context.bindir
options.local_lib = command_context.bindir.replace("bin", "fennec")
for file in os.listdir(os.path.join(command_context.topobjdir, "dist")):
if file.endswith(".apk") and file.startswith("fennec"):
options.local_apk = os.path.join(command_context.topobjdir, "dist", file)
log.info("using APK: " + options.local_apk)
break
try:
result = remotecppunittests.run_test_harness(options, tests)
except Exception as e:
log.error("Caught exception running cpp unit tests: %s" % str(e))
result = False
raise
return 0 if result else 1
def executable_name(name):
return name + ".exe" if sys.platform.startswith("win") else name
@Command(
"jstests",
category="testing",
description="Run SpiderMonkey JS tests in the JS shell.",
ok_if_tests_disabled=True,
)
@CommandArgument("--shell", help="The shell to be used")
@CommandArgument(
"params",
nargs=argparse.REMAINDER,
help="Extra arguments to pass down to the test harness.",
)
def run_jstests(command_context, shell, params):
import subprocess
command_context.virtualenv_manager.ensure()
python = command_context.virtualenv_manager.python_path
js = shell or os.path.join(command_context.bindir, executable_name("js"))
jstest_cmd = [
python,
os.path.join(command_context.topsrcdir, "js", "src", "tests", "jstests.py"),
js,
] + params
return subprocess.call(jstest_cmd)
@Command(
"jit-test",
category="testing",
description="Run SpiderMonkey jit-tests in the JS shell.",
ok_if_tests_disabled=True,
)
@CommandArgument("--shell", help="The shell to be used")
@CommandArgument(
"--cgc",
action="store_true",
default=False,
help="Run with the SM(cgc) job's env vars",
)
@CommandArgument(
"params",
nargs=argparse.REMAINDER,
help="Extra arguments to pass down to the test harness.",
)
def run_jittests(command_context, shell, cgc, params):
import subprocess
command_context.virtualenv_manager.ensure()
python = command_context.virtualenv_manager.python_path
js = shell or os.path.join(command_context.bindir, executable_name("js"))
jittest_cmd = [
python,
os.path.join(command_context.topsrcdir, "js", "src", "jit-test", "jit_test.py"),
js,
] + params
env = os.environ.copy()
if cgc:
env["JS_GC_ZEAL"] = "IncrementalMultipleSlices"
return subprocess.call(jittest_cmd, env=env)
@Command("jsapi-tests", category="testing", description="Run SpiderMonkey JSAPI tests.")
@CommandArgument(
"--list",
action="store_true",
default=False,
help="List all tests",
)
@CommandArgument(
"--frontend-only",
action="store_true",
default=False,
help="Run tests for frontend-only APIs, with light-weight entry point",
)
@CommandArgument(
"test_name",
nargs="?",
metavar="N",
help="Test to run. Can be a prefix or omitted. If "
"omitted, the entire test suite is executed.",
)
def run_jsapitests(command_context, list=False, frontend_only=False, test_name=None):
import subprocess
jsapi_tests_cmd = [
os.path.join(command_context.bindir, executable_name("jsapi-tests"))
]
if list:
jsapi_tests_cmd.append("--list")
if frontend_only:
jsapi_tests_cmd.append("--frontend-only")
if test_name:
jsapi_tests_cmd.append(test_name)
test_env = os.environ.copy()
test_env["TOPSRCDIR"] = command_context.topsrcdir
result = subprocess.call(jsapi_tests_cmd, env=test_env)
if result != 0:
print(f"jsapi-tests failed, exit code {result}")
return result
def run_check_js_msg(command_context):
import subprocess
command_context.virtualenv_manager.ensure()
python = command_context.virtualenv_manager.python_path
check_cmd = [
python,
os.path.join(command_context.topsrcdir, "config", "check_js_msg_encoding.py"),
]
return subprocess.call(check_cmd)
def get_jsshell_parser():
from jsshell.benchmark import get_parser
return get_parser()
@Command(
"jsshell-bench",
category="testing",
parser=get_jsshell_parser,
description="Run benchmarks in the SpiderMonkey JS shell.",
)
def run_jsshelltests(command_context, **kwargs):
from jsshell import benchmark
return benchmark.run(**kwargs)
@Command(
"cramtest",
category="testing",
description="Mercurial style .t tests for command line applications.",
)
@CommandArgument(
"test_paths",
nargs="*",
metavar="N",
help="Test paths to run. Each path can be a test file or directory. "
"If omitted, the entire suite will be run.",
)
@CommandArgument(
"cram_args",
nargs=argparse.REMAINDER,
help="Extra arguments to pass down to the cram binary. See "
"'./mach python -m cram -- -h' for a list of available options.",
)
def cramtest(command_context, cram_args=None, test_paths=None, test_objects=None):
command_context.activate_virtualenv()
import mozinfo
from manifestparser import TestManifest
if test_objects is None:
from moztest.resolve import TestResolver
resolver = command_context._spawn(TestResolver)
if test_paths:
# If we were given test paths, try to find tests matching them.
test_objects = resolver.resolve_tests(paths=test_paths, flavor="cram")
else:
# Otherwise just run everything in CRAMTEST_MANIFESTS
test_objects = resolver.resolve_tests(flavor="cram")
if not test_objects:
message = "No tests were collected, check spelling of the test paths."
command_context.log(logging.WARN, "cramtest", {}, message)
return 1
mp = TestManifest()
mp.tests.extend(test_objects)
tests = mp.active_tests(disabled=False, **mozinfo.info)
python = command_context.virtualenv_manager.python_path
cmd = [python, "-m", "cram"] + cram_args + [t["relpath"] for t in tests]
return subprocess.call(cmd, cwd=command_context.topsrcdir)
from datetime import date, timedelta
@Command(
"test-info", category="testing", description="Display historical test results."
)
def test_info(command_context):
"""
All functions implemented as subcommands.
"""
@SubCommand(
"test-info",
"tests",
description="Display historical test result summary for named tests.",
)
@CommandArgument("test_names", nargs=argparse.REMAINDER, help="Test(s) of interest.")
@CommandArgument(
"--start",
default=(date.today() - timedelta(7)).strftime("%Y-%m-%d"),
help="Start date (YYYY-MM-DD)",
)
@CommandArgument(
"--end", default=date.today().strftime("%Y-%m-%d"), help="End date (YYYY-MM-DD)"
)
@CommandArgument(
"--show-info",
action="store_true",
help="Retrieve and display general test information.",
)
@CommandArgument(
"--show-bugs",
action="store_true",
help="Retrieve and display related Bugzilla bugs.",
)
@CommandArgument("--verbose", action="store_true", help="Enable debug logging.")
def test_info_tests(
command_context,
test_names,
start,
end,
show_info,
show_bugs,
verbose,
):
import testinfo
ti = testinfo.TestInfoTests(verbose)
ti.report(
test_names,
start,
end,
show_info,
show_bugs,
)
@SubCommand(
"test-info",
"report",
description="Generate a json report of test manifests and/or tests "
"categorized by Bugzilla component and optionally filtered "
"by path, component, and/or manifest annotations.",
)
@CommandArgument(
"--components",
default=None,
help="Comma-separated list of Bugzilla components."
" eg. Testing::General,Core::WebVR",
)
@CommandArgument(
"--flavor",
help='Limit results to tests of the specified flavor (eg. "xpcshell").',
)
@CommandArgument(
"--subsuite",
help='Limit results to tests of the specified subsuite (eg. "devtools").',
)
@CommandArgument(
"paths", nargs=argparse.REMAINDER, help="File system paths of interest."
)
@CommandArgument(
"--show-manifests",
action="store_true",
help="Include test manifests in report.",
)
@CommandArgument(
"--show-tests", action="store_true", help="Include individual tests in report."
)
@CommandArgument(
"--show-summary", action="store_true", help="Include summary in report."
)
@CommandArgument(
"--show-annotations",
action="store_true",
help="Include list of manifest annotation conditions in report.",
)
@CommandArgument(
"--show-testruns",
action="store_true",
help="Include total number of runs the test has if there are failures.",
)
@CommandArgument(
"--filter-values",
help="Comma-separated list of value regular expressions to filter on; "
"displayed tests contain all specified values.",
)
@CommandArgument(
"--filter-keys",
help="Comma-separated list of test keys to filter on, "
'like "skip-if"; only these fields will be searched '
"for filter-values.",
)
@CommandArgument(
"--no-component-report",
action="store_false",
dest="show_components",
default=True,
help="Do not categorize by bugzilla component.",
)
@CommandArgument("--output-file", help="Path to report file.")
@CommandArgument("--runcounts-input-file", help="Optional path to report file.")
@CommandArgument("--verbose", action="store_true", help="Enable debug logging.")
@CommandArgument(
"--start",
default=(date.today() - timedelta(30)).strftime("%Y-%m-%d"),
help="Start date (YYYY-MM-DD)",
)
@CommandArgument(
"--end", default=date.today().strftime("%Y-%m-%d"), help="End date (YYYY-MM-DD)"
)
def test_report(
command_context,
components,
flavor,
subsuite,
paths,
show_manifests,
show_tests,
show_summary,
show_annotations,
filter_values,
filter_keys,
show_components,
output_file,
verbose,
start,
end,
show_testruns,
runcounts_input_file,
):
import testinfo
from mozbuild import build_commands
try:
command_context.config_environment
except BuildEnvironmentNotFoundException:
print("Looks like configure has not run yet, running it now...")
build_commands.configure(command_context)
ti = testinfo.TestInfoReport(verbose)
ti.report(
components,
flavor,
subsuite,
paths,
show_manifests,
show_tests,
show_summary,
show_annotations,
filter_values,
filter_keys,
show_components,
output_file,
start,
end,
show_testruns,
runcounts_input_file,
)
@SubCommand(
"test-info",
"report-diff",
description='Compare two reports generated by "test-info reports".',
)
@CommandArgument(
"--before",
default=None,
help="The first (earlier) report file; path to local file or url.",
)
@CommandArgument(
"--after", help="The second (later) report file; path to local file or url."
)
@CommandArgument(
"--output-file",
help="Path to report file to be written. If not specified, report"
"will be written to standard output.",
)
@CommandArgument("--verbose", action="store_true", help="Enable debug logging.")
def test_report_diff(command_context, before, after, output_file, verbose):
import testinfo
ti = testinfo.TestInfoReport(verbose)
ti.report_diff(before, after, output_file)
@SubCommand(
"test-info",
"testrun-report",
description="Generate report of number of runs for each test group (manifest)",
)
@CommandArgument("--output-file", help="Path to report file.")
def test_info_testrun_report(command_context, output_file):
import json
import testinfo
ti = testinfo.TestInfoReport(verbose=True)
if os.environ.get("GECKO_HEAD_REPOSITORY", "") in [
]:
runcounts = ti.get_runcounts()
if output_file:
output_file = os.path.abspath(output_file)
output_dir = os.path.dirname(output_file)
if not os.path.isdir(output_dir):
os.makedirs(output_dir)
with open(output_file, "w") as f:
json.dump(runcounts, f)
else:
print(runcounts)
@SubCommand(
"test-info",
"failure-report",
description="Display failure line groupings and frequencies for "
"single tracking intermittent bugs.",
)
@CommandArgument(
"--start",
default=(date.today() - timedelta(30)).strftime("%Y-%m-%d"),
help="Start date (YYYY-MM-DD)",
)
@CommandArgument(
"--end", default=date.today().strftime("%Y-%m-%d"), help="End date (YYYY-MM-DD)"
)
@CommandArgument(
"--bugid",
default=None,
help="bugid for treeherder intermittent failures data query.",
)
def test_info_failures(
command_context,
start,
end,
bugid,
):
# bugid comes in as a string, we need an int:
try:
bugid = int(bugid)
except ValueError:
bugid = None
if not bugid:
print("Please enter a valid bugid (i.e. '1760132')")
return
# get bug info
url = (
% bugid
)
r = requests.get(url, headers={"User-agent": "mach-test-info/1.0"})
if r.status_code != 200:
print("%s error retrieving url: %s" % (r.status_code, url))
data = r.json()
if not data:
print("unable to get bugzilla information for %s" % bugid)
return
summary = data["bugs"][0]["summary"]
parts = summary.split("|")
if not summary.endswith("single tracking bug") or len(parts) != 2:
print("this query only works with single tracking bugs")
return
# get depends_on bugs:
buglist = [bugid]
if "depends_on" in data["bugs"][0]:
buglist.extend(data["bugs"][0]["depends_on"])
testname = parts[0].strip().split(" ")[-1]
# now query treeherder to get details about annotations
data = []
for b in buglist:
url += "?startday=%s&endday=%s&tree=trunk&bug=%s" % (start, end, b)
r = requests.get(url, headers={"User-agent": "mach-test-info/1.0"})
r.raise_for_status()
bdata = r.json()
data.extend(bdata)
if len(data) == 0:
print("no failures were found for given bugid, please ensure bug is")
return
# query VCS to get current list of variants:
import yaml
r = requests.get(url, headers={"User-agent": "mach-test-info/1.0"})
variants = yaml.safe_load(r.text)
print(
"\nQuerying data for bug %s annotated from %s to %s on trunk.\n\n"
% (buglist, start, end)
)
jobs = {}
lines = {}
for failure in data:
# config = platform/buildtype
# testsuite (<suite>[-variant][-<chunk>])
# lines - group by patterns that contain test name
config = "%s/%s" % (failure["platform"], failure["build_type"])
variant = ""
suite = ""
varpos = len(failure["test_suite"])
for v in variants.keys():
var = "-%s" % variants[v]["suffix"]
if var in failure["test_suite"]:
if failure["test_suite"].find(var) < varpos:
variant = var
if variant:
suite = failure["test_suite"].split(variant)[0]
parts = failure["test_suite"].split("-")
try:
int(parts[-1])
suite = "-".join(parts[:-1])
except ValueError:
pass # if this works, then the last '-X' is a number :)
if suite == "":
print("Error: failure to find variant in %s" % failure["test_suite"])
job = "%s-%s%s" % (config, suite, variant)
if job not in jobs.keys():
jobs[job] = 0
jobs[job] += 1
# lines - sum(hash) of all lines where we match testname
hvalue = 0
for line in failure["lines"]:
if len(line.split(testname)) <= 1:
continue
# strip off timestamp and mozharness status
parts = line.split("TEST-UNEXPECTED")
l = "TEST-UNEXPECTED%s" % parts[-1]
# only keep 25 characters of the failure, often longer is random numbers
parts = l.split(testname)
l = "%s%s%s" % (parts[0], testname, parts[1][:25])
hvalue += hash(l)
if not failure["lines"]:
hvalue = 1
if not hvalue:
continue
if hvalue not in lines.keys():
lines[hvalue] = {"lines": failure["lines"], "config": []}
lines[hvalue]["config"].append(job)
for h in lines.keys():
print("%s errors with:" % (len(lines[h]["config"])))
for l in lines[h]["lines"]:
print(l)
else:
print(
"... no failure lines recorded in"
)
for job in jobs:
count = len([x for x in lines[h]["config"] if x == job])
if count > 0:
print(" %s: %s" % (job, count))
print("")
@Command(
"rusttests",
category="testing",
conditions=[conditions.is_non_artifact_build],
description="Run rust unit tests (via cargo test).",
)
def run_rusttests(command_context, **kwargs):
return command_context._mach_context.commands.dispatch(
"build",
command_context._mach_context,
what=["pre-export", "export", "recurse_rusttests"],
)
@Command(
"fluent-migration-test",
category="testing",
description="Test Fluent migration recipes.",
)
@CommandArgument(
"--l10n-git",
action="store_true",
dest="l10n_git",
help="Use git rather than hg source repository",
)
@CommandArgument("test_paths", nargs="*", metavar="N", help="Recipe paths to test.")
def run_migration_tests(command_context, l10n_git=False, test_paths=None, **kwargs):
if not test_paths:
test_paths = []
command_context.activate_virtualenv()
from test_fluent_migrations import fmt
rv = 0
with_context = []
for to_test in test_paths:
try:
context = fmt.inspect_migration(to_test)
for issue in context["issues"]:
command_context.log(
logging.ERROR,
"fluent-migration-test",
{
"error": issue["msg"],
"file": to_test,
},
"ERROR in {file}: {error}",
)
if context["issues"]:
continue
with_context.append(
{
"to_test": to_test,
"references": context["references"],
}
)
except Exception as e:
command_context.log(
logging.ERROR,
"fluent-migration-test",
{"error": str(e), "file": to_test},
"ERROR in {file}: {error}",
)
rv |= 1
obj_dir, repo_dir = fmt.prepare_directories(command_context, l10n_git)
for context in with_context:
rv |= fmt.test_migration(
command_context, obj_dir, repo_dir, l10n_git, **context
)
return rv
@Command(
"manifest",
category="testing",
description="Manifest operations",
virtualenv_name="manifest",
)
def manifest(_command_context):
"""
All functions implemented as subcommands.
"""
@SubCommand(
"manifest",
"skip-fails",
description="Update manifests to skip failing tests",
)
@CommandArgument("try_url", nargs=1, help="Treeherder URL for try (please use quotes)")
@CommandArgument(
"-b",
"--bugzilla",
default=None,
dest="bugzilla",
help="Bugzilla instance (or disable)",
)
@CommandArgument(
"-m", "--meta-bug-id", default=None, dest="meta_bug_id", help="Meta Bug id"
)
@CommandArgument(
"-s",
"--turbo",
action="store_true",
dest="turbo",
help="Skip all secondary failures",
)
@CommandArgument(
"-t", "--save-tasks", default=None, dest="save_tasks", help="Save tasks to file"
)
@CommandArgument(
"-T", "--use-tasks", default=None, dest="use_tasks", help="Use tasks from file"
)
@CommandArgument(
"-f",
"--save-failures",
default=None,
dest="save_failures",
help="Save failures to file",
)
@CommandArgument(
"-F",
"--use-failures",
default=None,
dest="use_failures",
help="Use failures from file",
)
@CommandArgument(
"-M",
"--max-failures",
default=-1,
dest="max_failures",
help="Maximum number of failures to skip (-1 == no limit)",
)
@CommandArgument("-v", "--verbose", action="store_true", help="Verbose mode")
@CommandArgument(
"-d",
"--dry-run",
action="store_true",
help="Determine manifest changes, but do not write them",
)
def skipfails(
command_context,
try_url,
bugzilla=None,
meta_bug_id=None,
turbo=False,
save_tasks=None,
use_tasks=None,
save_failures=None,
use_failures=None,
max_failures=-1,
verbose=False,
dry_run=False,
):
from skipfails import Skipfails
if meta_bug_id is not None:
try:
meta_bug_id = int(meta_bug_id)
except ValueError:
meta_bug_id = None
if max_failures is not None:
try:
max_failures = int(max_failures)
except ValueError:
max_failures = -1
else:
max_failures = -1
Skipfails(command_context, try_url, verbose, bugzilla, dry_run, turbo).run(
meta_bug_id,
save_tasks,
use_tasks,
save_failures,
use_failures,
max_failures,
)