commit ac9bd776a41cc591eb47c9d588f236a9aeb7b218
author Manoj Gupta <manojgupta@google.com> Tue Nov 15 13:47:52 2016 -0800
committer chrome-bot <chrome-bot@chromium.org> Tue Nov 15 16:52:59 2016 -0800
tree e2f839472b1152bf314b851eafe817ed7006b76b
parent b187da39bb5da01bcb729c0bc2b2921c581fa7e3

Do no exit or raise exception if autotest files are not found.

If autotest files are missing from server, just emit an warning
and continue.

BUG=chromium:665483
TEST=crosperf unit tests

Change-Id: Ieaec9b53cbd270bc50f2c5b412b153ddc568104d
Reviewed-on: https://chrome-internal-review.googlesource.com/304808
Commit-Ready: Manoj Gupta <manojgupta@google.com>
Tested-by: Manoj Gupta <manojgupta@google.com>
Reviewed-by: Caroline Tice <cmtice@google.com>
Reviewed-by: Yunlian Jiang <yunlian@google.com>

crosperf/download_images.py

diff --git a/crosperf/download_images.py b/crosperf/download_images.py
index 14a9142..a2a7800 100644
--- a/crosperf/download_images.py
+++ b/crosperf/download_images.py
@@ -162,12 +162,39 @@
     if retval != 0:
       print('(Warning: Could not remove file %s .)' % package_file_name)
 
+  def VerifyAutotestFilesExist(self, chromeos_root, build_id, package_file):
+    # Quickly verify if the files are there
+    status = 0
+    gs_package_name = ('gs://chromeos-image-archive/%s/%s' %
+                       (build_id, package_file))
+    if not test_flag.GetTestMode():
+      cmd = 'gsutil ls %s' % gs_package_name
+      if self.log_level != 'verbose':
+        self._logger.LogOutput('CMD: %s' % cmd)
+      status = self._ce.ChrootRunCommand(chromeos_root, cmd)
+      if status != 0:
+        print('(Warning: Could not find file %s )' % gs_package_name)
+        return 1
+    # Package exists on server
+    return 0
+
   def DownloadAutotestFiles(self, chromeos_root, build_id):
     # Download autest package files (3 files)
     autotest_packages_name = ('autotest_packages.tar')
     autotest_server_package_name = ('autotest_server_package.tar.bz2')
     autotest_control_files_name = ('control_files.tar')
 
+    # Quickly verify if the files are there
+    # If not, just exit with warning
+    status = self.VerifyAutotestFilesExist(chromeos_root, build_id,
+                                      autotest_packages_name)
+    if status != 0:
+      default_autotest_dir = '~/trunk/src/third_party/autotest/files'
+      print('(Warning: Could not download autotest packages .)\n'
+            '(Warning: Defaulting autotest path to %s .' %
+            default_autotest_dir)
+      return default_autotest_dir
+
     # Autotest directory relative path wrt chroot
     autotest_rel_path = os.path.join('/tmp', build_id, 'autotest_files')
     # Absolute Path to download files