doAnswer（）來修改傳遞的對象沒有被要求虛空方法

Question

使用案例：

我想打電話給使用對象的名單嘲笑類的方法，並希望修改傳遞的參數。它工作正常，如果我工作的返回類型的一些方法，並不適用於Void返回方法。

方法測試：

public class TestClass{ 
        @Autowired 

       PersistActivity activity; 

       public void methodToTest(List<URLObject> urlObjects) { 
          List<URLObject> urlObjects2PerformAction = new ArrayList<URLObject>(); 

        for (URLObject urlObject : urlObjects) { 
          // Added this check to make sure, we are not running a previous 
          // activity which is successful. 
          if (urlObject.getRetryCountMap().get(currentActivityName) != null) { 
            urlObjects2PerformAction.add(urlObject); 
          } 
        } 
        boolean activityStatus = true; 
        try { 
         log.info("Calling activity " + currentActivityName + " with " + urlObjects2PerformAction); 
         activity.doActivity(urlObjects2PerformAction); 
         log.info("UrlObjects are " + urlObjects2PerformAction); 
        } catch (Exception e) { 
         //another set of actions. 
        } 
        for (URLObject urlObject : urlObjects) { 
         if (!urlObject.isActivitySuccessful()) { 
           // If complete call failed or partial data failed, update 
           log.info("Failed for message " + urlObject); 
         } else { 
          log.info("Passed for message " + urlObject); 
         } 
        }//end for 
      }//end method 
     }//end class  

在這個例子中，我想修改urlObjects來填充它的布爾變量之一是真/假。

我已經在測試情況下加入一個doAnswer()子句是這樣的：

測試類：

public class TestMe{ 
    @Mock 
    private PersistActivity activity; 

    @Captor 
    private ArgumentCaptor<List<URLObject>> argument; 

    @InjectMocks 
    private DCUrlPersistor dcUrlPersisor = new DCUrlPersistor(); 


    @Test 
public void runIndividualDeamonWithMultipleMessagesSomeFailing() throws Exception { 


    URLObject urlObject1 = getUrlObject(normalizedUrl, normalizedUrl, scraperName); 
    URLObject urlObject2 = getUrlObject("test2.com", "test2.com", "gpsUS"); 


Map<String,Integer> retryCountMap = new HashMap<String, Integer>(); 
retryCountMap.put(PersisterActivitiesEnum.CSI.getName(), 0); 

urlObject1.setRetryCountMap(retryCountMap); 
urlObject2.setRetryCountMap(retryCountMap); 

/** 
* Change argument passed to the method. 
*/  
doAnswer(new Answer<Object>() { 

    @Override 
    public Object answer(InvocationOnMock invocation) 
     throws Throwable 
    { 
     Object[] arguments = invocation.getArguments(); 

     if ( arguments != null 
       && 
       arguments.length > 0 
       && 
       arguments[0] != null) 
     { 
      List<URLObject> objs = (List<URLObject>) arguments[0]; 
      Assert.assertNotNull(objs.get(0).getRetryCountMap()); 
      objs.get(0).setActivitySuccessful(false); 
     } 
     return null; 
    }  
}).when(activity).doActivity(anyList()); 


doNothing().when(activity).doActivity(Arrays.asList(urlObject1, urlObject2)); 

/** 
* Call method. 
*/ 
dcUrlPersisor.methodToTest(Arrays.asList(urlObject1, urlObject2)); 
/** 
* Verify various instances and logs. 
*/ 
verify(activity, times(1)).doActivity(argument.capture()); 
assertEquals(2, argument.getValue().size()); 
assertEquals(normalizedUrl, argument.getValue().get(0).getNormalizedURL());  
Assert.assertTrue(logContains("Calling activity csi with ")); 
//-----This is the part which fails as log never prints this message----// 
Assert.assertTrue(logContains("Failed for message")); 


    } 
} 

問題：

如果更改doActivity方法的返回類型爲List<ABC>比Void ，那麼它會被調用並按預期工作（不會更改測試代碼）。

我們是否需要對void方法進行一些更改？我已經將它設置爲在doAnswer中返回void。

任何幫助，將不勝感激。

Answer 1

嘗試刪除以下行：

doNothing().when(activity).doActivity(Arrays.asList(urlObject1, urlObject2)); 

對我來說，這似乎有點違背設立doAnswer它前面。

Answer 2

將Void更改爲Object這樣。

... 
doAnswer(new Answer<Object>() { 
    @Override 
    public Object answer(InvocationOnMock invocation) throws Throwable { 
    ...