Fix run-task detection in verify

Author: Eijebong

src/taskgraph/util/verify.py

@@ -4,6 +4,7 @@
 
 
 import logging
+import os
 import re
 import sys
 import warnings
@@ -333,7 +334,7 @@ def verify_run_task_caches(task, taskgraph, scratch_pad, graph_config, parameter
     command = payload.get("command") or [""]
 
     main_command = command[0] if isinstance(command[0], str) else ""
-    run_task = main_command.endswith("run-task")
+    run_task = os.path.basename(main_command).startswith("run-task")
 
     for cache in payload.get("cache", {}).get(
         "task-reference", payload.get("cache", {})

test/test_util_verify.py

@@ -302,6 +302,22 @@ def make_task_treeherder(label, symbol, platform="linux/opt"):
             pytest.raises(Exception, match="cache name is not dep"),
             id="using run-task without cache suffix",
         ),
+        pytest.param(
+            "verify_run_task_caches",
+            make_graph(
+                make_task(
+                    "task1",
+                    task_def={
+                        "payload": {
+                            "cache": {"test-domain-level-1-checkouts": {}},
+                            "command": ["run-task-hg"],
+                        }
+                    },
+                )
+            ),
+            pytest.raises(Exception, match="cache name is not dep"),
+            id="using run-task-hg without cache suffix",
+        ),
     ),
 )
 @pytest.mark.filterwarnings("error")