51 lines
1.7 KiB
Kotlin
51 lines
1.7 KiB
Kotlin
/* 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.fenix.helpers.idlingresource
|
|
|
|
import android.net.ConnectivityManager
|
|
import androidx.core.content.getSystemService
|
|
import androidx.test.espresso.IdlingResource
|
|
import androidx.test.platform.app.InstrumentationRegistry
|
|
import org.mozilla.fenix.ext.isOnline
|
|
|
|
/**
|
|
* An IdlingResource implementation that waits until the network connection is online or offline.
|
|
* The networkConnected parameter sets the expected connection status.
|
|
* Only after connecting/disconnecting has completed further actions will be performed.
|
|
*/
|
|
|
|
class NetworkConnectionIdlingResource(private val networkConnected: Boolean) : IdlingResource {
|
|
private var resourceCallback: IdlingResource.ResourceCallback? = null
|
|
private val connectionManager =
|
|
InstrumentationRegistry.getInstrumentation().context.getSystemService<ConnectivityManager>()
|
|
|
|
override fun getName(): String {
|
|
return this::javaClass.name
|
|
}
|
|
|
|
override fun isIdleNow(): Boolean {
|
|
val idle =
|
|
if (networkConnected) {
|
|
isOnline()
|
|
} else {
|
|
!isOnline()
|
|
}
|
|
if (idle) {
|
|
resourceCallback?.onTransitionToIdle()
|
|
}
|
|
return idle
|
|
}
|
|
|
|
override fun registerIdleTransitionCallback(callback: IdlingResource.ResourceCallback?) {
|
|
if (callback != null) {
|
|
resourceCallback = callback
|
|
}
|
|
}
|
|
|
|
private fun isOnline(): Boolean {
|
|
return connectionManager!!.isOnline()
|
|
}
|
|
}
|