[fenix] Bug 1635488 - part 2: Add missing branch in version bump payload (https://github.com/mozilla-mobile/fenix/pull/16650)

taskcluster/fenix_taskgraph/transforms/version_bump.py

@@ -34,7 +34,9 @@ def resolve_keys(config, tasks):
 @transforms.add
 def build_worker_definition(config, tasks):
     for task in tasks:
-        worker_definition = {}
+        worker_definition = {
+            "branch": config.params["head_ref"].decode("utf-8"),
+        }
         task["worker"].update(worker_definition)
 
         yield task

taskcluster/fenix_taskgraph/worker_types.py

@@ -172,6 +172,7 @@ def build_github_release_payload(config, task, task_def):
         Required("bump"): bool,
         Optional("bump-files"): [text_type],
         Optional("push"): bool,
+        Optional("branch"): text_type,
     },
 )
 def build_version_bump_payload(config, task, task_def):
@@ -196,3 +197,6 @@ def build_version_bump_payload(config, task, task_def):
 
     if worker.get('force-dry-run'):
         task_def['payload']['dry_run'] = True
+
+    if worker.get("branch"):
+        task_def["payload"]["branch"] = worker["branch"]