Java Mockito - 如何测试调用同一类中的方法以及其他类中的方法的方法？

Question

我有这样的代码结构。 ClassA 中的方法methodToTest 正在调用ClassA 中的方法以及ClassB 中的方法

public class ClassA {
    @Autowired
    ClassB classB;

    public void methodToTest() {
        Object c = methodInA();
        Object d = classB.methodInB(c);
    }

    public Object methodInA() {
        //do something
    }
}

public class ClassB {
    public Object methodInB(Object obj) {
        //do something
    }
}

我试图模拟两个调用，即调用methodInA 和methodInB

以下是单元测试代码

@RunWith(SpringJUnit4ClassRunner.class)
@SpringApplicationConfiguration(classes = Application.class)
@WebAppConfiguration
public class ClassATest {
    @InjectMocks
    ClassA classA;

    @Mock
    ClassB classB;

    @Spy
    ClassA classASpy;

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

    //Test method in ClassA
    @Test
    public void testMethodInA() {
        Object mockObjA = mock(Object.class);
        Object mockObjB = mock(Object.class);

        doReturn(mockObjA).when(classASpy).methodInA();
        when(classB.methodInB(mockObjA)).thenReturn(mockObjB);

        classA.methodToTest();

        //assertions
    }

}

在运行测试时，Object c 为空，但我希望它等于 mockObjA。

我也从测试中尝试使用classASpy.methodToTest() 进行调用。但随后它在Object d = classB.methodInB(c) 行methodToTest() 中给出NullPointerException。

测试这种情况的正确方法是什么？

Answer 1

当您调用methodToTest 时，您正在测试单元ClassA。通过模拟该单元的某些内容来更改被测单元没有多大意义。因为那样你会测试一些与ClassA完全不同的东西。

您应该做的是测试classB.methodInB 是否使用正确的参数调用。为此，您可以使用 ArgumentCaptor 作为调用参数 classB.methodInB。

ArgumentCaptor<Object > argument = ArgumentCaptor.forClass(Object .class);
verify(mockObjB).methodInB(argument.capture());
Object capturedArgument = argument.getValue();

然后您对capturedArgument 进行断言，以验证ClassA.methodInA 是否正常工作。