时间戳

首页 > 解决方案 > 如何使用回调对挂起函数进行单元测试

android - 如何使用回调对挂起函数进行单元测试

问题描述

我有以下代码：

suspend fun initialize(sdk: Sdk) =
    suspendCoroutine<Unit> { continuation ->
        try {
            sdk.initialize(
                callback = { continuation.resume(Unit) },
                onFailure = { error -> continuation.resumeWithException(SdkException(error.message)) })
        } catch (exception: Exception) {
            continuation.resumeWithException(
                SdkException("Crash inside SDK", exception)
            )
        }
    }

Sdk 是一个第三方库。我suspendCoroutine用来暂停协程并在 sdk 完成初始化时恢复。一切正常，但是当我尝试编写这样的单元测试时，我得到以下结果IllegalStateException: This job has not completed yet。我正在使用mockito-kotlin编写以下测试：

    @Test
    fun `should initialize sdk correctly`() = runBlockingTest {
        val sdk = mock<Sdk>()

        initializeSdk(sdk)

        verify(sdk).initialize(any(), any())
    }

基本上我想做的是能够测试resume和resumeWithException

标签： androidunit-testingkotlinmockitokotlin-coroutines

解决方案

就个人而言，我不是模拟的粉丝。如果Sdk是一个接口，您可以只提供自己的测试实现来执行您的测试（用于成功和错误结果）。您可以准确控制何时/是否调用回调等。

如果 Sdk 是一个你无法控制的类，你可以创建一个接口来抽象这个Sdk类，并让你的initialize方法使用你自己的接口。但是我不得不承认这并不理想。

如果您坚持使用模拟，通常模拟库可以让您使用调用参数来模拟对方法调用的响应。使用 Mockito，它应该是这样的：

val sdk = mock<Sdk> {
    on { initialize(any(), any()) } doAnswer { invocation ->
        @Suppress("UNCHECKED_CAST")
        val successCallback = invocation.arguments[0] as (() -> Unit) // use callback's function type here
        successCallback.invoke()
    }
}

（虽然我不是 Mockito 方面的专家，所以可能有更简洁或类型安全的方法：D）

推荐阅读