Mockito：如何在迭代列表時獲取模擬對象？

Question

@Mock 
Object obj; 

List<Obejct obj> objs; 

如何獲取模擬對象obj當我迭代objs？

Answer 1

代碼：

@Mock 
private Iterator<YourObject> iteratorMock; 
//... 
when(yourListMock.iterator()).thenReturn(iteratorMock); 
when(iteratorMock.hasNext()).thenReturn(Boolean.TRUE, Boolean.TRUE, 
    Boolean.TRUE, Boolean.FALSE); 
when(iteratorMock.next()).thenReturn(yourMock1, yourMock1, yourMock2); 

---

正如你可以在代碼中看到，嘲諷的迭代器是爲第二，第三，e.t.c調用返回不同的嘲弄一個非常方便的方式。

這樣你也可以很容易地讓第二次調用拋出異常，第三次調用再次工作，能夠測試第二次調用的異常將被處理，例如第三次結果仍然會被處理。

使用真正的列表是一個選項，但我認爲嘲笑更強大，代碼更緊密地針對您實際需要的特定測試。