Set nest step status when test failure for Chromedriver recipe.

scripts/slave/recipe_modules/chromedriver/api.py

 # Copyright 2016 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.
 
 from recipe_engine import recipe_api
 
 GS_CHROMEDRIVER_DATA_BUCKET = 'chromedriver-data'
 GS_PREBUILTS_URL = GS_CHROMEDRIVER_DATA_BUCKET + '/prebuilts'
 GS_SERVER_LOGS_URL = GS_CHROMEDRIVER_DATA_BUCKET + '/server_logs'
 
@@ skipping 113 matching lines @@
       cmd.extend(['--verbose'])
     if self.m.platform.is_linux:
       cmd = ['xvfb-run', '-a'] + cmd
     if android_package:
       cmd.extend(['--android-package', android_package])
     return cmd
 
   def _run_test(self, test_name, script_path, chromedriver,
                 ref_chromedriver=None, android_package=None, verbose=False,
                 archive_server_log=True, **kwargs): 
-    with self.m.step.nest(test_name):
+    with self.m.step.nest(test_name) as nest_step:
       with self.m.tempfile.temp_dir('server_log') as server_log_dir:
         build_number = self.m.properties['buildnumber']
         server_log = server_log_dir.join(
             ('%s_%s' % (test_name, build_number)).replace(' ', '_'))
-
-        self.m.step('Run test %s' % test_name,
-                    self._generate_test_command(
-                        script_path, chromedriver, server_log,
-                        ref_chromedriver=ref_chromedriver,
-                        android_package=android_package,
-                        verbose=verbose),
-                    **kwargs)
-        if archive_server_log:
-          self.archive_server_log(server_log)
+        try:
+          self.m.step('Run Tests',
+                      self._generate_test_command(
+                          script_path, chromedriver, server_log,
+                          ref_chromedriver=ref_chromedriver,
+                          android_package=android_package,
+                          verbose=verbose),
+                      **kwargs)
+        except self.m.step.StepFailure:
+          nest_step.presentation.status = self.m.step.FAILURE
+          raise
+        finally:
+          if archive_server_log:
+            self.archive_server_log(server_log)
 
   def run_python_tests(self, chromedriver, ref_chromedriver,
                        chrome_version_name=None, android_package=None,
                        archive_server_log=True, **kwargs):
     """Run the Chromedriver Python tests."""
     test_name = 'python_tests%s' % (
         ' %s' % chrome_version_name if chrome_version_name else '')
     test_script_path = self.m.path['checkout'].join(
         'chrome', 'test', 'chromedriver', 'test', 'run_py_tests.py')
     self._run_test(test_name, test_script_path, chromedriver,
@@ skipping 42 matching lines @@
                               chrome_version_name=package,
                               android_package=package,
                               env=test_env,
                               archive_server_log=archive_server_logs)
         self.run_java_tests(chromedriver,
                             chrome_version_name=package,
                             android_package=package,
                             verbose=True,
                             env=test_env,
                             archive_server_log=archive_server_logs)