mozilla-mobile/reference-browser/app/src/androidTest/java/org/mozilla/reference/browser/ext/UiDevice.kt

Revision control

Copy as Markdown

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
package org.mozilla.reference.browser.ext
import androidx.test.uiautomator.SearchCondition
import androidx.test.uiautomator.UiDevice
import androidx.test.uiautomator.UiObject2
import org.junit.Assert.assertNotNull
import org.mozilla.reference.browser.helpers.TestAssetHelper.waitingTime
/**
* Wait for an [UiObject2] and perform an interaction on it.
*
* @throws AssertionError if no [UiObject2] matches the condition
*/
fun UiDevice.waitAndInteract(condition: SearchCondition<UiObject2>, interaction: UiObject2.() -> Unit) {
val obj = this.wait(condition, waitingTime)
assertNotNull(obj)
interaction.invoke(obj)
obj.recycle()
}