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