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工作正常。