Mockito ArgumentCaptor for Kotlin功能

Question

考虑一个将接口实现作为参数的函数,如下所示:

interface Callback {
    fun done()
}

class SomeClass {        

    fun doSomeThing(callback: Callback) {

        // do something

        callback.done()

    }    
}

当我想测试这个函数的调用者时,我可以做类似的事情

val captor = ArgumentCaptor.forClass(Callback::class)
Mockito.verify(someClass).doSomeThing(captor.capture())

为了测试调用回调时其他类的作用,我可以这样做

captor.value.done()

问题:如果我用高阶函数替换回调接口,我该怎么做呢？

fun doSomeThing(done: () -> Unit) {

    // do something

    done.invoke()

}

这可以用ArgumentCaptor来完成,我必须使用哪个类 ArgumentCaptor.forClass(???)

Answer 1

我推荐nhaarman/mockito-kotlin:将Mockito与Kotlin一起使用

它通过带有reified类型参数的内联函数解决了这个问题:

inline fun <reified T : Any> argumentCaptor() = ArgumentCaptor.forClass(T::class.java)

来源:的Mockito-科特林/ ArgumentCaptor.kt在a6f860461233ba92c7730dd42b0faf9ba2ce9281·nhaarman /的Mockito,科特林

例如:

val captor = argumentCaptor<() -> Unit>()
verify(someClass).doSomeThing(captor.capture())

要么

val captor: () -> Unit = argumentCaptor()
verify(someClass).doSomeThing(captor.capture())

Answer 2

我刚才遇到了这个问题，并通过内联解决了argumentCaptor它mockito-kotlin：

argumentCaptor<String>().apply {
  verify(myClass, times(2)).setItems(capture())

  assertEquals(2, allValues.size)
  assertEquals("test", firstValue)
}

firstValue是对第一个捕获的对象的引用。

来源：https ://github.com/mockito/mockito-kotlin/wiki/Mocking-and-verifying#argument-captors

Answer 3

我尝试了@mfulton26 的建议，但收到一条错误消息，说captor.capture() must not be null. 这对我有用。

声明了一个captor带@Captor注解的成员变量，

@Captor private lateinit var captor: ArgumentCaptor<Callback>

在我的@Test 中，

verify(someClass).doSomething(capture(captor))