Provision Android devices after the 'device_status_check' step.

build/android/provision_devices.py

+#!/usr/bin/env python
+#
+# Copyright (c) 2013 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""Provisions Android devices with settings required for bots.
+
+Usage:
+  ./provision_devices.py [-d <device serial number>]
+"""
+
+import optparse
+import os
+import re
+import subprocess
+import sys
+import time
+
+from pylib import android_commands
+from pylib import constants
+
+
+def PushAndLaunchAdbReboot(device, target):
+  """Pushes and launches the adb_reboot binary on the device.
+
+  Arguments:
+    device: The serial number of the device to which the
+            adb_reboot binary should be pushed.
+    target: The build target (example, Debug or Release) which helps in
+            locating the adb_reboot binary.
+  """
+  print 'Will push and launch adb_reboot on %s' % device
+  android_cmd = android_commands.AndroidCommands(device)
+  # Kill if adb_reboot is already running.
+  android_cmd.KillAllBlocking('adb_reboot', 2)
+  # Push adb_reboot
+  print '  Pushing adb_reboot ...'
+  adb_reboot = os.path.join(constants.CHROME_DIR,
+                            'out/%s/adb_reboot' % target)
+  android_cmd.PushIfNeeded(adb_reboot, '/data/local/')
+  # Launch adb_reboot
+  print '  Launching adb_reboot ...'
+  p = subprocess.Popen(['adb', '-s', device, 'shell'], stdin=subprocess.PIPE)
+  p.communicate('/data/local/adb_reboot; exit\n')
+
+
+def LaunchHostHeartbeat():
+  ps = subprocess.Popen(['ps', 'aux'], stdout = subprocess.PIPE)
+  stdout, _ = ps.communicate()
+  matches = re.findall('\\n.*host_heartbeat.*', stdout)
+  for match in matches:
+    print 'An instance of host heart beart already running... will kill'
+    pid = re.findall('(\d+)', match)[1]
+    subprocess.call(['kill', str(pid)])
+  # Launch a new host_heartbeat
+  print 'Spawing host heartbeat...'
+  subprocess.Popen([os.path.join(constants.CHROME_DIR,
+                                 'build/android/host_heartbeat.py')])
+
+
+def ProvisionDevices(options):
+  if options.device is not None:
+    devices = [options.device]
+  else:
+    devices = android_commands.GetAttachedDevices()
+  for device in devices:
+    android_cmd = android_commands.AndroidCommands(device)
+    android_cmd.RunShellCommand('su -c date -u %f' % time.time())
+    PushAndLaunchAdbReboot(device, options.target)
+  LaunchHostHeartbeat()
+
+
+def main(argv):
+  parser = optparse.OptionParser()
+  parser.add_option('-d', '--device',
+                    help='The serial number of the device to be provisioned')
+  parser.add_option('-t', '--target',
+                    help='Path to the adb_reboot binary')
+  options, args = parser.parse_args(argv[1:])
+
+  if args:
+    print >> sys.stderr, 'Unused args %s' % args
+    return 1
+
+  if not options.target:
+    print >> sys.stderr, 'Build target not specified'
+    return 1
+
+  ProvisionDevices(options)
+
+
+if __name__ == '__main__':
+  sys.exit(main(sys.argv))