summaryrefslogtreecommitdiffstats
path: root/testing/mozharness/mach_commands.py
diff options
context:
space:
mode:
Diffstat (limited to 'testing/mozharness/mach_commands.py')
-rw-r--r--testing/mozharness/mach_commands.py196
1 files changed, 196 insertions, 0 deletions
diff --git a/testing/mozharness/mach_commands.py b/testing/mozharness/mach_commands.py
new file mode 100644
index 000000000..f453397db
--- /dev/null
+++ b/testing/mozharness/mach_commands.py
@@ -0,0 +1,196 @@
+# 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/.
+
+from __future__ import absolute_import, print_function, unicode_literals
+
+import argparse
+import os
+import re
+import subprocess
+import sys
+import urllib
+import urlparse
+
+import mozinfo
+
+from mach.decorators import (
+ CommandArgument,
+ CommandProvider,
+ Command,
+)
+
+from mozbuild.base import MachCommandBase, MozbuildObject
+from mozbuild.base import MachCommandConditions as conditions
+from argparse import ArgumentParser
+
+def get_parser():
+ parser = argparse.ArgumentParser()
+ parser.add_argument("suite_name", nargs=1, type=str, action="store",
+ help="Suite to run in mozharness")
+ parser.add_argument("mozharness_args", nargs=argparse.REMAINDER,
+ help="Extra arguments to pass to mozharness")
+ return parser
+
+class MozharnessRunner(MozbuildObject):
+ def __init__(self, *args, **kwargs):
+ MozbuildObject.__init__(self, *args, **kwargs)
+
+
+ self.test_packages_url = self._test_packages_url()
+ self.installer_url = self._installer_url()
+
+ desktop_unittest_config = [
+ "--config-file", lambda: self.config_path("unittests",
+ "%s_unittest.py" % mozinfo.info['os']),
+ "--config-file", lambda: self.config_path("developer_config.py")]
+
+ self.config = {
+ "__defaults__": {
+ "config": ["--no-read-buildbot-config",
+ "--download-symbols", "ondemand",
+ "--installer-url", self.installer_url,
+ "--test-packages-url", self.test_packages_url]
+ },
+
+ "mochitest-valgrind": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--mochitest-suite", "valgrind-plain"]
+ },
+ "mochitest": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--mochitest-suite", "plain"]
+ },
+ "mochitest-chrome": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--mochitest-suite", "chrome"]
+ },
+ "mochitest-browser-chrome": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--mochitest-suite", "browser-chrome"]
+ },
+ "mochitest-devtools-chrome": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--mochitest-suite", "mochitest-devtools-chrome"]
+ },
+ "crashtest": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--reftest-suite", "crashtest"]
+ },
+ "jsreftest": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--reftest-suite", "jsreftest"]
+ },
+ "reftest": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--reftest-suite", "reftest"]
+ },
+ "reftest-no-accel": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--reftest-suite", "reftest-no-accel"]
+ },
+ "cppunittest": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--cppunittest-suite", "cppunittest"]
+ },
+ "xpcshell": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--xpcshell-suite", "xpcshell"]
+ },
+ "xpcshell-addons": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--xpcshell-suite", "xpcshell-addons"]
+ },
+ "jittest": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--jittest-suite", "jittest"]
+ },
+ "mozbase": {
+ "script": "desktop_unittest.py",
+ "config": desktop_unittest_config + [
+ "--mozbase-suite", "mozbase"]
+ },
+ "marionette": {
+ "script": "marionette.py",
+ "config": ["--config-file", self.config_path("marionette",
+ "test_config.py")]
+ },
+ "web-platform-tests": {
+ "script": "web_platform_tests.py",
+ "config": ["--config-file", self.config_path("web_platform_tests",
+ self.wpt_config)]
+ },
+ }
+
+
+ def path_to_url(self, path):
+ return urlparse.urljoin('file:', urllib.pathname2url(path))
+
+ def _installer_url(self):
+ package_re = {
+ "linux": re.compile("^firefox-\d+\..+\.tar\.bz2$"),
+ "win": re.compile("^firefox-\d+\..+\.installer\.exe$"),
+ "mac": re.compile("^firefox-\d+\..+\.mac(?:64)?\.dmg$"),
+ }[mozinfo.info['os']]
+ dist_path = os.path.join(self.topobjdir, "dist")
+ filenames = [item for item in os.listdir(dist_path) if
+ package_re.match(item)]
+ assert len(filenames) == 1
+ return self.path_to_url(os.path.join(dist_path, filenames[0]))
+
+ def _test_packages_url(self):
+ dist_path = os.path.join(self.topobjdir, "dist")
+ filenames = [item for item in os.listdir(dist_path) if
+ item.endswith('test_packages.json')]
+ assert len(filenames) == 1
+ return self.path_to_url(os.path.join(dist_path, filenames[0]))
+
+ def config_path(self, *parts):
+ return self.path_to_url(os.path.join(self.topsrcdir, "testing", "mozharness",
+ "configs", *parts))
+
+ @property
+ def wpt_config(self):
+ return "test_config.py" if mozinfo.info['os'] != "win" else "test_config_windows.py"
+
+ def run_suite(self, suite, **kwargs):
+ default_config = self.config.get("__defaults__")
+ suite_config = self.config.get(suite)
+
+ if suite_config is None:
+ print("Unknown suite %s" % suite)
+ return 1
+
+ script = os.path.join(self.topsrcdir, "testing", "mozharness",
+ "scripts", suite_config["script"])
+ options = [item() if callable(item) else item
+ for item in default_config["config"] + suite_config["config"]]
+
+ cmd = [script] + options
+
+ rv = subprocess.call(cmd, cwd=os.path.dirname(script))
+ return rv
+
+
+@CommandProvider
+class MozharnessCommands(MachCommandBase):
+ @Command('mozharness', category='testing',
+ description='Run tests using mozharness.',
+ conditions=[conditions.is_firefox],
+ parser=get_parser)
+ def mozharness(self, **kwargs):
+ runner = self._spawn(MozharnessRunner)
+ return runner.run_suite(kwargs.pop("suite_name")[0], **kwargs)