OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # | 2 # |
3 # Copyright 2013 The Chromium Authors. All rights reserved. | 3 # Copyright 2013 The Chromium Authors. All rights reserved. |
4 # Use of this source code is governed by a BSD-style license that can be | 4 # Use of this source code is governed by a BSD-style license that can be |
5 # found in the LICENSE file. | 5 # found in the LICENSE file. |
6 | 6 |
7 """A class to keep track of devices across builds and report state.""" | 7 """A class to keep track of devices across builds and report state.""" |
8 | 8 |
9 import argparse | 9 import argparse |
10 import json | 10 import json |
11 import logging | 11 import logging |
12 import os | 12 import os |
13 import psutil | 13 import psutil |
14 import re | 14 import re |
15 import signal | 15 import signal |
16 import sys | 16 import sys |
17 | 17 |
18 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) | 18 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) |
| 19 import devil_chromium |
19 from devil.android import battery_utils | 20 from devil.android import battery_utils |
20 from devil.android import device_blacklist | 21 from devil.android import device_blacklist |
21 from devil.android import device_errors | 22 from devil.android import device_errors |
22 from devil.android import device_list | 23 from devil.android import device_list |
23 from devil.android import device_utils | 24 from devil.android import device_utils |
24 from devil.android.sdk import adb_wrapper | 25 from devil.android.sdk import adb_wrapper |
25 from devil.utils import lsusb | 26 from devil.utils import lsusb |
26 from devil.utils import reset_usb | 27 from devil.utils import reset_usb |
27 from devil.utils import run_tests_helper | 28 from devil.utils import run_tests_helper |
28 from pylib import constants | 29 from pylib import constants |
(...skipping 270 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
299 parser.add_argument('--json-output', | 300 parser.add_argument('--json-output', |
300 help='Output JSON information into a specified file.') | 301 help='Output JSON information into a specified file.') |
301 parser.add_argument('--blacklist-file', help='Device blacklist JSON file.') | 302 parser.add_argument('--blacklist-file', help='Device blacklist JSON file.') |
302 parser.add_argument('-v', '--verbose', action='count', default=1, | 303 parser.add_argument('-v', '--verbose', action='count', default=1, |
303 help='Log more information.') | 304 help='Log more information.') |
304 | 305 |
305 args = parser.parse_args() | 306 args = parser.parse_args() |
306 | 307 |
307 run_tests_helper.SetLogLevel(args.verbose) | 308 run_tests_helper.SetLogLevel(args.verbose) |
308 | 309 |
| 310 devil_chromium.Initialize() |
| 311 |
309 blacklist = (device_blacklist.Blacklist(args.blacklist_file) | 312 blacklist = (device_blacklist.Blacklist(args.blacklist_file) |
310 if args.blacklist_file | 313 if args.blacklist_file |
311 else None) | 314 else None) |
312 | 315 |
313 last_devices_path = os.path.join( | 316 last_devices_path = os.path.join( |
314 args.out_dir, device_list.LAST_DEVICES_FILENAME) | 317 args.out_dir, device_list.LAST_DEVICES_FILENAME) |
315 try: | 318 try: |
316 expected_devices = set( | 319 expected_devices = set( |
317 device_list.GetPersistentDeviceList(last_devices_path)) | 320 device_list.GetPersistentDeviceList(last_devices_path)) |
318 except IOError: | 321 except IOError: |
(...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
381 live_devices = [status['serial'] for status in statuses | 384 live_devices = [status['serial'] for status in statuses |
382 if (status['adb_status'] == 'device' | 385 if (status['adb_status'] == 'device' |
383 and not _IsBlacklisted(status['serial'], blacklist))] | 386 and not _IsBlacklisted(status['serial'], blacklist))] |
384 | 387 |
385 # If all devices failed, or if there are no devices, it's an infra error. | 388 # If all devices failed, or if there are no devices, it's an infra error. |
386 return 0 if live_devices else constants.INFRA_EXIT_CODE | 389 return 0 if live_devices else constants.INFRA_EXIT_CODE |
387 | 390 |
388 | 391 |
389 if __name__ == '__main__': | 392 if __name__ == '__main__': |
390 sys.exit(main()) | 393 sys.exit(main()) |
OLD | NEW |