如何在Kotlin中Mock第三方类的顶层扩展函数？

你遇到的问题核心在于：Kotlin顶层扩展函数会被编译成静态方法，而Mockito无法直接处理静态方法的Mock。必须使用MockK的mockkStatic来解决这个场景，以下是具体步骤和代码示例：

1. 确保引入MockK依赖 #

如果你的测试还没加MockK，先在构建脚本中添加依赖（以Gradle为例）：

testImplementation("io.mockk:mockk:1.13.8") // 可替换为最新稳定版本

2. 正确编写Mock代码 #

关键要点： #

• 用mockkStatic指定顶层扩展函数所在的编译后类名（Kotlin顶层函数会生成[包名].[文件名]Kt的Java类）
• 用MockK的every来Stub扩展函数的返回值

完整测试代码： #

// MyLibTest.kt
import io.mockk.every
import io.mockk.mockk
import io.mockk.mockkStatic
import org.junit.Test
import software.amazon.awssdk.services.rds.RdsClient
import com.tecton.orchestrator.monitoring.anExtensionMethod

@Test
fun testIt() {
    // 第一步：mock顶层扩展函数对应的编译类
    // 替换为你实际的包名+文件名Kt
    mockkStatic("com.tecton.orchestrator.monitoring.MyLibKt")
    
    // 第二步：创建RdsClient的mock实例
    val mockRdsClient = mockk<RdsClient>()
    
    // 第三步：Stub扩展函数，指定当mock实例调用该方法时返回"bob"
    every { mockRdsClient.anExtensionMethod() } returns "bob"
    
    // 验证逻辑
    check(mockRdsClient.anExtensionMethod() == "bob")
    // 后续业务逻辑测试...
}

为什么之前的Mockito写法不行？ #

Kotlin的顶层扩展函数RdsClient.anExtensionMethod()编译后会变成Java静态方法：

public final class MyLibKt {
    public static String anExtensionMethod(RdsClient $this$anExtensionMethod) {
        return "hi";
    }
}

Mockito无法识别这种静态方法调用是对mock对象的操作，所以抛出MissingMethodInvocationException。而MockK的mockkStatic专门用来处理静态成员的Mock，能正确关联mock实例和扩展函数的调用。

额外注意事项 #

• 如果多个测试用例都需要Mock这个扩展函数，建议在@BeforeEach中初始化mockkStatic，在@AfterEach中调用unmockkStatic("com.xxx.MyLibKt")或unmockkAll()清理mock状态，避免用例间的状态污染。
• 确认mockkStatic的类名拼写正确，包名+文件名+Kt，比如你的文件是MyLib.kt，包是com.foo.bar，类名就是com.foo.bar.MyLibKt。

内容的提问来源于stack exchange，提问作者Stephen Gross