Fix run-task detection in the task transform

Author: Eijebong

src/taskgraph/transforms/task.py

@@ -649,7 +649,7 @@ def build_docker_worker_payload(config, task, task_def):
     if "max-run-time" in worker:
         payload["maxRunTime"] = worker["max-run-time"]
 
-    run_task = payload.get("command", [""])[0].endswith("run-task")
+    run_task = os.path.basename(payload.get("command", [""])[0]).startswith("run-task")
 
     # run-task exits EXIT_PURGE_CACHES if there is a problem with caches.
     # Automatically retry the tasks and purge caches if we see this exit

test/test_transforms_task.py

@@ -871,6 +871,36 @@ def test_default_expires_after(run_transform, graph_config, expires_after, test_
         assert task_dict["task"]["expires"] == {"relative-datestamp": "28 days"}
 
 
+@pytest.mark.parametrize(
+    "command",
+    (
+        ["run-task"],
+        ["run-task-hg"],
+        ["/usr/local/bin/run-task"],
+        ["/usr/bin/run-task-hg"],
+    ),
+)
+def test_run_task_exit_status(run_transform, make_transform_config, command):
+    task_def = {
+        "description": "fake description",
+        "name": "fake-task-name",
+        "worker-type": "t-linux",
+        "worker": {
+            "docker-image": "fake-image-name",
+            "max-run-time": 1800,
+            "command": command,
+        },
+    }
+
+    task_dict = run_transform(
+        task.transforms, task_def, config=make_transform_config()
+    )[0]
+    payload = task_dict["task"]["payload"]
+    # If run-task is properly detected, then the transform sets those onExitStatus
+    assert 72 in payload["onExitStatus"]["retry"]
+    assert 72 in payload["onExitStatus"]["purgeCaches"]
+
+
 @pytest.mark.parametrize(
     "test_task",
     (