模拟 Java 函数调用链：深度解析与实践

class MyClass {
    public int func1(int arg) {
        // 实际逻辑
        return arg * 2;
    }

    public int func2(int arg) {
        // 实际逻辑
        return arg + 1;
    }
}

class MyObject {
    public int func3() {
        // 实际逻辑
        return 5;
    }
}

class MainClass {
    private MyClass myClass;
    private MyObject myObject;

    public MainClass(MyClass myClass, MyObject myObject) {
        this.myClass = myClass;
        this.myObject = myObject;
    }

    public int calculate() {
        return myClass.func1(myClass.func2(myObject.func3()));
    }
}

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() {
        // 1. 创建 Mock 对象
        MyClass myClassMock = Mockito.mock(MyClass.class);
        MyObject myObjectMock = Mockito.mock(MyObject.class);

        // 2. 设置 Mock 对象的行为
        // 模拟 myObject.func3() 返回 5
        when(myObjectMock.func3()).thenReturn(5);

        // 模拟 myClass.func2(anyInt()) 返回 10 (假设 func3() 返回 5, func2(5) 返回 10)
        when(myClassMock.func2(anyInt())).thenReturn(10);

        // 模拟 myClass.func1(anyInt()) 返回 20 (假设 func2() 返回 10, func1(10) 返回 20)
        when(myClassMock.func1(anyInt())).thenReturn(20);

        // 3. 创建被测试对象
        MainClass mainClass = new MainClass(myClassMock, myObjectMock);

        // 4. 执行测试
        int result = mainClass.calculate();

        // 5. 验证结果
        assertEquals(20, result);
    }
}