IdlingResource.use

IdlingResourceExtension.kt

import android.support.test.espresso.IdlingRegistry
import android.support.test.espresso.IdlingResource

inline fun <T: IdlingResource?, R> T.use(block: (T) -> R): R {
    var exception: Throwable? = null
    try {
        IdlingRegistry.getInstance().register(this)
        return block(this)
    } catch (e: Throwable) {
        exception = e
        throw e
    } finally {
        when {
            this == null -> {}
            exception == null -> IdlingRegistry.getInstance().unregister(this)
            else ->
                try {
                    IdlingRegistry.getInstance().unregister(this)
                } catch (closeException: Throwable) {
                    // ignore
                }
        }
    }
}