Question

我需要对函数doClassAction进行单元测试，并在其内部调用对象doStoreAction上的函数store。这个doStoreAction接受两个参数，一个整数和一个函数，如下例所示

class MyStore() {

    fun doStoreAction(num: Int, callback: (text: String) -> Unit) {
        // It does something 
        // ...
        // and when it's done it calls the callback

        return callback("some text")
    }
}

class MyClass(private val store: MyStore,
              private val objectX: MyObject) {

    fun doClassAction(num: Int) {
        store.doStoreAction(num) { callbackResult ->
            // Action (potentially asynchronous)
            // that will happen after obtaining the value via the callback

            objectX.doObjectAction(callbackResult)
        }
    }
}

我正在使用Mockito，到目前为止，我的测试有以下几行

@Test
fun doClassActionTest() {
    val mockStore = mock(MyStore::class.java)
    val mockObject = mock(MyObject::class.java)
    val class = MyClass(mockStore, mockObject)
    class.doClassAction(42)
}

但是它没有编译，因为对方法的调用没有指定回调...我不想覆盖它，因为我希望测试使用原始的并确认内部的东西被正确调用，像：

when(objectX.doObjectAction(anyString())).doNothing()
verify(objectX, times(1)).doObjectAction(callbackResult)

有没有办法调用该方法，并模拟我在测试中设置的值的回调响应，但仍然让它通过原始代码流？

Answer 1

经过几天的尝试和阅读后，我终于找到了一个有效的解决方案，我会在这里分享它，因为谁发现自己在同一个泡菜中

@Test
fun doClassActionTest() {
    val mockStore = mock(MyStore::class.java)
    val mockObject = mock(MyObject::class.java)
    val myClass = MyClass(mockStore, mockObject)
    val input = 42
    val resultCallback = "test result"

    `when`(mockStore.doStoreAction(any(), any())).then { invocation ->
        (invocation.arguments[1] as (String) -> Unit).invoke(resultCallback)
    }

    myClass.doClassAction(input)

    verify(mockObject, only()).doObjectAction(resultCallback)
}

这里需要注意的是记住将你正在嘲笑的类和这些类调用的方法放在Kotlin中open，否则可能会出现误报

P.S。我还使用了从mockito-kotlin库中提取的几个内联函数，例如：

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

inline fun <reified T : Any> createInstance(): T = createInstance(T::class)

fun <T : Any> createInstance(kClass: KClass<T>): T = castNull()

@Suppress("UNCHECKED_CAST")
private fun <T> castNull(): T = null as T

使用Kotlin在Android中测试lambda函数

1 个答案: