| 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()
|
|
|