Kotlin单元测试：如何Mock不属于任何类的独立函数？

没问题！针对你的需求，我们可以通过两种常见方式来实现——一种是用PowerMock直接Mock顶层函数（不需要修改生产代码），另一种是通过重构代码来支持依赖注入（更符合测试友好的设计）。下面分别详解：

方案一：使用PowerMock Mock顶层函数（无需修改生产代码） #

Kotlin的顶层函数会被编译成对应文件名+Kt后缀的静态类中的静态方法，比如如果add()函数在MathUtils.kt文件中，编译后对应的类是MathUtilsKt。PowerMock可以帮助我们Mock这类静态方法，具体步骤如下：

1. 添加依赖 #

首先在你的构建文件中添加PowerMock和Mockito的相关依赖（以Gradle为例）：

testImplementation "org.powermock:powermock-module-junit4:2.0.9"
testImplementation "org.powermock:powermock-api-mockito2:2.0.9"
testImplementation "org.mockito:mockito-core:3.12.4"
testImplementation "org.assertj:assertj-core:3.21.0"

2. 修改测试类 #

给测试类添加PowerMock的注解，并在测试方法中Mock顶层函数：

import org.junit.Test
import org.junit.runner.RunWith
import org.powermock.api.mockito.PowerMockito
import org.powermock.core.classloader.annotations.PrepareForTest
import org.powermock.modules.junit4.PowerMockRunner
import org.assertj.core.api.Assertions.assertThat

// 指定用PowerMockRunner运行测试
@RunWith(PowerMockRunner::class)
// 准备需要Mock的顶层函数所在的编译类（替换成你实际的文件名+Kt）
@PrepareForTest(MathUtilsKt::class)
class CalculatorTest {
    private val c = Calculator()

    @Test
    fun MathUtilsSuccess() {
        // Mock顶层函数add()，让它始终返回2
        PowerMockito.mockStatic(MathUtilsKt::class.java)
        PowerMockito.`when`(MathUtilsKt.add()).thenReturn(2)

        val result = c.MathUtils()
        // 2 + 22 = 24，与断言一致，测试通过
        assertThat(result).isEqualTo(24)
    }
}

方案二：重构代码实现依赖注入（更优雅的测试友好设计） #

如果不想引入PowerMock这类重型框架，可以通过重构代码，把顶层函数的逻辑抽成可替换的依赖，这样测试时就能轻松替换实现：

1. 重构生产代码 #

创建一个接口定义add()的行为，然后让顶层函数委托给这个接口的实例：

// 定义行为接口
interface MathOperations {
    fun add(): Int
}

// 在原来的顶层函数文件中（比如MathUtils.kt），定义默认实现
var mathOperations: MathOperations = object : MathOperations {
    override fun add(): Int = 4 + 1 // 原来的add()逻辑
}

// 原来的顶层函数现在委托给mathOperations
fun add(): Int = mathOperations.add()

2. 编写测试用例 #

测试时只需要替换mathOperations的实现即可：

import org.junit.Test
import org.assertj.core.api.Assertions.assertThat

class CalculatorTest {
    private val c = Calculator()

    @Test
    fun MathUtilsSuccess() {
        // 替换成测试用的实现，让add()返回2
        mathOperations = object : MathOperations {
            override fun add(): Int = 2
        }

        val result = c.MathUtils()
        assertThat(result).isEqualTo(24)
    }
}

两种方案对比 #

• PowerMock方案：优点是不需要修改生产代码，适合快速解决问题；缺点是引入了额外框架，可能和某些测试工具存在兼容性问题，且Mock静态方法的测试可读性稍差。
• 重构方案：优点是代码设计更灵活、测试更优雅，符合依赖注入的原则；缺点是需要调整现有生产代码结构。

你可以根据自己的项目情况选择合适的方案~

内容的提问来源于stack exchange，提问作者Ayush Naithani