forked from mirrors/gecko-dev
d024a31639
This patch was generated via: $ pyupgrade --py36-plus $(find taskcluster/taskgraph -name "*.py" -type f) $ autoflake --remove-all-unused-imports -i $(find taskcluster/taskgraph -name "*.py" -type f) The same set of commands are being applied to standalone taskgraph as well so they remain in sync. Differential Revision: https://phabricator.services.mozilla.com/D123234
46 lines
1.3 KiB
Python
46 lines
1.3 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/.
|
|
"""
|
|
Support for running mach python-test tasks (via run-task)
|
|
"""
|
|
|
|
|
|
from taskgraph.transforms.job import run_job_using, configure_taskdesc_for_run
|
|
from taskgraph.util.schema import Schema
|
|
from voluptuous import Required, Optional
|
|
|
|
python_test_schema = Schema(
|
|
{
|
|
Required("using"): "python-test",
|
|
# Python version to use
|
|
Required("python-version"): int,
|
|
# The subsuite to run
|
|
Required("subsuite"): str,
|
|
# Base work directory used to set up the task.
|
|
Optional("workdir"): str,
|
|
}
|
|
)
|
|
|
|
|
|
defaults = {
|
|
"python-version": 3,
|
|
"subsuite": "default",
|
|
}
|
|
|
|
|
|
@run_job_using(
|
|
"docker-worker", "python-test", schema=python_test_schema, defaults=defaults
|
|
)
|
|
@run_job_using(
|
|
"generic-worker", "python-test", schema=python_test_schema, defaults=defaults
|
|
)
|
|
def configure_python_test(config, job, taskdesc):
|
|
run = job["run"]
|
|
worker = job["worker"]
|
|
|
|
# defer to the mach implementation
|
|
run["mach"] = ("python-test --subsuite {subsuite} --run-slow").format(**run)
|
|
run["using"] = "mach"
|
|
del run["subsuite"]
|
|
configure_taskdesc_for_run(config, job, taskdesc, worker["implementation"])
|