mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-12-01 00:32:11 +00:00
5237b5f25c
This takes advantage of the browser-selection patch proposed to Puppeteer. A --product option allows choice between 'firefox' and 'chrome'. Puppeteer takes care of profile creation for Firefox. Additional Puppeteer Launcher options can be passed along with --setopt. Depends on D52313 Differential Revision: https://phabricator.services.mozilla.com/D52314 --HG-- extra : moz-landing-system : lando
331 lines
10 KiB
Python
331 lines
10 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/.
|
|
|
|
from __future__ import (
|
|
absolute_import,
|
|
print_function,
|
|
unicode_literals,
|
|
)
|
|
|
|
import json
|
|
import sys
|
|
import os
|
|
import tempfile
|
|
import shutil
|
|
import subprocess
|
|
|
|
from mach.decorators import (
|
|
Command,
|
|
CommandArgument,
|
|
CommandProvider,
|
|
SubCommand,
|
|
)
|
|
|
|
from mozbuild.base import (
|
|
MachCommandBase,
|
|
MozbuildObject,
|
|
)
|
|
from mozbuild import nodeutil
|
|
import mozprofile
|
|
|
|
|
|
EX_CONFIG = 78
|
|
EX_SOFTWARE = 70
|
|
EX_USAGE = 64
|
|
|
|
DEFAULT_REPO = "https://github.com/andreastt/puppeteer.git"
|
|
DEFAULT_COMMITISH = "firefox"
|
|
|
|
|
|
def setup():
|
|
# add node and npm from mozbuild to front of system path
|
|
npm, _ = nodeutil.find_npm_executable()
|
|
if not npm:
|
|
exit(EX_CONFIG, "could not find npm executable")
|
|
path = os.path.abspath(os.path.join(npm, os.pardir))
|
|
os.environ["PATH"] = "{}:{}".format(path, os.environ["PATH"])
|
|
|
|
|
|
@CommandProvider
|
|
class RemoteCommands(MachCommandBase):
|
|
def __init__(self, context):
|
|
MachCommandBase.__init__(self, context)
|
|
self.remotedir = os.path.join(self.topsrcdir, "remote")
|
|
|
|
@Command("remote", category="misc",
|
|
description="Remote protocol related operations.")
|
|
def remote(self):
|
|
self.parser.print_usage()
|
|
exit(EX_USAGE)
|
|
|
|
@SubCommand("remote", "vendor-puppeteer",
|
|
"Pull in latest changes of the Puppeteer client.")
|
|
@CommandArgument("--repository",
|
|
metavar="REPO",
|
|
default=DEFAULT_REPO,
|
|
help="The (possibly remote) repository to clone from. "
|
|
"Defaults to {}.".format(DEFAULT_REPO))
|
|
@CommandArgument("--commitish",
|
|
metavar="COMMITISH",
|
|
default=DEFAULT_COMMITISH,
|
|
help="The commit or tag object name to check out. "
|
|
"Defaults to \"{}\".".format(DEFAULT_COMMITISH))
|
|
def vendor_puppeteer(self, repository, commitish):
|
|
puppeteerdir = os.path.join(self.remotedir, "test", "puppeteer")
|
|
|
|
shutil.rmtree(puppeteerdir, ignore_errors=True)
|
|
os.makedirs(puppeteerdir)
|
|
with TemporaryDirectory() as tmpdir:
|
|
git("clone", "-q", repository, tmpdir)
|
|
git("checkout", commitish, worktree=tmpdir)
|
|
git("checkout-index", "-a", "-f",
|
|
"--prefix", "{}/".format(puppeteerdir),
|
|
worktree=tmpdir)
|
|
|
|
# remove files which may interfere with git checkout of central
|
|
try:
|
|
os.remove(os.path.join(puppeteerdir, ".gitattributes"))
|
|
os.remove(os.path.join(puppeteerdir, ".gitignore"))
|
|
except OSError:
|
|
pass
|
|
|
|
import yaml
|
|
annotation = {
|
|
"schema": 1,
|
|
"bugzilla": {
|
|
"product": "Remote Protocol",
|
|
"component": "Agent",
|
|
},
|
|
"origin": {
|
|
"name": "puppeteer",
|
|
"description": "Headless Chrome Node API",
|
|
"url": repository,
|
|
"license": "Apache-2.0",
|
|
"release": commitish,
|
|
},
|
|
}
|
|
with open(os.path.join(puppeteerdir, "moz.yaml"), "w") as fh:
|
|
yaml.safe_dump(annotation, fh,
|
|
default_flow_style=False,
|
|
encoding="utf-8",
|
|
allow_unicode=True)
|
|
|
|
|
|
def git(*args, **kwargs):
|
|
cmd = ("git",)
|
|
if kwargs.get("worktree"):
|
|
cmd += ("-C", kwargs["worktree"])
|
|
cmd += args
|
|
|
|
pipe = kwargs.get("pipe")
|
|
git_p = subprocess.Popen(cmd,
|
|
env={"GIT_CONFIG_NOSYSTEM": "1"},
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE)
|
|
pipe_p = None
|
|
if pipe:
|
|
pipe_p = subprocess.Popen(pipe, stdin=git_p.stdout, stderr=subprocess.PIPE)
|
|
|
|
if pipe:
|
|
_, pipe_err = pipe_p.communicate()
|
|
out, git_err = git_p.communicate()
|
|
|
|
# use error from first program that failed
|
|
if git_p.returncode > 0:
|
|
exit(EX_SOFTWARE, git_err)
|
|
if pipe and pipe_p.returncode > 0:
|
|
exit(EX_SOFTWARE, pipe_err)
|
|
|
|
return out
|
|
|
|
|
|
def npm(*args, **kwargs):
|
|
env = None
|
|
if kwargs.get("env"):
|
|
env = os.environ.copy()
|
|
env.update(kwargs["env"])
|
|
|
|
p = subprocess.Popen(("npm",) + args,
|
|
cwd=kwargs.get("cwd"),
|
|
env=env)
|
|
|
|
p.wait()
|
|
if p.returncode > 0:
|
|
exit(p.returncode, "npm: exit code {}".format(p.returncode))
|
|
|
|
|
|
# tempfile.TemporaryDirectory missing from Python 2.7
|
|
class TemporaryDirectory(object):
|
|
def __init__(self):
|
|
self.path = tempfile.mkdtemp()
|
|
self._closed = False
|
|
|
|
def __repr__(self):
|
|
return "<{} {!r}>".format(self.__class__.__name__, self.path)
|
|
|
|
def __enter__(self):
|
|
return self.path
|
|
|
|
def __exit__(self, exc, value, tb):
|
|
self.clean()
|
|
|
|
def __del__(self):
|
|
self.clean()
|
|
|
|
def clean(self):
|
|
if self.path and not self._closed:
|
|
shutil.rmtree(self.path)
|
|
self._closed = True
|
|
|
|
|
|
class PuppeteerRunner(MozbuildObject):
|
|
def __init__(self, *args, **kwargs):
|
|
super(PuppeteerRunner, self).__init__(*args, **kwargs)
|
|
|
|
self.remotedir = os.path.join(self.topsrcdir, "remote")
|
|
self.puppeteerdir = os.path.join(self.remotedir, "test", "puppeteer")
|
|
|
|
def run_test(self, *tests, **params):
|
|
"""
|
|
Runs Puppeteer unit tests with npm.
|
|
|
|
Possible optional test parameters:
|
|
|
|
`binary`:
|
|
Path for the browser binary to use. Defaults to the local
|
|
build.
|
|
`jobs`:
|
|
Number of tests to run in parallel. Defaults to not
|
|
parallelise, e.g. `-j1`.
|
|
`headless`:
|
|
Boolean to indicate whether to activate Firefox' headless mode.
|
|
`extra_prefs`:
|
|
Dictionary of extra preferences to write to the profile,
|
|
before invoking npm. Overrides default preferences.
|
|
"""
|
|
setup()
|
|
|
|
binary = params.get("binary") or self.get_binary_path()
|
|
product = params.get("product", "firefox")
|
|
|
|
env = {"DUMPIO": "1"}
|
|
extra_options = {}
|
|
for k, v in params.get("extra_launcher_options", {}).items():
|
|
extra_options[k] = json.loads(v)
|
|
|
|
if product == "firefox":
|
|
env["BINARY"] = binary
|
|
command = ["run", "funit", "--verbose"]
|
|
elif product == "chrome":
|
|
command = ["run", "unit", "--verbose"]
|
|
|
|
if params.get("jobs"):
|
|
env["PPTR_PARALLEL_TESTS"] = str(params["jobs"])
|
|
env["HEADLESS"] = str(params.get("headless", False))
|
|
|
|
prefs = {}
|
|
for k, v in params.get("extra_prefs", {}).items():
|
|
prefs[k] = mozprofile.Preferences.cast(v)
|
|
|
|
if prefs:
|
|
extra_options["extraPrefs"] = prefs
|
|
|
|
if extra_options:
|
|
env["EXTRA_LAUNCH_OPTIONS"] = json.dumps(extra_options)
|
|
|
|
return npm(*command, cwd=self.puppeteerdir, env=env)
|
|
|
|
|
|
@CommandProvider
|
|
class PuppeteerTest(MachCommandBase):
|
|
@Command("puppeteer-test", category="testing",
|
|
description="Run Puppeteer unit tests.")
|
|
@CommandArgument("--product",
|
|
type=str,
|
|
default="firefox",
|
|
choices=["chrome", "firefox"])
|
|
@CommandArgument("--binary",
|
|
type=str,
|
|
help="Path to browser binary. Defaults to local Firefox build.")
|
|
@CommandArgument("-z", "--headless",
|
|
action="store_true",
|
|
help="Run browser in headless mode.")
|
|
@CommandArgument("--setpref",
|
|
action="append",
|
|
dest="extra_prefs",
|
|
metavar="<pref>=<value>",
|
|
help="Defines additional user preferences.")
|
|
@CommandArgument("--setopt",
|
|
action="append",
|
|
dest="extra_options",
|
|
metavar="<option>=<value>",
|
|
help="Defines additional options for `puppeteer.launch`.")
|
|
@CommandArgument("-j",
|
|
dest="jobs",
|
|
type=int,
|
|
metavar="<N>",
|
|
help="Optionally run tests in parallel.")
|
|
@CommandArgument("tests", nargs="*")
|
|
def puppeteer_test(self, binary=None, headless=False, extra_prefs=None,
|
|
extra_options=None, jobs=1, tests=None, product="firefox", **kwargs):
|
|
# moztest calls this programmatically with test objects or manifests
|
|
if "test_objects" in kwargs and tests is not None:
|
|
raise ValueError("Expected either 'test_objects' or 'tests'")
|
|
|
|
if product != "firefox" and extra_prefs is not None:
|
|
raise ValueError("User preferences are not recognized by %s" % product)
|
|
|
|
if "test_objects" in kwargs:
|
|
tests = []
|
|
for test in kwargs["test_objects"]:
|
|
tests.append(test["path"])
|
|
|
|
prefs = {}
|
|
for s in (extra_prefs or []):
|
|
kv = s.split("=")
|
|
if len(kv) != 2:
|
|
exit(EX_USAGE, "syntax error in --setpref={}".format(s))
|
|
prefs[kv[0]] = kv[1].strip()
|
|
|
|
options = {}
|
|
for s in (extra_options or []):
|
|
kv = s.split("=")
|
|
if len(kv) != 2:
|
|
exit(EX_USAGE, "syntax error in --setopt={}".format(s))
|
|
options[kv[0]] = kv[1].strip()
|
|
|
|
self.install_puppeteer(product)
|
|
|
|
params = {"binary": binary,
|
|
"headless": headless,
|
|
"extra_prefs": prefs,
|
|
"product": product,
|
|
"jobs": jobs,
|
|
"extra_launcher_options": options}
|
|
puppeteer = self._spawn(PuppeteerRunner)
|
|
try:
|
|
return puppeteer.run_test(*tests, **params)
|
|
except Exception as e:
|
|
exit(EX_SOFTWARE, e)
|
|
|
|
def install_puppeteer(self, product):
|
|
setup()
|
|
env = {}
|
|
if product != "chrome":
|
|
env["PUPPETEER_SKIP_CHROMIUM_DOWNLOAD"] = "1"
|
|
npm("install",
|
|
cwd=os.path.join(self.topsrcdir, "remote", "test", "puppeteer"),
|
|
env=env)
|
|
|
|
|
|
def exit(code, error=None):
|
|
if error is not None:
|
|
if isinstance(error, Exception):
|
|
import traceback
|
|
traceback.print_exc()
|
|
else:
|
|
message = str(error).split("\n")[0].strip()
|
|
print("{}: {}".format(sys.argv[0], message), file=sys.stderr)
|
|
sys.exit(code)
|