diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
commit | 36d22d82aa202bb199967e9512281e9a53db42c9 (patch) | |
tree | 105e8c98ddea1c1e4784a60a5a6410fa416be2de /taskcluster/test/test_generate_params.py | |
parent | Initial commit. (diff) | |
download | firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip |
Adding upstream version 115.7.0esr.upstream/115.7.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'taskcluster/test/test_generate_params.py')
-rw-r--r-- | taskcluster/test/test_generate_params.py | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/taskcluster/test/test_generate_params.py b/taskcluster/test/test_generate_params.py new file mode 100644 index 0000000000..87f1fe341c --- /dev/null +++ b/taskcluster/test/test_generate_params.py @@ -0,0 +1,57 @@ +import json +import os +import subprocess + +import pytest +from gecko_taskgraph import GECKO +from mozunit import main +from taskgraph.taskgraph import TaskGraph + +pytestmark = pytest.mark.slow +PARAMS_DIR = os.path.join(GECKO, "taskcluster", "test", "params") + + +@pytest.fixture(scope="module") +def get_graph_from_spec(tmpdir_factory): + outdir = tmpdir_factory.mktemp("graphs") + + # Use a mach subprocess to leverage the auto parallelization of + # parameters when specifying a directory. + cmd = [ + "./mach", + "taskgraph", + "morphed", + "--json", + f"--parameters={PARAMS_DIR}", + f"--output-file={outdir}/graph.json", + ] + subprocess.run(cmd, cwd=GECKO) + assert len(outdir.listdir()) > 0 + + def inner(param_spec): + outfile = f"{outdir}/graph_{param_spec}.json" + with open(outfile) as fh: + output = fh.read() + try: + return TaskGraph.from_json(json.loads(output))[1] + except ValueError: + return output + + return inner + + +@pytest.mark.parametrize( + "param_spec", [os.path.splitext(p)[0] for p in os.listdir(PARAMS_DIR)] +) +def test_generate_graphs(get_graph_from_spec, param_spec): + ret = get_graph_from_spec(param_spec) + if isinstance(ret, str): + print(ret) + pytest.fail("An exception was raised during graph generation!") + + assert isinstance(ret, TaskGraph) + assert len(ret.tasks) > 0 + + +if __name__ == "__main__": + main() |