Use an API to determine if draw listener logic should be invoked.

Currently ViewCapture hardcodes a Robolectric check and skips certain logic.
This commit changes this logic to use a ControlledLooper API instead,
so in future the real logic can be executed when Robolectric supports it.

PiperOrigin-RevId: 615567737

Author: brettchabot

core/CHANGELOG.md

@@ -6,6 +6,8 @@
 
 **Bug Fixes**
 
+* make ViewCapture use ControlledLooper API instead of hardcoding is Robolectric check
+
 **New Features**
 
 **Breaking Changes**

core/java/androidx/test/core/view/ViewCapture.kt

@@ -35,6 +35,8 @@ import androidx.annotation.RequiresApi
 import androidx.concurrent.futures.ResolvableFuture
 import androidx.test.annotation.ExperimentalTestApi
 import androidx.test.core.internal.os.HandlerExecutor
+import androidx.test.internal.platform.ServiceLoaderWrapper
+import androidx.test.internal.platform.os.ControlledLooper
 import androidx.test.internal.platform.reflect.ReflectiveField
 import androidx.test.internal.platform.reflect.ReflectiveMethod
 import androidx.test.platform.graphics.HardwareRendererCompat
@@ -74,17 +76,23 @@ fun View.captureToBitmap(rect: Rect? = null): ListenableFuture<Bitmap> {
   }
 
   mainExecutor.execute {
-    if (Build.FINGERPRINT.contains("robolectric")) {
-      generateBitmap(bitmapFuture, rect)
-    } else {
+    if (getControlledLooper().areDrawCallbacksSupported()) {
       val forceRedrawFuture = forceRedraw()
       forceRedrawFuture.addListener({ generateBitmap(bitmapFuture, rect) }, mainExecutor)
+    } else {
+      generateBitmap(bitmapFuture, rect)
     }
   }
 
   return bitmapFuture
 }
 
+private fun getControlledLooper(): ControlledLooper {
+  return ServiceLoaderWrapper.loadSingleService(ControlledLooper::class.java) {
+    ControlledLooper.NO_OP_CONTROLLED_LOOPER
+  }
+}
+
 /**
  * Equivalent to [captureToBitmap] for now. In the future [captureToBitmap] will be changed to a
  * suspend function.

runner/monitor/CHANGELOG.md

@@ -16,6 +16,7 @@
 * Upstream TestStorage.getInputFileUri and getOutputFileUri  to PlatformTestStorage
 * Change PlatformTestStorage methods to throw FileNotFoundException instead of 
   IOException
+* Add internal ControlledLooper#isDrawCallbacksSupported.
 
 **Breaking API Changes**
 

runner/monitor/java/androidx/test/api/current_internal.txt

@@ -53,6 +53,7 @@ package androidx.test.internal.platform.content {
 package androidx.test.internal.platform.os {
 
   @RestrictTo(androidx.annotation.RestrictTo.Scope.LIBRARY_GROUP) public interface ControlledLooper {
+    method public default boolean areDrawCallbacksSupported();
     method public void drainMainThreadUntilIdle();
     method public void simulateWindowFocus(android.view.View!);
     field public static final androidx.test.internal.platform.os.ControlledLooper! NO_OP_CONTROLLED_LOOPER;

runner/monitor/java/androidx/test/internal/platform/os/ControlledLooper.java

@@ -39,12 +39,22 @@ public interface ControlledLooper {
   /** Generate window focus event for given view. */
   void simulateWindowFocus(View decorView);
 
+  /** Returns true if registerFrameCommitCallback() and addOnDrawListener callbacks are supported */
+  default boolean areDrawCallbacksSupported() {
+    return false;
+  }
+
   public static final ControlledLooper NO_OP_CONTROLLED_LOOPER =
       new ControlledLooper() {
         @Override
         public void drainMainThreadUntilIdle() {}
 
         @Override
         public void simulateWindowFocus(View decorView) {}
+
+        @Override
+        public boolean areDrawCallbacksSupported() {
+          return true;
+        }
       };
 }