Android中Kotlin函数作为参数的Mock测试问题

问题场景 #

你需要测试MyClass中的toBeTested函数，该函数会调用依赖实例myInstance的doSomething方法，其中第二个参数是otherFunction(number)的计算结果。现在想通过Mockito的verify来验证这个调用的参数是否符合预期。

解决方案 #

根据otherFunction的特性，有几种不同的实现方式：

1. 如果otherFunction是纯函数（无副作用、输入固定输出固定）

这种情况最简单，直接计算出预期结果，传入verify即可：

@Test 
fun testFunction() { 
    val mockInstance = mock(MyObject::class.java) 
    val myClass = MyClass(mockInstance) // 别忘了初始化MyClass实例
    
    // 计算预期的参数结果
    val expectedNumberResult = otherFunction(15)
    
    myClass.toBeTested("test", 15) 
    
    // 验证调用次数和参数
    verify(mockInstance, times(1)).doSomething("test", expectedNumberResult)
}

纯函数的好处是你可以安全地直接调用它来获取预期值，不用担心产生额外副作用。

2. 用ArgumentCaptor捕获参数并验证

如果otherFunction有副作用、或者你想隔离它的调用（比如它依赖其他未初始化的资源），可以用Mockito的ArgumentCaptor来捕获doSomething的调用参数，再对捕获到的值进行断言：

import org.mockito.ArgumentCaptor
import org.mockito.kotlin.captor
import org.mockito.kotlin.verify

@Test 
fun testFunction() { 
    val mockInstance = mock(MyObject::class.java) 
    val myClass = MyClass(mockInstance)
    
    myClass.toBeTested("test", 15) 
    
    // 定义参数捕获器
    val textCaptor: ArgumentCaptor<String> = captor()
    val numberCaptor: ArgumentCaptor<Int> = captor()
    
    // 捕获doSomething的调用参数
    verify(mockInstance).doSomething(textCaptor.capture(), numberCaptor.capture())
    
    // 验证参数是否符合预期
    assertEquals("test", textCaptor.value)
    // 这里根据otherFunction的逻辑断言结果，比如如果逻辑是num*2，就断言30
    assertEquals(30, numberCaptor.value)
}

这种方式不需要直接调用otherFunction，而是通过捕获实际传入doSomething的参数来验证计算逻辑是否正确。

3. 隔离otherFunction（如果它是可注入的依赖）

如果otherFunction是MyClass的依赖（比如作为构造参数传入），那你可以mock这个依赖，然后验证它的调用，同时指定返回值：

// 假设otherFunction封装为接口方法
interface NumberProcessor {
    fun otherFunction(num: Int): Int
}

// 修改MyClass构造函数，注入依赖
class MyClass(val myInstance: MyObject, val numberProcessor: NumberProcessor) { 
    fun toBeTested(text: String, number: Int) { 
        myInstance.doSomething(text, numberProcessor.otherFunction(number)) 
    } 
}

// 测试代码
@Test 
fun testFunction() { 
    val mockInstance = mock(MyObject::class.java) 
    val mockProcessor = mock(NumberProcessor::class.java)
    // 指定mockProcessor的返回值
    whenever(mockProcessor.otherFunction(15)).thenReturn(30)
    
    val myClass = MyClass(mockInstance, mockProcessor)
    myClass.toBeTested("test", 15) 
    
    // 先验证numberProcessor的调用
    verify(mockProcessor).otherFunction(15)
    // 再验证myInstance的调用参数
    verify(mockInstance, times(1)).doSomething("test", 30)
}

这种方式适合otherFunction逻辑复杂或者依赖外部资源的场景，通过依赖注入实现完全的测试隔离。

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