java - Mockito When().thenReturn 返回一个迭代器

Question

我试图从when().theReturn 返回一个迭代器，但我不断收到此错误:

org.mockito.exceptions.misusing.WrongTypeOfReturnValue: 
Itr cannot be returned by findAll()
findAll() should return List

这就是我想做的:

List<Client> iterList = mockClientList1.findAll();
final Iterator<Client> iter = newMockListClient.iterator();
when(iterList.iterator()).thenReturn(iter);

mockClientList1是一个ClientList对象，findAll()是一个返回Client列表的方法。我看到一篇关于 Mockito 关于 when.thenReturn 上堆叠方法的限制的帖子，但我不确定这是否是失败的原因？任何提示将非常感激。

Answer 1

如果你想这样做，你还需要为 findAll() 的返回值提供一个 stub 。

List<Client> mockList = mock(List.class)
when(mockClientList1.findAll()).thenReturn(mockList);
final Iterator<Client> iter = newMockListClient.iterator();
when(mockList.iterator()).thenReturn(iter);

但是除非有特定原因您只想仅 stub 迭代器，否则您也可以直接返回列表:

when(mockClientList1.findAll()).thenReturn(mockListClient);