From 96f02619db02f90f232a2fd6e35094a929edafba Mon Sep 17 00:00:00 2001
From: Colin Lee <mncolinlee@gmail.com>
Date: Fri, 2 Aug 2019 14:39:11 -0500
Subject: [PATCH] For #4507 Enable codecov

---
 automation/taskcluster/decision_task.py       |  2 +-
 automation/taskcluster/lib/tasks.py           | 25 +++++++++++++++----
 .../taskcluster/upload_coverage_report.sh     | 19 ++++++++++++++
 codecov.yml                                   |  8 ++++++
 4 files changed, 48 insertions(+), 6 deletions(-)
 create mode 100755 automation/taskcluster/upload_coverage_report.sh
 create mode 100644 codecov.yml

diff --git a/automation/taskcluster/decision_task.py b/automation/taskcluster/decision_task.py
index cacb1d96dd87..1f54b46c9676 100644
--- a/automation/taskcluster/decision_task.py
+++ b/automation/taskcluster/decision_task.py
@@ -63,7 +63,7 @@ def pr():
 
     for variant in get_variants_for_build_type('debug'):
         assemble_task_id = taskcluster.slugId()
-        build_tasks[assemble_task_id] = BUILDER.craft_assemble_task(variant)
+        build_tasks[assemble_task_id] = BUILDER.craft_assemble_pr_task(variant, True)
         build_tasks[taskcluster.slugId()] = BUILDER.craft_test_task(variant)
 
     for craft_function in (
diff --git a/automation/taskcluster/lib/tasks.py b/automation/taskcluster/lib/tasks.py
index aee5808f0a16..1071836a30f7 100644
--- a/automation/taskcluster/lib/tasks.py
+++ b/automation/taskcluster/lib/tasks.py
@@ -129,11 +129,22 @@ def craft_assemble_raptor_task(self, variant):
             },
         )
 
-    def craft_assemble_task(self, variant):
+    def craft_assemble_pr_task(self, variant, run_coverage=False):
+        coverage_flag = '-Pcoverage ' if run_coverage else ''
+        assemble_gradle_command = ('{} assemble{}'.format(coverage_flag, variant.for_gradle_command),)
+        post_gradle_command = ('automation/taskcluster/upload_coverage_report.sh' if run_coverage else '',)
+
+        command = ' && '.join(
+            cmd
+            for commands in (assemble_gradle_command, post_gradle_command)
+            for cmd in commands
+            if cmd
+        )
+
         return self._craft_clean_gradle_task(
             name='assemble: {}'.format(variant.raw),
             description='Building and testing variant {}'.format(variant.raw),
-            gradle_task='assemble{}'.format(variant.for_gradle_command),
+            gradle_task=command,
             artifacts=_craft_artifacts_from_variant(variant),
             treeherder={
                 'groupSymbol': variant.build_type,
@@ -144,6 +155,9 @@ def craft_assemble_task(self, variant):
                 'symbol': 'A',
                 'tier': 1,
             },
+            scopes=[
+                'secrets:get:project/mobile/fenix/pr'
+            ]
         )
 
     def craft_test_task(self, variant):
@@ -299,7 +313,7 @@ def craft_dependencies_task(self):
         )
 
     def _craft_clean_gradle_task(
-        self, name, description, gradle_task, artifacts=None, routes=None, treeherder=None
+        self, name, description, gradle_task, artifacts=None, routes=None, treeherder=None, scopes=None
     ):
         return self._craft_build_ish_task(
             name=name,
@@ -308,6 +322,7 @@ def _craft_clean_gradle_task(
             artifacts=artifacts,
             routes=routes,
             treeherder=treeherder,
+            scopes=scopes,
         )
 
     def craft_compare_locales_task(self):
@@ -329,8 +344,8 @@ def craft_compare_locales_task(self):
         )
 
     def _craft_build_ish_task(
-        self, name, description, command, dependencies=None, artifacts=None, scopes=None,
-        routes=None, treeherder=None, env_vars=None,
+        self, name, description, command, dependencies=None, artifacts=None,
+        routes=None, treeherder=None, env_vars=None, scopes=None
     ):
         dependencies = [] if dependencies is None else dependencies
         artifacts = {} if artifacts is None else artifacts
diff --git a/automation/taskcluster/upload_coverage_report.sh b/automation/taskcluster/upload_coverage_report.sh
new file mode 100755
index 000000000000..137b1fa6bdd7
--- /dev/null
+++ b/automation/taskcluster/upload_coverage_report.sh
@@ -0,0 +1,19 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+# If a command fails then do not proceed and fail this script too.
+set -ex
+
+# Get token for uploading to codecov and append it to codecov.yml
+python automation/taskcluster/helper/get-secret.py \
+    -s project/mobile/fenix/pr \
+    -k codecov \
+    -f .cc_token \
+
+# Set some environment variables that will help codecov detect the CI
+export CI_BUILD_URL="https://tools.taskcluster.net/tasks/$TASK_ID"
+
+# Execute codecov script for uploading report
+# bash <(curl -s https://codecov.io/bash)
+bash <(curl -s https://codecov.io/bash) -t @.cc_token
diff --git a/codecov.yml b/codecov.yml
new file mode 100644
index 000000000000..733485b27af3
--- /dev/null
+++ b/codecov.yml
@@ -0,0 +1,8 @@
+codecov:
+  ci:
+    - tools.taskcluster.net
+
+coverage:
+  precision: 2
+  round: down
+  range: 60..80