在某些場景中，不光要對(duì)方法的返回值和調(diào)用進(jìn)行驗(yàn)證，同時(shí)需要驗(yàn)證一系列交互后所傳入方法的參數(shù)。那么我們可以用參數(shù)捕獲器來捕獲傳入方法的參數(shù)進(jìn)行驗(yàn)證，看它是否符合我們的要求。

ArgumentCaptor介紹

通過ArgumentCaptor對(duì)象的forClass(Class<T> clazz)方法來構(gòu)建ArgumentCaptor對(duì)象。然后便可在驗(yàn)證時(shí)對(duì)方法的參數(shù)進(jìn)行捕獲，最后驗(yàn)證捕獲的參數(shù)值。如果方法有多個(gè)參數(shù)都要捕獲驗(yàn)證，那就需要?jiǎng)?chuàng)建多個(gè)ArgumentCaptor對(duì)象處理。

ArgumentCaptor的API

argument.capture() 捕獲方法參數(shù)
argument.getValue() 獲取方法參數(shù)值，如果方法進(jìn)行了多次調(diào)用，它將返回最后一個(gè)參數(shù)值
argument.getAllValues() 方法進(jìn)行多次調(diào)用后，返回多個(gè)參數(shù)值

應(yīng)用實(shí)例

@Test  
public void argumentCaptorTest() {  
    List mock = mock(List.class);  
    List mock2 = mock(List.class);  
    mock.add("John");  
    mock2.add("Brian");  
    mock2.add("Jim");  
      
    ArgumentCaptor argument = ArgumentCaptor.forClass(String.class);  
      
    verify(mock).add(argument.capture());  
    assertEquals("John", argument.getValue());  
      
    verify(mock2, times(2)).add(argument.capture());  
  
    assertEquals("Jim", argument.getValue());  
    assertArrayEquals(new Object[]{"Brian","Jim"},argument.getAllValues().toArray());  
}  

首先構(gòu)建ArgumentCaptor需要傳入捕獲參數(shù)的對(duì)象，例子中是String。接著要在verify方法的參數(shù)中調(diào)用argument.capture()方法來捕獲輸入的參數(shù)，之后argument變量中就保存了參數(shù)值，可以用argument.getValue()獲取。當(dāng)某個(gè)對(duì)象進(jìn)行了多次調(diào)用后，如mock2對(duì)象，這時(shí)調(diào)用argument.getValue()獲取到的是最后一次調(diào)用的參數(shù)。如果要獲取所有的參數(shù)值可以調(diào)用argument.getAllValues()，它將返回參數(shù)值的List。