feat: expose RUN_ID and TASKCLUSTER_ROOT_URL to scriptworkers

This will be needed to fix https://bugzilla.mozilla.org/show_bug.cgi?id=1967787.

Author: bhearsum

src/scriptworker/task.py

@@ -657,6 +657,8 @@ async def run_task(context, to_cancellable_process):
     """
     env = deepcopy(os.environ)
     env["TASK_ID"] = context.task_id or "None"
+    env["RUN_ID"] = get_run_id(context.claim_task)
+    env["TASKCLUSTER_ROOT_URL"] = context.config["taskcluster_root_url"]
     kwargs = {"stdout": PIPE, "stderr": PIPE, "stdin": None, "close_fds": True, "preexec_fn": lambda: os.setsid(), "env": env}  # pragma: no branch
 
     subprocess = await asyncio.create_subprocess_exec(*context.config["task_script"], **kwargs)

tests/test_task.py

@@ -41,7 +41,8 @@ def mobile_context(mobile_rw_context):
 def _craft_context(rw_context):
     rw_context.config["reclaim_interval"] = 0.001
     rw_context.config["task_max_timeout"] = 1
-    rw_context.config["task_script"] = ("bash", "-c", ">&2 echo bar && echo foo && exit 1")
+    rw_context.config["taskcluster_root_url"] = "https://tc"
+    rw_context.config["task_script"] = ("bash", "-c", ">&2 echo $TASK_ID && echo $RUN_ID && echo $TASKCLUSTER_ROOT_URL && exit 1")
     rw_context.claim_task = {
         "credentials": {"a": "b"},
         "status": {"taskId": "taskId"},
@@ -496,7 +497,7 @@ def test_prepare_to_run_task(context):
 async def test_run_task(context):
     status = await swtask.run_task(context, noop_to_cancellable_process)
     log_file = log.get_log_filename(context)
-    assert read(log_file) in ("bar\nfoo\nexit code: 1\n", "foo\nbar\nexit code: 1\n")
+    assert read(log_file) in ("taskId\nrunId\nhttps://tc\nexit code: 1\n", "taskId\nrunId\nhttps://tc\nexit code: 1\n")
     assert status == 1
 
 

tests/test_worker.py

@@ -25,7 +25,14 @@
 # constants helpers and fixtures {{{1
 @pytest.fixture(scope="function")
 def context(rw_context):
+    rw_context.config["taskcluster_root_url"] = "https://tc"
     rw_context.credentials_timestamp = arrow.utcnow().shift(minutes=-10).int_timestamp
+    rw_context.claim_task = {
+        "credentials": {"a": "b"},
+        "status": {"taskId": "taskId"},
+        "task": {"dependencies": ["dependency1", "dependency2"], "taskGroupId": "dependency0"},
+        "runId": "runId",
+    }
     yield rw_context