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.tv.firefox.helpers
import androidx.test.core.app.ApplicationProvider
import androidx.test.espresso.IdlingResource
import org.mozilla.tv.firefox.FirefoxApplication
/**
* 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
var ignoreLoading = false // Used in YouTubeNavigationTest to temporarily ignore the session idler
override fun getName(): String {
return SessionLoadedIdlingResource::class.java.simpleName
}
override fun isIdleNow(): Boolean {
val context = ApplicationProvider.getApplicationContext<FirefoxApplication>()
val sessionManager = context.components.sessionManager
val session = sessionManager.selectedSession
return if (session?.loading == true && !ignoreLoading) {
false
} else {
invokeCallback()
true
}
}
private fun invokeCallback() {
if (resourceCallback != null) {
resourceCallback!!.onTransitionToIdle()
}
}
override fun registerIdleTransitionCallback(callback: IdlingResource.ResourceCallback) {
this.resourceCallback = callback
}
}