From 38dd5f92e514573fccc8728839a31ca760ffa086 Mon Sep 17 00:00:00 2001
From: Mike Hommey <mh+mozilla@glandium.org>
Date: Sat, 15 Apr 2017 07:30:42 +0900
Subject: [PATCH] Bug 1356700 - Set MOZ_AUTOMATION on all taskcluster jobs.
 r=dustin

--HG--
extra : rebase_source : d55a62f6f31b6815e7f94f10761e4208ad03379b
---
 taskcluster/taskgraph/transforms/task.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/taskcluster/taskgraph/transforms/task.py b/taskcluster/taskgraph/transforms/task.py
index 299904ff667a..15dc15c34d01 100644
--- a/taskcluster/taskgraph/transforms/task.py
+++ b/taskcluster/taskgraph/transforms/task.py
@@ -955,6 +955,18 @@ def build_task(config, tasks):
         attributes = task.get('attributes', {})
         attributes['run_on_projects'] = task.get('run-on-projects', ['all'])
 
+        # Set MOZ_AUTOMATION on all jobs.
+        if task['worker']['implementation'] in (
+            'generic-worker',
+            'docker-engine',
+            'native-engine',
+            'docker-worker',
+        ):
+            payload = task_def.get('payload')
+            if payload:
+                env = payload.setdefault('env', {})
+                env['MOZ_AUTOMATION'] = '1'
+
         yield {
             'label': task['label'],
             'task': task_def,