Properly update chromedriver.

tools/testing/webdriver_test_setup.py

 #!/usr/bin/python
 
 # Copyright (c) 2011, 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
 # of steps.
  
 # TODO(efortuna): Rewrite this script in Dart when the Process module has a
 # better high level API.
 import optparse
 import os
 import platform
 import re
+import shutil
 import subprocess
 import sys
 import urllib
 import urllib2
 import zipfile
 
 def run_cmd(cmd, stdin=None):
   """Run the command on the command line in the shell. We print the output of
   the command.
   """
@@ skipping 105 matching lines @@
         # 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 as used on Google Code.
     """
     os_str = 'win'
     if 'darwin' in sys.platform:
       os_str = 'mac'
     elif 'linux' in sys.platform:
       os_str = 'linux32'
@@ skipping 92 matching lines @@
       lambda x: 'chromedriver_%(os)s_%(version)s.zip' % x).run()
   if 'win32' not in sys.platform and 'cygwin' not in sys.platform:
     GoogleCodeInstaller('selenium', os.path.dirname(os.path.abspath(__file__)),
         lambda x: 'selenium-server-standalone-%(version)s.jar' % x).run()
 
   if args.firefox:
     FirefoxInstaller().run()
 
 if __name__ == '__main__':
   main()