OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # | 2 # |
3 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 3 # Copyright (c) 2012 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 """Enables dalvik vm asserts in the android device.""" | 7 """Enables dalvik vm asserts in the android device.""" |
8 | 8 |
9 import argparse | 9 import argparse |
10 import sys | 10 import sys |
11 | 11 |
| 12 import devil_chromium |
12 from devil.android import device_blacklist | 13 from devil.android import device_blacklist |
13 from devil.android import device_utils | 14 from devil.android import device_utils |
14 | 15 |
15 | 16 |
16 def main(): | 17 def main(): |
17 parser = argparse.ArgumentParser() | 18 parser = argparse.ArgumentParser() |
18 | 19 |
19 parser.add_argument('--blacklist-file', help='Device blacklist JSON file.') | 20 parser.add_argument('--blacklist-file', help='Device blacklist JSON file.') |
20 | 21 |
21 set_asserts_group = parser.add_mutually_exclusive_group(required=True) | 22 set_asserts_group = parser.add_mutually_exclusive_group(required=True) |
22 set_asserts_group.add_argument( | 23 set_asserts_group.add_argument( |
23 '--enable_asserts', dest='set_asserts', action='store_true', | 24 '--enable_asserts', dest='set_asserts', action='store_true', |
24 help='Sets the dalvik.vm.enableassertions property to "all"') | 25 help='Sets the dalvik.vm.enableassertions property to "all"') |
25 set_asserts_group.add_argument( | 26 set_asserts_group.add_argument( |
26 '--disable_asserts', dest='set_asserts', action='store_false', | 27 '--disable_asserts', dest='set_asserts', action='store_false', |
27 help='Removes the dalvik.vm.enableassertions property') | 28 help='Removes the dalvik.vm.enableassertions property') |
28 | 29 |
29 args = parser.parse_args() | 30 args = parser.parse_args() |
30 | 31 |
| 32 devil_chromium.Initialize() |
| 33 |
31 blacklist = (device_blacklist.Blacklist(args.blacklist_file) | 34 blacklist = (device_blacklist.Blacklist(args.blacklist_file) |
32 if args.blacklist_file | 35 if args.blacklist_file |
33 else None) | 36 else None) |
34 | 37 |
35 # TODO(jbudorick): Accept optional serial number and run only for the | 38 # TODO(jbudorick): Accept optional serial number and run only for the |
36 # specified device when present. | 39 # specified device when present. |
37 devices = device_utils.DeviceUtils.parallel( | 40 devices = device_utils.DeviceUtils.parallel( |
38 device_utils.DeviceUtils.HealthyDevices(blacklist)) | 41 device_utils.DeviceUtils.HealthyDevices(blacklist)) |
39 | 42 |
40 def set_java_asserts_and_restart(device): | 43 def set_java_asserts_and_restart(device): |
41 if device.SetJavaAsserts(args.set_asserts): | 44 if device.SetJavaAsserts(args.set_asserts): |
42 device.RunShellCommand('stop') | 45 device.RunShellCommand('stop') |
43 device.RunShellCommand('start') | 46 device.RunShellCommand('start') |
44 | 47 |
45 devices.pMap(set_java_asserts_and_restart) | 48 devices.pMap(set_java_asserts_and_restart) |
46 return 0 | 49 return 0 |
47 | 50 |
48 | 51 |
49 if __name__ == '__main__': | 52 if __name__ == '__main__': |
50 sys.exit(main()) | 53 sys.exit(main()) |
OLD | NEW |