@InjectMocks mock子方法

如果被@InjectMocks注解的类中有一个需要mock的子方法，可以使用Mockito的Spy对象来部分mock该对象的行为，同时保留其它方法的真实行为。

举个例子，我们有一个被@InjectMocks注解的类A，其中包含一个子方法B，我们想要mock子方法B的行为，同时保留其它方法的真实行为，可以按如下方式进行：

public class A {
    public int add(int a, int b) {
        return a + b;
    }
    public int multiply(int a, int b) {
        return a * b;
    }
    public int calculate(int a, int b) {
        int sum = add(a, b);
        int product = multiply(a, b);
        return sum + product;
    }
}

public class ATest {
    @InjectMocks
    private A a;

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testCalculate() {
        A spyA = Mockito.spy(a);
        Mockito.doReturn(10).when(spyA).add(Mockito.anyInt(), Mockito.anyInt());
        int result = spyA.calculate(2, 3);
        Assert.assertEquals(result, 26);
    }
}

在这个例子中，我们使用@Spy注解将类A的对象spyA进行部分mock，然后使用Mockito.doReturn()方法来模拟spyA对象的add()方法返回10，这样在调用calculate()方法时，add()方法会返回10，multiply()方法会返回6，最终返回的结果为16+10=26。需要注意的是，我们使用的是spyA对象而非a对象来进行mock，以保留其它方法的真实行为。