diff options
Diffstat (limited to 'tests/common/src/com/android/tv/testing/uihelper/UiDeviceAsserts.java')
-rw-r--r-- | tests/common/src/com/android/tv/testing/uihelper/UiDeviceAsserts.java | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tests/common/src/com/android/tv/testing/uihelper/UiDeviceAsserts.java b/tests/common/src/com/android/tv/testing/uihelper/UiDeviceAsserts.java index ea9b5460..c096d7d2 100644 --- a/tests/common/src/com/android/tv/testing/uihelper/UiDeviceAsserts.java +++ b/tests/common/src/com/android/tv/testing/uihelper/UiDeviceAsserts.java @@ -17,6 +17,7 @@ package com.android.tv.testing.uihelper; import static com.android.tv.testing.uihelper.Constants.FOCUSED_VIEW; import static junit.framework.Assert.assertEquals; +import static junit.framework.Assert.assertNotNull; import static junit.framework.Assert.assertTrue; import android.support.test.uiautomator.By; @@ -38,6 +39,12 @@ public final class UiDeviceAsserts { assertEquals("Has " + bySelector, expected, uiDevice.hasObject(bySelector)); } + public static void assertWaitUntilFocused(UiDevice uiDevice, BySelector bySelector) { + UiObject2 uiObject = uiDevice.findObject(bySelector); + assertNotNull(uiObject); + assertTrue(uiObject.wait(Until.focused(true), Constants.MAX_FOCUSED_DELAY_MILLIS)); + } + /** * Assert that {@code searchCondition} becomes true within * {@value Constants#MAX_SHOW_DELAY_MILLIS} milliseconds. |