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

Question

我有这样的代码结构。方法methodToTest在ClassA呼吁在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()进行调用。但是，它在methodToTest()的Object d = classB.methodInB(c)处给出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工作正常。