模拟 kotlin 函数作为 Android 中的参数

Question

我需要测试一个接受另一个函数作为参数的 Kotlin 函数。 这个类看起来像这样：

class MyClass(val myInstance: MyObject) {

    fun toBeTested(text: String, number: Int) {
        myInstance.doSomething(text, otherFunction(number))
        ...
    }
}

到目前为止我的测试是这样的

@Test
fun testFunction() {
    val mockInstance = mock(MyObject::class.java)

    myClass.toBeTested("test", 15)
}

我想用verify(mockInstance, times(1)).doSomething()，但是我需要在doSomething里面指定它需要的参数，我不知道如何指定函数。

这是正确的方向吗？或者我应该采取其他方式（也许是重新组织代码或做一些其他的魔法）？

非常感谢

Answer 1

Mockito.any() 需要Class<T> 作为第一个参数，因此您要做的是获取() -> Unit 的Class 实例。 Kotlin 不允许您编写 (() -> Unit)::class.java，但它确实允许您访问 () -> Unit 的底层类：Function0。

所以要获得您需要的Class<() -> Unit>，您可以执行以下操作：

val f : Class<() -> Unit> = Function0::class.java as Class<Function0<Unit>>

您可以将此类传递给any 调用：

verify(mockInstance, times(1)).doSomething(any(f))

---

现在这不是很漂亮，因此您可以定义一个自定义的 any() 函数来为您完成所有这些魔法：

inline fun <reified T : Any> myAny() = Mockito.any(T::class.java)

这使您可以执行以下操作：

verify(mockInstance, times(1)).doSomething(myAny<() -> Unit>())
// or:
verify(mockInstance, times(1)).doSomething(myAny())

这就是 Mockito-Kotlin 在后台为您所做的事情。