Updated webdriver setup script to use new location for downloads.

tools/testing/webdriver_test_setup.py

 #!/usr/bin/python
 
 # Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 # for details. All rights reserved. Use of this source code is governed by a
 # BSD-style license that can be found in the LICENSE file.
 
 # Run to install the necessary components to run webdriver on the buildbots or
 # on your local machine.
 # Note: The setup steps can be done fairly easily by hand. This script is
 # intended to simply and reduce the time for setup since there are a fair number
@@ skipping 63 matching lines @@
       depot_tools = os.path.join('e:', depot_tools)
     return depot_tools
   else:
     path = os.environ['PATH'].split(os.pathsep)
     for loc in path:
       if 'depot_tools' in loc:
         return loc
   raise Exception("Could not find depot_tools in your path.")
 
 
+class ChromeDriverInstaller(object):

[Emily Fortuna, 2013/10/21 20:29:46]

can this be refactored? can in it extend GoogleCodeInstaller? Seems like there's
repeated code for get_os_str and others.

[Andrei Mouravski, 2013/10/22 00:11:51]

Aye, I'll do that.

+  """Install ChromeDriver from Google Storage."""
+
+  def __init__(self, download_location):
+    """ Create an object that will install ChromeDriver from Google Storage.
+    Arguments:
+    download_location - Where to download the desired file on our filesystem.
+     """
+    self.download_location = download_location
+    self.project_name = 'chromedriver'
+    self.google_storage_downloads_page = 'http://chromedriver.storage.googleapis.com'

[Emily Fortuna, 2013/10/21 20:29:46]

80 char

+
+
+  def find_latest_version(self):
+    """Find the latest version number of ChromeDriver."""
+    f = urllib2.urlopen(self.google_storage_downloads_page)
+    latest = ''
+    l = f.read()
+    r = re.compile('(?:<Key>)(\d+\.\d+)')
+    v = max(r.findall(l))
+    return v
+
+  def run(self):
+    """Download and install the Google Code."""
+    print 'Installing from %s' % self.project_name
+    os_str = self.get_os_str
+    version = self.find_latest_version()
+    download_name = 'chromedriver_%s.zip' % os_str
+
+    urllib.urlretrieve(self.google_storage_downloads_page + '/' + version +
+        '/chromedriver_' + os_str + '.zip',
+        os.path.join(self.download_location, download_name))
+
+    if platform.system() != 'Windows':
+      # The Python zip utility does not preserve executable permissions, but
+      # this does not seem to be a problem for Windows, which does not have a
+      # built in zip utility. :-/
+      run_cmd('unzip -u %s -d %s' % (os.path.join(self.download_location,
+              download_name), self.download_location), stdin='y')
+    else:
+      z = zipfile.ZipFile(os.path.join(self.download_location, download_name))
+      z.extractall(self.download_location)
+      z.close()
+    os.remove(os.path.join(self.download_location, download_name))
+
+    chrome_path = os.path.join(os.path.dirname(os.path.abspath(__file__)),
+        'orig-chromedriver')
+    if self.project_name == 'chromedriver' and os.path.exists(chrome_path):
+      # We have one additional location to make sure chromedriver is updated.
+      # TODO(efortuna): Remove this. See move_chrome_driver_if_needed in
+      # perf_testing/run_perf_tests.py
+      driver = 'chromedriver'
+      if platform.system() == 'Windows':
+        driver += '.exe'
+      shutil.copy(os.path.join(self.download_location, driver),
+          os.path.join(chrome_path, driver))
+
+  @property
+  def get_os_str(self):
+    """The strings to indicate what OS a download is for.
+    """
+    os_str = 'win'
+    if 'darwin' in sys.platform:
+      os_str = 'mac'
+    elif 'linux' in sys.platform:
+      os_str = 'linux32'
+      if '64bit' in platform.architecture()[0]:
+        os_str = 'linux64'
+    if self.project_name == 'chromedriver' and (
+        os_str == 'mac' or os_str == 'win'):
+      os_str = os_str + '32'
+    return os_str
 class GoogleCodeInstaller(object):
   """Install code that is being hosted on Google Code."""
 
   def __init__(self, project_name, download_location, download_name_func):
     """ Create a object that will install code from a Google Code site.
     Arguments:
     project_name - The GoogleCode project name such as "selenium" or
     "chromedriver."
     download_location - Where to download the desired file on our filesystem.
     download_name_func - A function that takes a dictionary (currently with keys
@@ skipping 292 matching lines @@
       os_str = 'mac'
     elif 'linux' in sys.platform:
       os_str = 'linux'
     return os_str
 
 def main():
   args = parse_args()
   if not args.python:
     SeleniumBindingsInstaller(args.buildbot).run()
   if not args.chromedriver:
-    GoogleCodeInstaller('chromedriver',
-        find_depot_tools_location(args.buildbot),
-        lambda x: 'chromedriver_%(os)s_%(version)s.zip' % x).run()
+    ChromeDriverInstaller(find_depot_tools_location(args.buildbot)).run()
   if not args.seleniumrc:
     GoogleCodeInstaller('selenium', os.path.dirname(os.path.abspath(__file__)),
         lambda x: 'selenium-server-standalone-%(version)s.jar' % x).run()
   if not args.iedriver and platform.system() == 'Windows':
     GoogleCodeInstaller('selenium', find_depot_tools_location(args.buildbot),
         lambda x: 'IEDriverServer_Win32_%(version)s.zip' % x).run()
   if not args.firefox:
     FirefoxInstaller().run()
   if not args.opera:
     OperaInstaller().run()
 
 if __name__ == '__main__':
   main()