robolectric-utils.kt

fun runRobolectric(action: suspend () -> Unit) {
    val result = AtomicReference<Result<Unit>>()

    @Suppress("EXPERIMENTAL_API_USAGE")
    MainScope().launch {
        try {
            action()
        } catch (e: Throwable) {
            result.set(Result.failure(e))
        } finally {
            result.compareAndSet(null, Result.success(Unit))
        }
    }

    while (result.get() == null) {
        Thread.sleep(100)
        ShadowLooper.runUiThreadTasks()
    }

    result.get().exceptionOrNull()?.let { throw it }
}