Added the --screenshot option.

build/android/pylib/local/device/local_device_instrumentation_test_run.py

 # Copyright 2015 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 logging
 import re
 import time
 
 from devil.android import device_errors
 from devil.android import flag_changer
@@ skipping 245 matching lines @@
         if r.GetName() == test_name:
           r.SetName(test_display_name)
     if DidPackageCrashOnDevice(self._test_instance.test_package, device):
       for r in results:
         if r.GetType() == base_test_result.ResultType.UNKNOWN:
           r.SetType(base_test_result.ResultType.CRASH)
 
     if any(r.GetType() not in (base_test_result.ResultType.PASS,
                                base_test_result.ResultType.SKIP)
            for r in results):
+      if self._test_instance.screenshot_dir:
+        saved_dir = device.TakeScreenshot(self._test_instance.screenshot_dir)

[jbudorick, 2016/05/18 19:31:17]

Thought about this a bit more. Let's instead specify the full path here --
directory & file name -- and include the both the name of the first failed test
in results and the timestamp in the file name.

[BigBossZhiling, 2016/05/18 22:30:18]

Done.

+        logging.info('saved screenshot at %s', saved_dir)

[jbudorick, 2016/05/18 19:31:17]

nit: this should be

  logging.info('Saved screenshot for %s to %s', <test name>, <file path>)

[BigBossZhiling, 2016/05/18 22:30:18]

Done.

       logging.info('detected failure in %s. raw output:', test_display_name)
       for l in output:
         logging.info('  %s', l)
       if (not self._env.skip_clear_data
           and self._test_instance.package_info):
         permissions = (
             self._test_instance.apk_under_test.GetPermissions()
             if self._test_instance.apk_under_test
             else None)
         device.ClearApplicationState(self._test_instance.package_info.package,
@@ skipping 25 matching lines @@
         timeout = v
         break
     else:
       logging.warning('Using default 1 minute timeout for %s', test_name)
       timeout = 60
 
     timeout *= cls._GetTimeoutScaleFromAnnotations(annotations)
 
     return timeout