Bug 1408145: Report comm-central revision to treeherder when building from a comm branch; r=dustin

MozReview-Commit-ID: A1HUqmit7ew

--HG--
extra : rebase_source : 665c2001b862081bb4d2270542852a9e403695e4

taskcluster/taskgraph/transforms/task.py

@@ -577,6 +577,17 @@ TREEHERDER_ROUTE_ROOTS = {
     'staging': 'tc-treeherder-stage',
 }
 
+# Which repository repository revision to use when reporting results to treeherder.
+DEFAULT_BRANCH_REV_PARAM = 'head_rev'
+BRANCH_REV_PARAM = {
+    'comm-esr45': 'comm_head_rev',
+    'comm-esr52': 'comm_head_rev',
+    'comm-beta': 'comm_head_rev',
+    'comm-central': 'comm_head_rev',
+    'comm-aurora': 'comm_head_rev',
+    'try-comm-central': 'comm_head_rev',
+}
+
 COALESCE_KEY = '{project}.{job-identifier}'
 SUPERSEDER_URL = 'https://coalesce.mozilla-releng.net/v1/list/{age}/{size}/{key}'
 
@@ -1163,10 +1174,15 @@ def build_task(config, tasks):
             treeherder['jobKind'] = task_th['kind']
             treeherder['tier'] = task_th['tier']
 
+            treeherder_rev = config.params[
+                BRANCH_REV_PARAM.get(
+                    config.params['project'],
+                    DEFAULT_BRANCH_REV_PARAM)]
+
             routes.extend([
                 '{}.v2.{}.{}.{}'.format(TREEHERDER_ROUTE_ROOTS[env],
                                         config.params['project'],
-                                        config.params['head_rev'],
+                                        treeherder_rev,
                                         config.params['pushlog_id'])
                 for env in task_th['environments']
             ])
@@ -1218,7 +1234,7 @@ def build_task(config, tasks):
         if task_th:
             # link back to treeherder in description
             th_push_link = 'https://treeherder.mozilla.org/#/jobs?repo={}&revision={}'.format(
-                config.params['project'], config.params['head_rev'])
+                config.params['project'], treeherder_rev)
             task_def['metadata']['description'] += ' ([Treeherder push]({}))'.format(
                 th_push_link)