Make android-chrome bisect work with archives.

tools/auto_bisect/fetch_build.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.
 
 """This module contains functions for fetching and extracting archived builds.
 
 The builds may be stored in different places by different types of builders;
 for example, builders on tryserver.chromium.perf stores builds in one place,
 while builders on chromium.linux store builds in another.
 
@@ skipping 19 matching lines @@
 # Possible builder types.
 PERF_BUILDER = 'perf'
 FULL_BUILDER = 'full'
 ANDROID_CHROME_PERF_BUILDER = 'android-chrome-perf'
 
 # Maximum time in seconds to wait after posting build request to the try server.
 MAX_MAC_BUILD_TIME = 14400
 MAX_WIN_BUILD_TIME = 14400
 MAX_LINUX_BUILD_TIME = 14400
 
+# Try server status page URLs, used to get build status.
+PERF_TRY_SERVER_URL = 'http://build.chromium.org/p/tryserver.chromium.perf'
+LINUX_TRY_SERVER_URL = 'http://build.chromium.org/p/tryserver.chromium.linux'
+
 
 def GetBucketAndRemotePath(revision, builder_type=PERF_BUILDER,
                            target_arch='ia32', target_platform='chromium',
                            deps_patch_sha=None, extra_src=None):
   """Returns the location where a build archive is expected to be.
 
   Args:
     revision: Revision string, e.g. a git commit hash or SVN revision.
     builder_type: Type of build archive.
     target_arch: Architecture, e.g. "ia32".
@@ skipping 20 matching lines @@
                                target_platform='chromium', extra_src=None):
   """Gets builder bot name and build time in seconds based on platform."""
   logging.info('Getting builder name for builder "%s", "%s", "%s".',
                builder_type, target_arch, target_platform)
   build_archive = BuildArchive.Create(
       builder_type, target_arch=target_arch, target_platform=target_platform,
       extra_src=extra_src)
   return build_archive.GetBuilderName(), build_archive.GetBuilderBuildTime()
 
 
+def GetBuildBotUrl(builder_type=PERF_BUILDER, target_arch='ia32',
+                   target_platform='chromium', extra_src=None):
+  """Gets buildbot URL for a given builder type."""
+  logging.info('Getting buildbot URL for "%s", "%s", "%s".',
+               builder_type, target_arch, target_platform)
+  build_archive = BuildArchive.Create(
+      builder_type, target_arch=target_arch, target_platform=target_platform,
+      extra_src=extra_src)
+  return build_archive.GetBuildBotUrl()
+
+
 class BuildArchive(object):
   """Represents a place where builds of some type are stored.
 
   There are two pieces of information required to locate a file in Google
   Cloud Storage, bucket name and file path. Subclasses of this class contain
   specific logic about which bucket names and paths should be used to fetch
   a build.
   """
 
   @staticmethod
@@ skipping 12 matching lines @@
             target_arch, target_platform, loaded_extra_src)
       except (IOError, TypeError, ImportError):
         raise RuntimeError('Invalid or missing --extra_src. [%s]' % extra_src)
     raise NotImplementedError('Builder type "%s" not supported.' % builder_type)
 
   def __init__(self, target_arch='ia32', target_platform='chromium',
                extra_src=None):
     self._extra_src = extra_src
     if bisect_utils.IsLinuxHost() and target_platform == 'android':
       self._platform = 'android'
+    elif bisect_utils.IsLinuxHost() and target_platform == 'android-chrome':
+      self._platform = 'android-chrome'
     elif bisect_utils.IsLinuxHost():
       self._platform = 'linux'
     elif bisect_utils.IsMacHost():
       self._platform = 'mac'
     elif bisect_utils.Is64BitWindows() and target_arch == 'x64':
       self._platform = 'win64'
     elif bisect_utils.IsWindowsHost():
       self._platform = 'win'
     else:
       raise NotImplementedError('Unknown platform "%s".' % sys.platform)
@@ skipping 46 matching lines @@
       return 'mac'
     raise NotImplementedError('Unknown platform "%s".' % sys.platform)
 
   def GetBuilderName(self):
     raise NotImplementedError()
 
   def GetBuilderBuildTime(self):
     """Returns the time to wait for a build after requesting one."""
     if self._platform in ('win', 'win64'):
       return MAX_WIN_BUILD_TIME
-    if self._platform in ('linux', 'android'):
+    if self._platform in ('linux', 'android', 'android-chrome'):
       return MAX_LINUX_BUILD_TIME
     if self._platform == 'mac':
       return MAX_MAC_BUILD_TIME
     raise NotImplementedError('Unsupported Platform "%s".' % sys.platform)
 
+  def GetBuildBotUrl(self):
+    raise NotImplementedError()
+
 
 class PerfBuildArchive(BuildArchive):
 
   def BucketName(self):
     return 'chrome-perf'
 
   def FilePath(self, revision, deps_patch_sha=None):
     return '%s/%s' % (self._ArchiveDirectory(),
                       self._ZipFileName(revision, deps_patch_sha))
 
@@ skipping 16 matching lines @@
     elif self._platform == 'win':
       return 'win_perf_bisect_builder'
     elif self._platform == 'linux':
       return 'linux_perf_bisect_builder'
     elif self._platform == 'android':
       return 'android_perf_bisect_builder'
     elif self._platform == 'mac':
       return 'mac_perf_bisect_builder'
     raise NotImplementedError('Unsupported platform "%s".' % sys.platform)
 
+  def GetBuildBotUrl(self):
+    """Returns buildbot URL for fetching build info."""
+    return PERF_TRY_SERVER_URL
+
 
 class FullBuildArchive(BuildArchive):
 
   def BucketName(self):
     platform_to_bucket = {
         'android': 'chromium-android',
         'linux': 'chromium-linux-archive',
         'mac': 'chromium-mac-archive',
         'win64': 'chromium-win-archive',
         'win': 'chromium-win-archive',
@@ skipping 16 matching lines @@
     }
     assert self._platform in platform_to_directory
     return platform_to_directory.get(self._platform)
 
   def GetBuilderName(self):
     """Gets builder bot name based on platform."""
     if self._platform == 'linux':
       return 'linux_full_bisect_builder'
     raise NotImplementedError('Unsupported platform "%s".' % sys.platform)
 
+  def GetBuildBotUrl(self):
+    """Returns buildbot URL for fetching build info."""
+    return LINUX_TRY_SERVER_URL
+
+
 class AndroidChromeBuildArchive(BuildArchive):
   """Represents a place where builds of android-chrome type are stored.
 
   If AndroidChromeBuildArchive is used, it is assumed that the --extra_src
   is a valid Python module which contains the module-level functions
   GetBucketName and GetArchiveDirectory.
   """
 
   def BucketName(self):
     return self._extra_src.GetBucketName()
@@ skipping 20 matching lines @@
                       self._ZipFileName(revision, deps_patch_sha))
 
   def _ArchiveDirectory(self):
     """Returns the directory name to download builds from."""
     return self._extra_src.GetArchiveDirectory()
 
   def GetBuilderName(self):
     """Returns the builder name extra source."""
     return self._extra_src.GetBuilderName()
 
+  def GetBuildBotUrl(self):
+    """Returns buildbot URL for fetching build info."""
+    return self._extra_src.GetBuildBotUrl()
+
 
 def BuildIsAvailable(bucket_name, remote_path):
   """Checks whether a build is currently archived at some place."""
   logging.info('Checking existence: gs://%s/%s' % (bucket_name, remote_path))
   try:
     exists = cloud_storage.Exists(bucket_name, remote_path)
     logging.info('Exists? %s' % exists)
     return exists
   except cloud_storage.CloudStorageError:
     return False
@@ skipping 155 matching lines @@
     print 'Build is not available.'
     return 1
 
   FetchFromCloudStorage(bucket_name, remote_path, args.output_dir)
   print 'Build has been downloaded to and extracted in %s.' % args.output_dir
   return 0
 
 
 if __name__ == '__main__':
   sys.exit(Main(sys.argv))