Bug 1356700 - Set MOZ_AUTOMATION on all taskcluster jobs. r=dustin

--HG--
extra : rebase_source : d55a62f6f31b6815e7f94f10761e4208ad03379b

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,