Merge "Ensure waitOnMainUntil runs on the main thread" into udc-qpr-dev

2023-08-21 08:07:51 +00:00
parent 08e0db3d2d 30874b7782
commit 08497eca4a
1 changed files with 1 additions and 1 deletions
--- a/tests/InputMethodStressTest/src/com/android/inputmethod/stresstest/ImeStressTestUtil.java
+++ b/tests/InputMethodStressTest/src/com/android/inputmethod/stresstest/ImeStressTestUtil.java
@@ -154,7 +154,7 @@ public final class ImeStressTestUtil {
     * <p>The given {@code pred} will be called on the main thread.
     */
    public static void waitOnMainUntil(String message, Callable<Boolean> pred) {
-        eventually(() -> assertWithMessage(message).that(pred.call()).isTrue(), TIMEOUT);
+        eventually(() -> assertWithMessage(message).that(callOnMainSync(pred)).isTrue(), TIMEOUT);
    }

    /** Waits until IME is shown, or throws on timeout. */