Dump WindowHierarcy on getUiObject failure

This adds an additional dump of the window hierarcy when getUiObject
fails, to complement the dump of the screenshot on failure. This should
point why the screenshot shows the button being visible, but UiDevice
doesn't find it.

Flag: EXEMPT test debug
Bug: 395783236
Test: atest InputMethodServiceTest#testBackButtonClick
  InputMethodServiceTest#testBackButtonLongClick
  InputMethodServiceTest#testImeSwitchButtonClick
  InputMethodServiceTest#testImeSwitchButtonLongClick
Change-Id: I210249f43c6c53f3f185f17645e563732e13e072

1 files changed, 10 insertions, 0 deletions

diff --git a/services/tests/InputMethodSystemServerTests/src/com/android/inputmethodservice/InputMethodServiceTest.java b/services/tests/InputMethodSystemServerTests/src/com/android/inputmethodservice/InputMethodServiceTest.java
index 2cd860ae6c12..e263e85f020f 100644
--- a/services/tests/InputMethodSystemServerTests/src/com/android/inputmethodservice/InputMethodServiceTest.java
+++ b/services/tests/InputMethodSystemServerTests/src/com/android/inputmethodservice/InputMethodServiceTest.java
@@ -77,6 +77,8 @@ import org.junit.Test;
 import org.junit.rules.TestName;
 import org.junit.runner.RunWith;
 
+import java.io.ByteArrayOutputStream;
+import java.nio.charset.StandardCharsets;
 import java.util.List;
 import java.util.Objects;
 import java.util.concurrent.CountDownLatch;
@@ -1294,6 +1296,14 @@ public class InputMethodServiceTest {
         mInstrumentation.waitForIdleSync();
         final var postScreenshot = mInstrumentation.getUiAutomation().takeScreenshot();
         mDumpOnFailure.dumpOnFailure("post-getUiObject", postScreenshot);
+        try {
+            final var outputStream = new ByteArrayOutputStream();
+            mUiDevice.dumpWindowHierarchy(outputStream);
+            final String windowHierarchy = outputStream.toString(StandardCharsets.UTF_8);
+            mDumpOnFailure.dumpOnFailure("post-getUiObject", windowHierarchy);
+        } catch (Exception e) {
+            Log.i(TAG, "Failed to dump windowHierarchy", e);
+        }
         assertWithMessage("UiObject with " + bySelector + " was found").that(uiObject).isNotNull();
         return uiObject;
     }