feat: allow setting per-task priority (#534)

* feat: allow setting per-task priority

The run and task transforms only allowed using the default per-project priority.

* style: pre-commit.ci auto fixes [...]

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: jcristau

src/taskgraph/transforms/run/__init__.py

@@ -49,6 +49,7 @@
         # possibly modified by the run implementation.  See
         # taskcluster/taskgraph/transforms/task.py for the schema details.
         Required("description"): task_description_schema["description"],
+        Optional("priority"): task_description_schema["priority"],
         Optional("attributes"): task_description_schema["attributes"],
         Optional("task-from"): task_description_schema["task-from"],
         Optional("dependencies"): task_description_schema["dependencies"],

src/taskgraph/transforms/task.py

@@ -72,6 +72,15 @@ def _run_task_suffix():
                 ),
             ): object,
         },
+        Optional("priority"): Any(
+            "highest",
+            "very-high",
+            "high",
+            "medium",
+            "low",
+            "very-low",
+            "lowest",
+        ),
         # Soft dependencies of this task, as a list of tasks labels
         Optional("soft-dependencies"): [str],
         # Dependencies that must be scheduled in order for this task to run.

test/test_transforms_task.py

@@ -942,3 +942,70 @@ def test_default_expires_after(run_transform, graph_config, expires_after, test_
         assert task_dict["task"]["expires"] == {"relative-datestamp": expires_after}
     else:
         assert task_dict["task"]["expires"] == {"relative-datestamp": "28 days"}
+
+
+@pytest.mark.parametrize(
+    "test_task",
+    (
+        {
+            "description": "fake description",
+            "name": "fake-task-name",
+            "worker-type": "t-linux",
+            "worker": {
+                "docker-image": "fake-image-name",
+                "max-run-time": 1800,
+            },
+        },
+        {
+            "description": "fake description",
+            "name": "fake-task-name",
+            "worker-type": "t-linux",
+            "worker": {
+                "docker-image": "fake-image-name",
+                "max-run-time": 1800,
+            },
+            "priority": "high",
+        },
+    ),
+)
+def test_task_priority(run_transform, graph_config, test_task):
+    params = FakeParameters(
+        {
+            "base_repository": "[email protected]://github.com/mozilla/example.git",
+            "build_date": 0,
+            "build_number": 1,
+            "head_repository": "[email protected]://github.com/mozilla/example.git",
+            "head_rev": "abcdef",
+            "head_ref": "default",
+            "level": "1",
+            "moz_build_date": 0,
+            "next_version": "1.0.1",
+            "owner": "some-owner",
+            "project": "some-project",
+            "pushlog_id": 1,
+            "repository_type": "git",
+            "target_tasks_method": "test_method",
+            "tasks_for": "github-pull-request",
+            "try_mode": None,
+            "version": "1.0.0",
+        },
+    )
+
+    transform_config = TransformConfig(
+        "check_priority",
+        str(here),
+        {},
+        params,
+        {},
+        graph_config,
+        write_artifacts=False,
+    )
+
+    task_dict = deepcopy(test_task)
+
+    task_dict = run_transform(task.transforms, task_dict, config=transform_config)[0]
+    priority = test_task.get("priority")
+    if priority:
+        assert task_dict["task"]["priority"] == priority
+    else:
+        assert task_dict["task"]["priority"] == graph_config["task-priority"]