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
package org.mozilla.focus.idlingResources
import androidx.test.espresso.IdlingResource
import androidx.test.platform.app.InstrumentationRegistry
import mozilla.components.browser.state.selector.selectedTab
import org.mozilla.focus.FocusApplication
/**
* An IdlingResource implementation that waits until the current session is not loading anymore.
* Only after loading has completed further actions will be performed.
*/
class SessionLoadedIdlingResource : IdlingResource {
private var resourceCallback: IdlingResource.ResourceCallback? = null
override fun getName(): String {
return SessionLoadedIdlingResource::class.java.simpleName
}
override fun isIdleNow(): Boolean {
val context = InstrumentationRegistry.getInstrumentation().targetContext.applicationContext as FocusApplication
val tab = context.components.store.state.selectedTab
return if (tab?.content?.loading == true) {
false
} else {
if (tab?.content?.progress == 100) {
invokeCallback()
true
} else {
false
}
}
}
private fun invokeCallback() {
if (resourceCallback != null) {
resourceCallback!!.onTransitionToIdle()
}
}
override fun registerIdleTransitionCallback(callback: IdlingResource.ResourceCallback) {
this.resourceCallback = callback
}
}