From 67abe16bcb5ebfcaf6f53dfe3f8dde45c4963efc Mon Sep 17 00:00:00 2001
From: Mihai Tabara <mtabara@mozilla.com>
Date: Fri, 17 Apr 2020 18:56:18 +0100
Subject: [PATCH] Bug 1630587 - fix mark-as-shipped task Fenix releases
 (#10021)

---
 taskcluster/ci/mark-as-shipped/kind.yml            |  8 +++++++-
 .../fenix_taskgraph/transforms/mark_as_shipped.py  | 14 ++++++++++++++
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/taskcluster/ci/mark-as-shipped/kind.yml b/taskcluster/ci/mark-as-shipped/kind.yml
index d6d291241..86226ddd5 100644
--- a/taskcluster/ci/mark-as-shipped/kind.yml
+++ b/taskcluster/ci/mark-as-shipped/kind.yml
@@ -24,7 +24,13 @@ job-template:
     description: Mark Fenix as shipped in ship-it
     worker-type: ship-it
     scopes:
-        - project:releng:ship-it:action:mark-as-shipped
+        by-level:
+            '3':
+                - project:releng:ship-it:action:mark-as-shipped
+                - project:releng:ship-it:server:production
+            default:
+                - project:releng:ship-it:action:mark-as-shipped
+                - project:releng:ship-it:server:staging
     worker: {}
 
     treeherder:
diff --git a/taskcluster/fenix_taskgraph/transforms/mark_as_shipped.py b/taskcluster/fenix_taskgraph/transforms/mark_as_shipped.py
index ac963b915..af44124fe 100644
--- a/taskcluster/fenix_taskgraph/transforms/mark_as_shipped.py
+++ b/taskcluster/fenix_taskgraph/transforms/mark_as_shipped.py
@@ -15,6 +15,20 @@ from taskgraph.util.schema import resolve_keyed_by
 transforms = TransformSequence()
 
 
+@transforms.add
+def resolve_keys(config, tasks):
+    for task in tasks:
+        resolve_keyed_by(
+            task,
+            'scopes',
+            item_name=task["name"],
+            **{
+                'level': config.params["level"],
+            }
+        )
+        yield task
+
+
 @transforms.add
 def make_task_description(config, jobs):
     for job in jobs: