OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2012 The Swarming Authors. All rights reserved. | 2 # Copyright 2012 The Swarming Authors. All rights reserved. |
3 # Use of this source code is governed under the Apache License, Version 2.0 that | 3 # Use of this source code is governed under the Apache License, Version 2.0 that |
4 # can be found in the LICENSE file. | 4 # can be found in the LICENSE file. |
5 | 5 |
6 import cStringIO | 6 import cStringIO |
7 import hashlib | 7 import hashlib |
8 import json | 8 import json |
9 import logging | 9 import logging |
10 import optparse | 10 import optparse |
11 import os | 11 import os |
12 import shutil | 12 import shutil |
13 import subprocess | 13 import subprocess |
14 import sys | 14 import sys |
15 import tempfile | 15 import tempfile |
16 | 16 |
17 ROOT_DIR = unicode(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) | 17 ROOT_DIR = unicode(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) |
18 sys.path.insert(0, ROOT_DIR) | 18 sys.path.insert(0, ROOT_DIR) |
19 sys.path.insert(0, os.path.join(ROOT_DIR, 'third_party')) | 19 sys.path.insert(0, os.path.join(ROOT_DIR, 'third_party')) |
20 | 20 |
21 from depot_tools import auto_stub | 21 from depot_tools import auto_stub |
22 import auth | 22 import auth |
23 import isolate | 23 import isolate |
24 import isolate_format | 24 import isolate_format |
25 import isolated_format | 25 import isolated_format |
26 import isolateserver | 26 import isolateserver |
27 from utils import file_path | 27 from utils import file_path |
| 28 from utils import logging_utils |
28 from utils import tools | 29 from utils import tools |
29 import test_utils | 30 import test_utils |
30 | 31 |
31 ALGO = hashlib.sha1 | 32 ALGO = hashlib.sha1 |
32 | 33 |
33 | 34 |
34 NO_RUN_ISOLATE = { | 35 NO_RUN_ISOLATE = { |
35 'tests/isolate/files1/subdir/42.txt': | 36 'tests/isolate/files1/subdir/42.txt': |
36 'the answer to life the universe and everything\n', | 37 'the answer to life the universe and everything\n', |
37 'tests/isolate/files1/test_file1.txt': 'Foo\n', | 38 'tests/isolate/files1/test_file1.txt': 'Foo\n', |
(...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
96 | 97 |
97 class IsolateBase(auto_stub.TestCase): | 98 class IsolateBase(auto_stub.TestCase): |
98 def setUp(self): | 99 def setUp(self): |
99 super(IsolateBase, self).setUp() | 100 super(IsolateBase, self).setUp() |
100 self.mock(auth, 'ensure_logged_in', lambda _: None) | 101 self.mock(auth, 'ensure_logged_in', lambda _: None) |
101 self.old_cwd = os.getcwd() | 102 self.old_cwd = os.getcwd() |
102 self.cwd = file_path.get_native_path_case( | 103 self.cwd = file_path.get_native_path_case( |
103 unicode(tempfile.mkdtemp(prefix=u'isolate_'))) | 104 unicode(tempfile.mkdtemp(prefix=u'isolate_'))) |
104 # Everything should work even from another directory. | 105 # Everything should work even from another directory. |
105 os.chdir(self.cwd) | 106 os.chdir(self.cwd) |
| 107 self.mock( |
| 108 logging_utils.OptionParserWithLogging, 'logger_root', |
| 109 logging.Logger('unittest')) |
106 | 110 |
107 def tearDown(self): | 111 def tearDown(self): |
108 try: | 112 try: |
109 os.chdir(self.old_cwd) | 113 os.chdir(self.old_cwd) |
110 file_path.rmtree(self.cwd) | 114 file_path.rmtree(self.cwd) |
111 finally: | 115 finally: |
112 super(IsolateBase, self).tearDown() | 116 super(IsolateBase, self).tearDown() |
113 | 117 |
114 | 118 |
115 class IsolateTest(IsolateBase): | 119 class IsolateTest(IsolateBase): |
(...skipping 1253 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1369 }, f) | 1373 }, f) |
1370 | 1374 |
1371 self.mock(sys, 'stdout', cStringIO.StringIO()) | 1375 self.mock(sys, 'stdout', cStringIO.StringIO()) |
1372 cmd = [ | 1376 cmd = [ |
1373 '--isolate-server', 'http://localhost:1', | 1377 '--isolate-server', 'http://localhost:1', |
1374 '--dump-json', 'json_output.json', | 1378 '--dump-json', 'json_output.json', |
1375 join('x.isolated.gen.json'), | 1379 join('x.isolated.gen.json'), |
1376 join('y.isolated.gen.json'), | 1380 join('y.isolated.gen.json'), |
1377 ] | 1381 ] |
1378 self.assertEqual( | 1382 self.assertEqual( |
1379 0, isolate.CMDbatcharchive(tools.OptionParserWithLogging(), cmd)) | 1383 0, |
| 1384 isolate.CMDbatcharchive(logging_utils.OptionParserWithLogging(), cmd)) |
1380 expected = [ | 1385 expected = [ |
1381 { | 1386 { |
1382 'base_url': 'http://localhost:1', | 1387 'base_url': 'http://localhost:1', |
1383 'infiles': { | 1388 'infiles': { |
1384 join(isolated_file_x): { | 1389 join(isolated_file_x): { |
1385 'priority': '0', | 1390 'priority': '0', |
1386 }, | 1391 }, |
1387 join('foo'): { | 1392 join('foo'): { |
1388 'h': '520d41b29f891bbaccf31d9fcfa72e82ea20fcf0', | 1393 'h': '520d41b29f891bbaccf31d9fcfa72e82ea20fcf0', |
1389 's': 4, | 1394 's': 4, |
(...skipping 219 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1609 | 1614 |
1610 | 1615 |
1611 def clear_env_vars(): | 1616 def clear_env_vars(): |
1612 for e in ('ISOLATE_DEBUG', 'ISOLATE_SERVER'): | 1617 for e in ('ISOLATE_DEBUG', 'ISOLATE_SERVER'): |
1613 os.environ.pop(e, None) | 1618 os.environ.pop(e, None) |
1614 | 1619 |
1615 | 1620 |
1616 if __name__ == '__main__': | 1621 if __name__ == '__main__': |
1617 clear_env_vars() | 1622 clear_env_vars() |
1618 test_utils.main() | 1623 test_utils.main() |
OLD | NEW |