如何在 Mockk Kotlin 中測試異步 function

Question

我想測試這個function。

suspend fun fetchTwoDocs() =
        coroutineScope {
            val deferredOne = async { fetchDoc(1) }
            val deferredTwo = async { fetchDoc(2) }
            deferredOne.await()
            deferredTwo.await()
        }

如何在 mockk 中測試這個 function

Answer 1

我假設您不能或不會重寫您正在編寫的代碼以使用協程。 在這種情況下，Pablisco 在這里提出了很好的想法。 我最喜歡的一個是使用隊列：

// You create a sync queue
val queue = SynchronousQueue<String>() // Or the type of 'fetch2Docs'
// fetchTwoDocs runs async so it will return eventually
queue.put(fetchTwoDocs())
// queue.take() will wait until there is a value in the queue
assertThat(queue.take(), equalTo("expected value"))

這是一個關於如何在假設的異步回調中使用它的額外示例：

val queue = SynchronousQueue<String>()
asyncFunctionThatReturnsWithCallback(someParam) { callbackParam ->
    queue.put(callbackParam)
}
assertTrue(queue.take())

Answer 2

如果你只是想測試這個函數，你可以簡單地從你的測試中在runBlocking調用它，或者使用提供runBlockingTest的kotlinx-coroutines-test庫：

@Test
fun test() = runBlocking {
    val result = fetchTwoDocs()
    // then assert stuff
}