forked from mirrors/gecko-dev
757f0bf16a
These were not caught due to bug 1865672. Depends on D195431 Differential Revision: https://phabricator.services.mozilla.com/D195432
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
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",
|
|
]
|
|
# unset MOZ_AUTOMATION so we don't attempt to optimize out the graph
|
|
# entirely as having already run
|
|
env = os.environ.copy()
|
|
env.pop("MOZ_AUTOMATION", None)
|
|
subprocess.run(cmd, cwd=GECKO, env=env)
|
|
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) if p.endswith(".yml")],
|
|
)
|
|
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()
|