时间戳

首页 > 解决方案 > 在 kotlin 中，我如何模拟一个包装回调的挂起函数？

callback - 在 kotlin 中，我如何模拟一个包装回调的挂起函数？

问题描述

假设有一个带有回调的接口：

interface SomeInterface {
    fun doSomething(arg: String, callback: (Exception?, Long) -> Unit)
}

我将其扩展到这样的挂起函数：

suspend fun SomeInterface.doSomething(arg: String): Long = suspendCoroutine { cont ->
    this.doSomething(arg) { err, result ->
        if (err == null) {
            cont.resume(result)
        } else {
            cont.resumeWithException(err)
        }
    }
}

我想在测试中模拟这个，但我失败了。理想情况下，我想使用这样的东西：

@Test
fun checkService() {
    runBlocking {
        val myService = mock<SomeInterface>()
        whenever(myService.doSomething(anyString())).thenReturn(1234L)
        val result = myService.doSomething("")
        assertEquals(result, 1234L)
    }
}

上述语法因模拟异常而失败，因为它期望回调的匹配器。

org.mockito.exceptions.misusing.InvalidUseOfMatchersException: 
Invalid use of argument matchers!
2 matchers expected, 1 recorded:

我怎样才能模拟这样的挂起功能？如果不可能使用类似的语法，我如何才能使用所需的参数进行模拟回调，以便在整个代码中使用的挂起变体在测试期间返回所需的结果？

更新：当它是一个扩展功能时，它似乎是不可能的。根据Marko Topolnik的评论，我认为这是因为扩展只是一个静态函数，超出了 mockito 的能力。

当挂起函数是成员函数时，它会按预期工作，使用我的原始语法。

这是带有一些演示代码的要点： https ://gist.github.com/mirceanis/716bf019a47826564fa57a77065f2335

标签： callbackkotlinmockingmockitokotlinx.coroutines

解决方案

我建议使用MockK进行测试，这对协程更友好。

要模拟协程，您可以使用coEvery和returns如下所示：

val interf = mockk<SomeInterface>()
coEvery { a.doSomething(any()) } returns Outcome.OK

推荐阅读