[chromedriver] Only delete temp files that start with "chromedriver_".

chrome/test/chromedriver/run_buildbot_steps.py

 #!/usr/bin/env python
 # Copyright (c) 2013 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.
 
 """Runs all the buildbot steps for ChromeDriver except for update/compile."""
 
 import bisect
 import csv
 import datetime
@@ skipping 367 matching lines @@
   if slave_utils.GSUtilCopy(temp_latest_release_fname, latest_release_url,
                             mimetype='text/plain'):
     util.MarkBuildStepError()
 
 
 def _CleanTmpDir():
   tmp_dir = tempfile.gettempdir()
   print 'cleaning temp directory:', tmp_dir
   for file_name in os.listdir(tmp_dir):
     file_path = os.path.join(tmp_dir, file_name)
-    if os.path.isdir(file_path):
-      print 'deleting sub-directory', file_path
-      shutil.rmtree(file_path, True)
     if file_name.startswith('chromedriver_'):
-      print 'deleting file', file_path
-      os.remove(file_path)
+      if os.path.isdir(file_path):
+        print 'deleting sub-directory', file_path
+        shutil.rmtree(file_path, True)
+      else:
+        print 'deleting file', file_path
+        os.remove(file_path)
 
 
 def _GetCommitPositionFromGitHash(snapshot_hashcode):
   json_url = GS_GIT_LOG_URL % snapshot_hashcode
   try:
     response = urllib2.urlopen(json_url)
   except urllib2.HTTPError as error:
     util.PrintAndFlush('HTTP Error %d' % error.getcode())
     return None
   except urllib2.URLError as error:
@@ skipping 86 matching lines @@
 
   bitness = '32'
   if util.IsLinux() and platform_module.architecture()[0] == '64bit':
     bitness = '64'
   platform = '%s%s' % (util.GetPlatformName(), bitness)
   if options.android_packages:
     platform = 'android'
 
   _CleanTmpDir()
 
+  # Make sure any subprocesses (i.e. chromedriver) create temp files under
+  # $TMPDIR/chromedriver_*, so that they can be cleaned up by _CleanTempDir().
+  if util.IsWindows():
+    os.environ['TMP'] = os.environ['TEMP'] = util.MakeTempDir()
+  else:
+    os.environ['TMPDIR'] = util.MakeTempDir()
+
   if not options.revision:
     commit_position = None
   else:
     commit_position = _GetCommitPositionFromGitHash(options.revision)
 
   if platform == 'android':
     if not options.revision and options.update_log:
       parser.error('Must supply a --revision with --update-log')
     _DownloadPrebuilts()
   else:
@@ skipping 30 matching lines @@
     util.MarkBuildStepStart('run_all_tests.py')
     util.MarkBuildStepError()
 
   # Add a "cleanup" step so that errors from runtest.py or bb_device_steps.py
   # (which invoke this script) are kept in their own build step.
   util.MarkBuildStepStart('cleanup')
 
 
 if __name__ == '__main__':
   main()