mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-26 11:45:37 +00:00
193 lines
6.9 KiB
Python
Executable File
193 lines
6.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# ***** BEGIN LICENSE BLOCK *****
|
|
# 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/.
|
|
# ***** END LICENSE BLOCK *****
|
|
import os
|
|
import sys
|
|
import copy
|
|
|
|
# load modules from parent dir
|
|
sys.path.insert(1, os.path.dirname(sys.path[0]))
|
|
|
|
from mozharness.base.script import PreScriptAction
|
|
from mozharness.base.vcs.vcsbase import MercurialScript
|
|
from mozharness.mozilla.blob_upload import BlobUploadMixin, blobupload_config_options
|
|
from mozharness.mozilla.testing.testbase import TestingMixin, testing_config_options
|
|
|
|
from mozharness.mozilla.structuredlog import StructuredOutputParser
|
|
from mozharness.base.log import INFO
|
|
|
|
class WebPlatformTest(TestingMixin, MercurialScript, BlobUploadMixin):
|
|
config_options = [
|
|
[['--test-type'], {
|
|
"action": "extend",
|
|
"dest": "test_type",
|
|
"help": "Specify the test types to run."}
|
|
],
|
|
[['--e10s'], {
|
|
"action": "store_true",
|
|
"dest": "e10s",
|
|
"help": "Run with e10s enabled"}
|
|
],
|
|
[["--total-chunks"], {
|
|
"action": "store",
|
|
"dest": "total_chunks",
|
|
"help": "Number of total chunks"}
|
|
],
|
|
[["--this-chunk"], {
|
|
"action": "store",
|
|
"dest": "this_chunk",
|
|
"help": "Number of this chunk"}]
|
|
] + copy.deepcopy(testing_config_options) + \
|
|
copy.deepcopy(blobupload_config_options)
|
|
|
|
def __init__(self, require_config_file=True):
|
|
super(WebPlatformTest, self).__init__(
|
|
config_options=self.config_options,
|
|
all_actions=[
|
|
'clobber',
|
|
'read-buildbot-config',
|
|
'download-and-extract',
|
|
'create-virtualenv',
|
|
'pull',
|
|
'install',
|
|
'run-tests',
|
|
],
|
|
require_config_file=require_config_file,
|
|
config={'require_test_zip': True})
|
|
|
|
# Surely this should be in the superclass
|
|
c = self.config
|
|
self.installer_url = c.get('installer_url')
|
|
self.test_url = c.get('test_url')
|
|
self.test_packages_url = c.get('test_packages_url')
|
|
self.installer_path = c.get('installer_path')
|
|
self.binary_path = c.get('binary_path')
|
|
self.abs_app_dir = None
|
|
|
|
def query_abs_app_dir(self):
|
|
"""We can't set this in advance, because OSX install directories
|
|
change depending on branding and opt/debug.
|
|
"""
|
|
if self.abs_app_dir:
|
|
return self.abs_app_dir
|
|
if not self.binary_path:
|
|
self.fatal("Can't determine abs_app_dir (binary_path not set!)")
|
|
self.abs_app_dir = os.path.dirname(self.binary_path)
|
|
return self.abs_app_dir
|
|
|
|
def query_abs_dirs(self):
|
|
if self.abs_dirs:
|
|
return self.abs_dirs
|
|
abs_dirs = super(WebPlatformTest, self).query_abs_dirs()
|
|
|
|
dirs = {}
|
|
dirs['abs_app_install_dir'] = os.path.join(abs_dirs['abs_work_dir'], 'application')
|
|
dirs['abs_test_install_dir'] = os.path.join(abs_dirs['abs_work_dir'], 'tests')
|
|
dirs["abs_wpttest_dir"] = os.path.join(dirs['abs_test_install_dir'], "web-platform")
|
|
dirs['abs_blob_upload_dir'] = os.path.join(abs_dirs['abs_work_dir'], 'blobber_upload_dir')
|
|
|
|
abs_dirs.update(dirs)
|
|
self.abs_dirs = abs_dirs
|
|
|
|
return self.abs_dirs
|
|
|
|
@PreScriptAction('create-virtualenv')
|
|
def _pre_create_virtualenv(self, action):
|
|
dirs = self.query_abs_dirs()
|
|
|
|
requirements = os.path.join(dirs['abs_test_install_dir'],
|
|
'config',
|
|
'marionette_requirements.txt')
|
|
|
|
self.register_virtualenv_module(requirements=[requirements],
|
|
two_pass=True)
|
|
|
|
def _query_cmd(self):
|
|
if not self.binary_path:
|
|
self.fatal("Binary path could not be determined")
|
|
#And exit
|
|
|
|
c = self.config
|
|
dirs = self.query_abs_dirs()
|
|
abs_app_dir = self.query_abs_app_dir()
|
|
run_file_name = "runtests.py"
|
|
|
|
base_cmd = [self.query_python_path('python'), '-u']
|
|
base_cmd.append(os.path.join(dirs["abs_wpttest_dir"], run_file_name))
|
|
|
|
# Make sure that the logging directory exists
|
|
if self.mkdir_p(dirs["abs_blob_upload_dir"]) == -1:
|
|
self.fatal("Could not create blobber upload directory")
|
|
# Exit
|
|
|
|
base_cmd += ["--log-raw=-",
|
|
"--log-raw=%s" % os.path.join(dirs["abs_blob_upload_dir"],
|
|
"wpt_raw.log"),
|
|
"--binary=%s" % self.binary_path,
|
|
"--symbols-path=%s" % self.query_symbols_url(),
|
|
"--stackwalk-binary=%s" % self.query_minidump_stackwalk()]
|
|
|
|
for test_type in c.get("test_type", []):
|
|
base_cmd.append("--test-type=%s" % test_type)
|
|
|
|
if c.get("e10s"):
|
|
base_cmd.append("--e10s")
|
|
|
|
for opt in ["total_chunks", "this_chunk"]:
|
|
val = c.get(opt)
|
|
if val:
|
|
base_cmd.append("--%s=%s" % (opt.replace("_", "-"), val))
|
|
|
|
options = list(c.get("options", []))
|
|
|
|
str_format_values = {
|
|
'binary_path': self.binary_path,
|
|
'test_path': dirs["abs_wpttest_dir"],
|
|
'test_install_path': dirs["abs_test_install_dir"],
|
|
'abs_app_dir': abs_app_dir,
|
|
'abs_work_dir': dirs["abs_work_dir"]
|
|
}
|
|
|
|
opt_cmd = [item % str_format_values for item in options]
|
|
|
|
return base_cmd + opt_cmd
|
|
|
|
def download_and_extract(self):
|
|
super(WebPlatformTest, self).download_and_extract(
|
|
target_unzip_dirs=["bin/*",
|
|
"config/*",
|
|
"mozbase/*",
|
|
"marionette/*",
|
|
"web-platform/*"],
|
|
suite_categories=["web-platform"])
|
|
|
|
def run_tests(self):
|
|
dirs = self.query_abs_dirs()
|
|
cmd = self._query_cmd()
|
|
cmd = self.append_harness_extra_args(cmd)
|
|
|
|
parser = StructuredOutputParser(config=self.config,
|
|
log_obj=self.log_obj)
|
|
|
|
env = {'MINIDUMP_SAVE_PATH': dirs['abs_blob_upload_dir']}
|
|
env = self.query_env(partial_env=env, log_level=INFO)
|
|
|
|
return_code = self.run_command(cmd,
|
|
cwd=dirs['abs_work_dir'],
|
|
output_timeout=1000,
|
|
output_parser=parser,
|
|
env=env)
|
|
|
|
tbpl_status, log_level = parser.evaluate_parser(return_code)
|
|
|
|
self.buildbot_status(tbpl_status, level=log_level)
|
|
|
|
|
|
# main {{{1
|
|
if __name__ == '__main__':
|
|
web_platform_tests = WebPlatformTest()
|
|
web_platform_tests.run_and_exit()
|