Index: appengine/swarming/server/bot_code_test.py |
diff --git a/appengine/swarming/server/bot_code_test.py b/appengine/swarming/server/bot_code_test.py |
index ee9c888fa32cce685ea99cdfcdfc79caf6036422..34f54ccc7e73e238909084b74f1b802563b8594e 100755 |
--- a/appengine/swarming/server/bot_code_test.py |
+++ b/appengine/swarming/server/bot_code_test.py |
@@ -7,7 +7,6 @@ import StringIO |
import logging |
import os |
import re |
-import shutil |
import subprocess |
import sys |
import tempfile |
@@ -23,6 +22,13 @@ from test_support import test_case |
from server import bot_archive |
from server import bot_code |
+CLIENT_DIR = os.path.join( |
+ os.path.dirname(os.path.dirname(test_env.APP_DIR)), 'client') |
+sys.path.insert(0, CLIENT_DIR) |
+from third_party.depot_tools import fix_encoding |
+from utils import file_path |
+sys.path.pop(0) |
+ |
class BotManagementTest(test_case.TestCase): |
def setUp(self): |
@@ -73,10 +79,11 @@ class BotManagementTest(test_case.TestCase): |
out = proc.communicate()[0] |
self.assertEqual(0, proc.returncode, out) |
finally: |
- shutil.rmtree(temp_dir) |
+ file_path.rmtree(temp_dir) |
if __name__ == '__main__': |
+ fix_encoding.fix_encoding() |
logging.basicConfig( |
level=logging.DEBUG if '-v' in sys.argv else logging.ERROR) |
unittest.main() |