Сборник тестовых потоков Kotlin Coroutine с помощью viewModelScope - PullRequest
Новая
       18

Сборник тестовых потоков Kotlin Coroutine с помощью viewModelScope

0 голосов
/ 26 сентября 2019

Я хочу проверить метод моей ViewModel, который собирает поток.Внутри коллектора объект LiveData видоизменен, что я хочу проверить в конце.Примерно так выглядит установка: 

//Outside viewmodel
val f = flow { emit("Test") }.flowOn(Dispatchers.IO)

//Inside viewmodel
val liveData = MutableLiveData<String>()

fun action() {
    viewModelScope.launch { privateAction() }
}

suspend fun privateAction() {
    f.collect {
        liveData.value = it
    }
}

Когда я сейчас вызываю метод action() в моем модульном тесте, тест заканчивается до сбора потока.Вот как может выглядеть тест: 

@Test
fun example() = runBlockingTest {
    viewModel.action()
    assertEquals(viewModel.liveData.value, "Test")
}

Я использую TestCoroutineDispatcher через это расширение Junit5, а также расширение для мгновенного выполнения для LiveData: 

    class TestCoroutineDispatcherExtension : BeforeEachCallback, AfterEachCallback, ParameterResolver {
    @SuppressLint("NewApi") // Only used in unit tests
    override fun supportsParameter(parameterContext: ParameterContext?, extensionContext: ExtensionContext?): Boolean {
        return parameterContext?.parameter?.type === testDispatcher.javaClass
    }

    override fun resolveParameter(parameterContext: ParameterContext?, extensionContext: ExtensionContext?): Any {
        return testDispatcher
    }

    private val testDispatcher = TestCoroutineDispatcher()

    override fun beforeEach(context: ExtensionContext?) {
        Dispatchers.setMain(testDispatcher)
    }

    override fun afterEach(context: ExtensionContext?) {
        Dispatchers.resetMain()
        testDispatcher.cleanupTestCoroutines()
    }
}

    class InstantExecutorExtension : BeforeEachCallback, AfterEachCallback {
    override fun beforeEach(context: ExtensionContext?) {
        ArchTaskExecutor.getInstance()
            .setDelegate(object : TaskExecutor() {
                override fun executeOnDiskIO(runnable: Runnable) = runnable.run()
                override fun postToMainThread(runnable: Runnable) = runnable.run()
                override fun isMainThread(): Boolean = true
            })
    }

    override fun afterEach(context: ExtensionContext?) {
        ArchTaskExecutor.getInstance().setDelegate(null)
    }
}

1 Ответ

0 голосов
/ 26 сентября 2019

Вы можете попробовать либо, 

fun action() = viewModelScope.launch { privateAction() }

suspend fun privateAction() {
    f.collect {
        liveData.value = it
    }
}

@Test
fun example() = runBlockingTest {
    viewModel.action().join()
    assertEquals(viewModel.liveData.value, "Test")
}

или 

fun action() {
    viewModelScope.launch { privateAction()
}

suspend fun privateAction() {
    f.collect {
        liveData.value = it
    }
}

@Test
fun example() = runBlockingTest {
    viewModel.action()
    viewModel.viewModelScope.coroutineContext[Job]!!.join()
    assertEquals(viewModel.liveData.value, "Test")
}

Вы также можете попробовать это, 

suspend fun <T> LiveData<T>.awaitValue(): T? {
    return suspendCoroutine { cont ->
        val observer = object : Observer<T> {
            override fun onChanged(t: T?) {
                removeObserver(this)
                cont.resume(t)
            }
        }
        observeForever(observer)
    }
}

@Test
fun example() = runBlockingTest {
    viewModel.action()
    assertEquals(viewModel.liveData.awaitValue(), "Test")
}
...