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, adb_reboot):
+  """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.
+    adb_reboot: The path to 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 ...'
+  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', '%s' % str(pid)])

[Isaac (away), 2013/03/29 00:18:33]

'%s' % <string> is a no-op.  Just use <string> or in this case str(pid).

[Siva Chandra, 2013/03/29 01:47:06]

Done.

+  # 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.EnableAdbRoot()

[Isaac (away), 2013/03/29 00:18:33]

This will convert phone to root until the phone is restarted and the automated
restart is happening before this script is called.  So this will enable root for
the remainder of the build.

[Siva Chandra, 2013/03/29 01:47:06]

Removed enabling root.

+    android_cmd.RunShellCommand('date -u %f' % time.time())
+    PushAndLaunchAdbReboot(device, options.adb_reboot)
+  LaunchHostHeartbeat()
+
+
+def main(argv):
+  parser = optparse.OptionParser()
+  parser.add_option('-d', '--device', dest='device',

[Isaac (away), 2013/03/29 00:18:33]

remove dest param, unneeded

[Siva Chandra, 2013/03/29 01:47:06]

Done.

+                    help='The serial number of the device to be provisioned')
+  parser.add_option('-a', '--adb-reboot', dest='adb_reboot',

[Isaac (away), 2013/03/29 00:18:33]

dest param unneded.  Also I'd rather this script take the target (release vs.
debug) as a param rather than the full path and used constants to figure out the
full path to adb-reboot.  That way if we move adb_reboot, we can change this
script and let the buildbot scripts directory stay stable.

[Siva Chandra, 2013/03/29 01:47:06]

Done.

+                    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.adb_reboot:
+    print >> sys.stderr, 'Path to adb_reboot not specified'
+    return 1
+
+  ProvisionDevices(options)
+
+
+if __name__ == '__main__':
+  sys.exit(main(sys.argv))