chromium_tests: apply chromium.get_env() when running tests

scripts/slave/recipe_modules/chromium_tests/api.py

 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import collections
 import contextlib
 import copy
 import itertools
 import json
 
@@ skipping 511 matching lines @@
 
   def _make_legacy_build_url(self, master_config, mastername):
     return self.m.archive.legacy_download_url(
                master_config.get('build_gs_bucket'),
                extra_url_components=(
                    None if mastername.startswith('chromium.perf')
                    else self.m.properties['mastername']))
 
   @contextlib.contextmanager
   def wrap_chromium_tests(self, bot_config, tests=None):
-    bot_type = bot_config.get('bot_type', 'builder_tester')
+    with self.m.step.context({'env': self.m.chromium.get_env()}):
+      bot_type = bot_config.get('bot_type', 'builder_tester')
 
-    if bot_type in ('tester', 'builder_tester'):
-      isolated_targets = [
-          t.isolate_target(self.m) for t in tests if t.uses_swarming]
-      if isolated_targets:
-        self.m.isolate.find_isolated_tests(self.m.chromium.output_dir)
+      if bot_type in ('tester', 'builder_tester'):
+        isolated_targets = [
+            t.isolate_target(self.m) for t in tests if t.uses_swarming]
+        if isolated_targets:
+          self.m.isolate.find_isolated_tests(self.m.chromium.output_dir)
 
-    if bot_type == 'tester':
-      if (self.m.chromium.c.TARGET_PLATFORM == 'android' and
-          bot_config.get('root_devices')):
-        self.m.adb.root_devices()
+      if bot_type == 'tester':
+        if (self.m.chromium.c.TARGET_PLATFORM == 'android' and
+            bot_config.get('root_devices')):
+          self.m.adb.root_devices()
 
-    # Some recipes use this wrapper to setup devices and have their own way
-    # to run tests. If platform is Android and tests is None, run device steps.
-    require_device_steps = (tests is None or
-                            any([t.uses_local_devices for t in tests]))
+      # Some recipes use this wrapper to setup devices and have their own way
+      # to run tests. If platform is Android and tests is None, run device steps.
+      require_device_steps = (tests is None or
+                              any([t.uses_local_devices for t in tests]))
 
-    if self.m.chromium.c.TARGET_PLATFORM == 'android' and require_device_steps:
-      #TODO(prasadv): Remove this hack and implement specific functions
-      # at the point of call.
-      remove_system_webview = bot_config.get('remove_system_webview')
-      perf_setup = bot_config.matches_any_bot_id(lambda bot_id:
-          bot_id['mastername'].startswith('chromium.perf') or
-          bot_id['mastername'].startswith('tryserver.chromium.perf'))
-      self.m.chromium_android.common_tests_setup_steps(
-          perf_setup=perf_setup,
-          remove_system_webview=remove_system_webview)
+      if self.m.chromium.c.TARGET_PLATFORM == 'android' and require_device_steps:
+        #TODO(prasadv): Remove this hack and implement specific functions
+        # at the point of call.
+        remove_system_webview = bot_config.get('remove_system_webview')
+        perf_setup = bot_config.matches_any_bot_id(lambda bot_id:
+            bot_id['mastername'].startswith('chromium.perf') or
+            bot_id['mastername'].startswith('tryserver.chromium.perf'))
+        self.m.chromium_android.common_tests_setup_steps(
+            perf_setup=perf_setup,
+            remove_system_webview=remove_system_webview)
 
-    try:
-      yield
-    finally:
-      if self.m.platform.is_win:
-        self.m.chromium.process_dumps()
+      try:
+        yield
+      finally:
+        if self.m.platform.is_win:
+          self.m.chromium.process_dumps()
 
-      if self.m.chromium.c.TARGET_PLATFORM == 'android':
-        if require_device_steps:
-          self.m.chromium_android.common_tests_final_steps(
-              logcat_gs_bucket='chromium-android')
-        else:
-          self.m.chromium_android.test_report()
+        if self.m.chromium.c.TARGET_PLATFORM == 'android':
+          if require_device_steps:
+            self.m.chromium_android.common_tests_final_steps(
+                logcat_gs_bucket='chromium-android')
+          else:
+            self.m.chromium_android.test_report()
 
   def _resolve_fixed_revisions(self, bot_update_json):
     """Set all fixed revisions from the first sync to their respective
     got_X_revision values.
 
     If on the first sync, a revision was requested to be HEAD, this avoids
     using HEAD potentially resolving to a different revision on the second
     sync. Instead, we sync explicitly to whatever was checked out the first
     time.
 
@@ skipping 255 matching lines @@
   def get_compile_targets_for_scripts(self):
     return self.m.python(
         name='get compile targets for scripts',
         script=self.m.path['checkout'].join(
             'testing', 'scripts', 'get_compile_targets.py'),
         args=[
             '--output', self.m.json.output(),
             '--',
         ] + self.get_common_args_for_scripts(),
         step_test_data=lambda: self.m.json.test_api.output({}))