正如摘要所述，本文将深入探讨如何在 Java 中使用 Mockito 等框架来模拟嵌套函数调用，以实现对复杂代码逻辑的单元测试。通过分层模拟，我们可以精确控制每个函数的返回值，进而验证代码的正确性。
在编写单元测试时，经常会遇到需要模拟函数调用链的情况，例如 int var = func1(func2(obj.func3())); 。 直接模拟整个表达式的结果可能并不容易，更有效的方法是逐层模拟，控制每个函数的返回值，最终达到模拟整个表达式的目的。
分层模拟的实现步骤：
-
模拟最内层函数： 首先，模拟最内层的函数 obj.func3()，定义其返回值。
立即学习“Java免费学习笔记（深入）”；
模拟中间层函数： 接着，模拟中间层的函数 func2()，并使用 any() 匹配器来匹配其输入参数。定义 func2() 的返回值。
隐含模拟最外层函数： 由于 func1() 的返回值最终赋给了 var，我们实际上通过前两步已经控制了整个表达式的结果，func1() 的返回值无需显式模拟，测试代码会根据 func2() 的返回值自动计算得到。
示例代码（使用 Mockito）：
假设我们有以下代码：
class MyClass {
public int func3() {
// 实际代码
return 0;
}
}
class AnotherClass {
public int func2(int input) {
// 实际代码
return 0;
}
public int func1(int input) {
// 实际代码
return 0;
}
}
public class MainClass {
private MyClass obj;
private AnotherClass anotherObj;
public MainClass(MyClass obj, AnotherClass anotherObj) {
this.obj = obj;
this.anotherObj = anotherObj;
}
public int calculate() {
return anotherObj.func1(anotherObj.func2(obj.func3()));
}
}我们可以使用 Mockito 模拟 func3() 和 func2() 的返回值，从而控制 calculate() 的返回值。
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.Mockito.when;
public class MainClassTest {
@Test
public void testCalculate() {
// 创建 Mock 对象
MyClass mockObj = Mockito.mock(MyClass.class);
AnotherClass mockAnotherObj = Mockito.mock(AnotherClass.class);
// 定义模拟行为
when(mockObj.func3()).thenReturn(10);
when(mockAnotherObj.func2(anyInt())).thenReturn(20);
when(mockAnotherObj.func1(anyInt())).thenReturn(30); // 模拟 func1, 假设 func1 依赖于 func2 的结果
// 创建被测试对象，并注入 Mock 对象
MainClass mainClass = new MainClass(mockObj, mockAnotherObj);
// 执行测试
int result = mainClass.calculate();
// 断言结果
assertEquals(30, result);
}
}代码解释：
- Mockito.mock(MyClass.class) 和 Mockito.mock(AnotherClass.class) 创建了 MyClass 和 AnotherClass 的 Mock 对象。
- when(mockObj.func3()).thenReturn(10) 定义了当调用 mockObj.func3() 时，返回值为 10。
- when(mockAnotherObj.func2(anyInt())).thenReturn(20) 定义了当调用 mockAnotherObj.func2()，且参数为任意 int 值时，返回值为 20。
- when(mockAnotherObj.func1(anyInt())).thenReturn(30) 定义了当调用 mockAnotherObj.func1()，且参数为任意 int 值时，返回值为 30。
- anyInt() 是 Mockito 提供的参数匹配器，用于匹配任意 int 类型的参数。 any() 可以匹配任何类型的参数。
注意事项：
- 确保你的项目中引入了 Mockito 依赖。
- Mockito 只能模拟接口和类，不能模拟 final 类和 final 方法。
- 理解 when(...).thenReturn(...) 的含义，when() 括号内是方法调用，thenReturn() 括号内是返回值。
- 使用 any() 或 anyInt() 等参数匹配器时，需要确保匹配到所有可能的参数值，否则可能会导致测试失败。
- 当函数有多个参数时，可以使用 eq() 匹配特定值，或使用 any() 匹配任意值。 例如：when(mockObj.func(eq("specificValue"), anyInt())).thenReturn(true);
总结：
通过分层模拟函数调用链，我们可以更加精确地控制单元测试的输入和输出，提高测试的覆盖率和准确性。 掌握 Mockito 提供的各种参数匹配器，可以灵活应对不同的测试场景。 在实际开发中，根据代码的复杂程度选择合适的模拟策略，可以有效地提高代码质量和可维护性。
