Get data from LiveData via coroutines, based on https://github.com/googlesamples/android-sunflower/blob/master/app/src/androidTest/java/com/google/samples/apps/sunflower/utilities/LiveDataTestUtil.kt

LiveDataTestUtil.kt

package com.example

import androidx.lifecycle.LiveData
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.runBlocking
import kotlinx.coroutines.withContext
import java.util.concurrent.CountDownLatch
import java.util.concurrent.TimeUnit
import java.util.concurrent.atomic.AtomicReference

fun <T> LiveData<T>.blockingGet(): T =
    runBlocking {
        val data = AtomicReference<T?>()
        val latch = CountDownLatch(1)
        val observe = { o: T ->
            data.set(o)
            latch.countDown()
        }
        withContext(Dispatchers.Main) {
            this@blockingGet.observeForever(observe)
        }
        latch.await(2, TimeUnit.SECONDS)
        withContext(Dispatchers.Main) {
            this@blockingGet.removeObserver(observe)
        }
        data.get() ?: throw RuntimeException("Value was not recovered from LiveData in time.")
    }