使用 kotlin 协程时，如何对调用挂起函数的函数进行单元测试？

Question

我有这样的课

class SomeClass {
    fun someFun() {
        // ... Some synchronous code
        async {
            suspendfun() 
        }
    }

    private suspend fun suspendFun() {
         dependency.otherFun().await()
         // ... other code
    }
}

我想对someFun() 进行单元测试，所以我编写了一个如下所示的单元测试：

@Test
fun testSomeFun() {
    runBlocking {
        someClass.someFun()
    }

    // ... verifies & asserts
}

但这似乎不起作用，因为在 runBlocking 中的所有内容完成之前，runBlocking 实际上不会阻止执行。如果我直接在runBlocking 内部测试suspendFun()，它会按预期工作，但我希望能够一起测试someFun()。

知道如何使用同步代码和异步代码测试函数吗？

Answer 1

修复异步

在实施时，您的 someFun() 只会“触发并忘记”async 结果。因此，runBlocking 对该测试没有任何影响。

如果可能，让someFun() 返回async 的Deferred，然后在runBlocking 中调用await。

fun someFun(): Deferred<Unit> {
    // ... Some synchronous code
    return async {
        suspendFun()
    }
}

然后测试：

runBlocking {
    SomeClass().someFun().await()
}

此question/answer 是获取更多信息的好资源。

替代方案：使用启动

也可以避免使用async，而使用suspend 函数和launch-创建的协程：

suspend fun someFun() {
    // ... Some synchronous code
    suspendFun()
}

private suspend fun suspendFun() {
    delay(1000)
    println("executed")
    // ... other code
}

测试使用launch，外部runBlocking隐式等待其完成：

val myScope = GlobalScope
runBlocking {
    myScope.launch {
        SomeClass().someFun()
    }
}