Add an assert to ensure UTF-8 locale when handling file paths.

appengine/swarming/remote_smoke_test.py

 #!/usr/bin/env python
 # Copyright 2014 The Swarming Authors. All rights reserved.
 # Use of this source code is governed by the Apache v2.0 license that can be
 # found in the LICENSE file.
 
 """Integration test for the Swarming server."""
 
 import Queue
 import json
 import logging
 import optparse
 import os
-import shutil
 import subprocess
 import sys
 import tempfile
 import threading
 import time
 
 APP_DIR = os.path.dirname(os.path.abspath(__file__))
 CHECKOUT_DIR = os.path.dirname(os.path.dirname(APP_DIR))
 CLIENT_DIR = os.path.join(CHECKOUT_DIR, 'client')
 SWARMING_SCRIPT = os.path.join(CLIENT_DIR, 'swarming.py')
 
+sys.path.insert(0, CLIENT_DIR)
+from third_party.depot_tools import fix_encoding
+from utils import file_path
+sys.path.pop(0)
+
 
 def gen_isolated(isolate, script, includes=None):
   """Archives a script to `isolate` server."""
   tmp = tempfile.mkdtemp(prefix='swarming_smoke')
   data = {
     'variables': {
       'command': ['python', '-u', 'script.py'],
       'files': ['script.py'],
     },
   }
@@ skipping 19 matching lines @@
       data['includes'] = includes
       with open(isolated, 'wb') as f:
         json.dump(data, f, sort_keys=True, separators=(',', ':'))
       cmd = [
         os.path.join(CLIENT_DIR, 'isolateserver.py'), 'archive',
         '-I', isolate, '--namespace', 'default-gzip', isolated,
       ]
       out = subprocess.check_output(cmd)
     return out.split(' ', 1)[0]
   finally:
-    shutil.rmtree(tmp)
+    file_path.rmtree(tmp)
 
 
 def capture(cmd, **kwargs):
   """Captures output and return exit code."""
   proc = subprocess.Popen(
       cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, **kwargs)
   out = proc.communicate()[0]
   return out, proc.returncode
 
 
@@ skipping 111 matching lines @@
 def run_test(results, swarming, isolate, extra_flags, name, test_case):
   start = time.time()
   try:
     result = test_case(swarming, isolate, extra_flags)
   except Exception as e:
     result = e
   results.put((name, result, time.time() - start))
 
 
 def main():
+  fix_encoding.fix_encoding()
   # It's necessary for relative paths in .isolate.
   os.chdir(APP_DIR)
 
   parser = optparse.OptionParser()
   parser.add_option('-S', '--swarming', help='Swarming server')
   parser.add_option('-I', '--isolate-server', help='Isolate server')
   parser.add_option('-d', '--dimensions', nargs=2, default=[], action='append')
   parser.add_option('-v', '--verbose', action='store_true', help='Logs more')
   options, args = parser.parse_args()
 
@@ skipping 28 matching lines @@
   for i in xrange(len(tests)):
     name, result, duration = results.get()
     print('[%d/%d] %-*s: %4.1fs: %s' %
         (i, len(tests), maxlen, name, duration, result))
 
   return 0
 
 
 if __name__ == '__main__':
   sys.exit(main())