From 26a029d407be480d791972afb5975cf62c9360a6 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Fri, 19 Apr 2024 02:47:55 +0200
Subject: Adding upstream version 124.0.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 taskcluster/test/test_generate_params.py | 62 ++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 taskcluster/test/test_generate_params.py

(limited to 'taskcluster/test/test_generate_params.py')

diff --git a/taskcluster/test/test_generate_params.py b/taskcluster/test/test_generate_params.py
new file mode 100644
index 0000000000..3fb479879b
--- /dev/null
+++ b/taskcluster/test/test_generate_params.py
@@ -0,0 +1,62 @@
+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()
-- 
cgit v1.2.3