[Android] Add option to wait for wifi after Reboot

build/android/pylib/device/device_utils.py

Index: build/android/pylib/device/device_utils.py
diff --git a/build/android/pylib/device/device_utils.py b/build/android/pylib/device/device_utils.py
index 805df6517eb6580ebd43d9d2091a8401935709fa..84011a1a4753fd8492f3cb58169f011a14049af8 100644
--- a/build/android/pylib/device/device_utils.py
+++ b/build/android/pylib/device/device_utils.py
@@ -350,11 +350,13 @@ class DeviceUtils(object):
   @decorators.WithTimeoutAndRetriesDefaults(
       REBOOT_DEFAULT_TIMEOUT,
       REBOOT_DEFAULT_RETRIES)
-  def Reboot(self, block=True, timeout=None, retries=None):
+  def Reboot(self, block=True, wifi=False, timeout=None, retries=None):
     """Reboot the device.
 
     Args:
       block: A boolean indicating if we should wait for the reboot to complete.
+      wifi: A boolean indicaring if we should wait for wifi to be enabled after

[jbudorick, 2015/02/03 14:53:29]

nit: indicaring -> indicating

+        the reboot. The option has no effect unless also block=True.

[jbudorick, 2015/02/03 14:53:28]

nit: rephrase: "unless |block| is also True."

       timeout: timeout in seconds
       retries: number of retries
 
@@ -369,7 +371,7 @@ class DeviceUtils(object):
     self._cache = {}
     timeout_retry.WaitFor(device_offline, wait_period=1)
     if block:
-      self.WaitUntilFullyBooted()
+      self.WaitUntilFullyBooted(wifi=wifi)
 
   INSTALL_DEFAULT_TIMEOUT = 4 * _DEFAULT_TIMEOUT
   INSTALL_DEFAULT_RETRIES = _DEFAULT_RETRIES