From cc282f8470a1123cabc59fb576e8fad1f914ffa1 Mon Sep 17 00:00:00 2001
From: Pawan Wagh <waghpawan@google.com>
Date: Fri, 14 Mar 2025 06:46:47 +0000
Subject: [PATCH] Use WaitForDeviceAvailable

WaitForDeviceOnline doesn't gurantee that device
can execute commands

Test: atest -c Enable16KbTest
Bug: 401597066
Change-Id: Icf55035ebe4b4a028892dc22665dafc8a5b492b4
---
 .../Enable16KbTests/src/com/android/test/Enable16KbTest.java  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/Enable16KbTests/src/com/android/test/Enable16KbTest.java b/tests/Enable16KbTests/src/com/android/test/Enable16KbTest.java
index 75e29a1d15e..5af6cbbce62 100644
--- a/tests/Enable16KbTests/src/com/android/test/Enable16KbTest.java
+++ b/tests/Enable16KbTests/src/com/android/test/Enable16KbTest.java
@@ -115,9 +115,9 @@ public class Enable16KbTest extends BaseHostJUnit4Test {
 
     private void prepareDevice() throws Exception {
         // Verify that device is online before running test and enable root
-        getDevice().waitForDeviceOnline(DEVICE_WAIT_TIMEOUT);
+        getDevice().waitForDeviceAvailable(DEVICE_WAIT_TIMEOUT);
         getDevice().enableAdbRoot();
-        getDevice().waitForDeviceOnline(DEVICE_WAIT_TIMEOUT);
+        getDevice().waitForDeviceAvailable(DEVICE_WAIT_TIMEOUT);
 
         getDevice().executeShellCommand("input keyevent KEYCODE_WAKEUP");
         getDevice().executeShellCommand("wm dismiss-keyguard");