如何在界面中模拟 Kotlin 扩展功能？

Question

我有如下接口的扩展功能：

import javax.jms.ConnectionFactory

fun ConnectionFactory.foo() {
    println("do some stuff")
}

如何模拟函数 foo？

请注意，我在http://mockk.io/#extension-functions 中看到了类和对象的方法，但它不起作用。这个我试过了：

import io.mockk.classMockk
import io.mockk.every
import org.junit.Test
import javax.jms.ConnectionFactory

class ExtensionFunctionTest {
@Test
fun mockExtensionFunction() {
    val connectionFactory = classMockk(ConnectionFactory::class)
    every { connectionFactory.foo() } returns println("do other stuff")
    connectionFactory.foo()
}
}

抛出异常：

io.mockk.MockKException: Missing calls inside every { ... } block.

Answer 1

根据documentation，如果是模块范围的扩展函数，您需要为扩展函数创建staticMock“隐藏”类。 下面是一个例子（假设文件名为com/sample/extmockingtest/SampleTest.kt）：

fun <T> Iterable<T>.foo(): String  = "do some stuff"

class ExtensionFunctionTest {
    @Test
    fun mockExtensionFunction() {
        val itMock = classMockk(Iterable::class);
        staticMockk("com.sample.extmockingtest.SampleTestKt").use {
            every {
                itMock.foo()
            } returns "do other stuff"

            assertEquals("do other stuff", itMock.foo())

            verify {
                itMock.foo()
            }
        }
    }
}