summaryrefslogtreecommitdiffstats
path: root/taskcluster/gecko_taskgraph/transforms/job/python_test.py
blob: b57206121735424bfda9c27320dd8285ec52348a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# 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.util.schema import Schema
from voluptuous import Optional, Required

from gecko_taskgraph.transforms.job import configure_taskdesc_for_run, run_job_using

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"])