Retreive screenshots captured by tests from test devices.

build/android/pylib/instrumentation/instrumentation_test_instance.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 collections
 import copy
 import json
 import logging
 import os
 import pickle
@@ skipping 650 matching lines @@
     if os.path.exists(self._driver_apk):
       driver_apk = apk_helper.ApkHelper(self._driver_apk)
       self._driver_package = driver_apk.GetPackageName()
       self._driver_name = driver_apk.GetInstrumentationName()
     else:
       self._driver_apk = None
 
   def _initializeTestControlAttributes(self, args):
     self._screenshot_dir = args.screenshot_dir
     self._timeout_scale = args.timeout_scale or 1
+    self._ui_screenshot_dir = args.ui_screenshot_dir
 
   def _initializeTestCoverageAttributes(self, args):
     self._coverage_directory = args.coverage_dir
 
   def _initializeTombstonesAttributes(self, args):
     self._store_tombstones = args.store_tombstones
 
   def _initializeLogAttributes(self, args):
     self._gs_results_bucket = args.gs_results_bucket
     self._should_save_logcat = bool(args.json_results_file)
@@ skipping 113 matching lines @@
     return self._test_runner_junit4
 
   @property
   def timeout_scale(self):
     return self._timeout_scale
 
   @property
   def total_external_shards(self):
     return self._total_external_shards
 
+  @property
+  def ui_screenshot_dir(self):
+    return self._ui_screenshot_dir
+
   #override
   def TestType(self):
     return 'instrumentation'
 
   #override
   def SetUp(self):
     self._data_deps.extend(
         self._data_deps_delegate(self._runtime_deps_path))
 
   def GetDataDependencies(self):
@@ skipping 67 matching lines @@
 
   @staticmethod
   def GenerateTestResults(
       result_code, result_bundle, statuses, start_ms, duration_ms):
     return GenerateTestResults(result_code, result_bundle, statuses,
                                start_ms, duration_ms)
 
   #override
   def TearDown(self):
     pass