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

client/tests/file_path_test.py

 #!/usr/bin/env python
 # coding=utf-8
 # Copyright 2013 The Swarming Authors. All rights reserved.
 # Use of this source code is governed under the Apache License, Version 2.0 that
 # can be found in the LICENSE file.
 
 import logging
 import os
 import tempfile
 import unittest
-import shutil
 import StringIO
 import subprocess
 import sys
 import time
 
 BASE_DIR = unicode(os.path.dirname(os.path.abspath(__file__)))
 ROOT_DIR = os.path.dirname(BASE_DIR)
 sys.path.insert(0, ROOT_DIR)
 sys.path.insert(0, os.path.join(ROOT_DIR, 'third_party'))
 
 FILE_PATH = unicode(os.path.abspath(__file__))
 
 from depot_tools import auto_stub
+from depot_tools import fix_encoding
 import test_utils
 from utils import file_path
 
 
 def write_content(filepath, content):
   with open(filepath, 'wb') as f:
     f.write(content)
 
 
 class FilePathTest(auto_stub.TestCase):
   def setUp(self):
     super(FilePathTest, self).setUp()
     self._tempdir = None
 
   def tearDown(self):
-    if self._tempdir:
-      for dirpath, dirnames, filenames in os.walk(self._tempdir, topdown=True):
-        for filename in filenames:
-          file_path.set_read_only(os.path.join(dirpath, filename), False)
-        for dirname in dirnames:
-          file_path.set_read_only(os.path.join(dirpath, dirname), False)
-      shutil.rmtree(self._tempdir)
-    super(FilePathTest, self).tearDown()
+    try:
+      if self._tempdir:
+        for dirpath, dirnames, filenames in os.walk(
+            self._tempdir, topdown=True):
+          for filename in filenames:
+            file_path.set_read_only(os.path.join(dirpath, filename), False)
+          for dirname in dirnames:
+            file_path.set_read_only(os.path.join(dirpath, dirname), False)
+        file_path.rmtree(self._tempdir)
+    finally:
+      super(FilePathTest, self).tearDown()
 
   @property
   def tempdir(self):
     if not self._tempdir:
       self._tempdir = tempfile.mkdtemp(prefix=u'run_isolated_test')
     return self._tempdir
 
   def assertFileMode(self, filepath, mode, umask=None):
     umask = test_utils.umask() if umask is None else umask
     actual = os.stat(filepath).st_mode
@@ skipping 185 matching lines @@
             file_path.get_native_path_case(filepath + data_suffix))
 
         open(filepath + '$DATA', 'w').close()
         self.assertEqual(
             filepath + data_suffix,
             file_path.get_native_path_case(filepath + data_suffix))
         # Ensure the ADS weren't created as separate file. You love NTFS, don't
         # you?
         self.assertEqual([basename], os.listdir(tempdir))
       finally:
-        shutil.rmtree(tempdir)
+        file_path.rmtree(tempdir)
 
     def test_rmtree_win(self):
       # Mock our sleep for faster test case execution.
       sleeps = []
       self.mock(time, 'sleep', sleeps.append)
       self.mock(sys, 'stderr', StringIO.StringIO())
 
       # Open a child process, so the file is locked.
       subdir = os.path.join(self.tempdir, 'to_be_deleted')
       os.mkdir(subdir)
@@ skipping 84 matching lines @@
           os.path.join(BASE_DIR, 'trace_inputs'))
       self.assertEqual('trace_inputs', os.path.basename(actual))
 
       # Make sure the symlink is not resolved.
       actual = file_path.get_native_path_case(
           os.path.join(BASE_DIR, 'trace_inputs', 'files2'))
       self.assertEqual('files2', os.path.basename(actual))
 
 
 if __name__ == '__main__':
+  fix_encoding.fix_encoding()
   logging.basicConfig(
       level=logging.DEBUG if '-v' in sys.argv else logging.ERROR)
   if '-v' in sys.argv:
     unittest.TestCase.maxDiff = None
   unittest.main()