如何用Mockito嘲笑每个行为

Question

我想使以下工作，但我不知道如何正确地模拟每个行为。(代码取自一个相关问题 用 Mockito 测试 Java 增强的行为)

@Test
public void aa() {
Collection<String> fruits;
Iterator<String> fruitIterator;
fruitIterator = mock(Iterator.class);
when(fruitIterator.hasNext()).thenReturn(true, true, true, false);
when(fruitIterator.next()).thenReturn("Apple")
.thenReturn("Banana").thenReturn("Pear");
fruits = mock(Collection.class);
when(fruits.iterator()).thenReturn(fruitIterator);
doCallRealMethod().when(fruits).forEach(any(Consumer.class));
// this doesn't work (it doesn't print anything)
fruits.forEach(f -> {
mockObject.someMethod(f); 
});
// this works fine
/*
int iterations = 0;
for (String fruit : fruits) {
mockObject.someMethod(f); 
}
*/
// I want to verify something like this
verify(mockObject, times(3)).someMethod(anyString());
}

任何帮助将不胜感激。

Answer 1

Iterator mockIterator = mock(Iterator.class);
doCallRealMethod().when(fruits).forEach(any(Consumer.class));
when(fruits.iterator()).thenReturn(mockIterator);
when(mockIterator.hasNext()).thenReturn(true, false);
when(mockIterator.next()).thenReturn(mockObject);

Answer 2

Collection接口的方法forEach是一个"defender"方法;它不使用Iterator而是调用传递给该方法的Consumer。

如果您使用的是 Mockito 版本 2+ (*)，则可以询问要调用的Collection接口的默认方法forEach：

Mockito.doCallRealMethod().when(fruits).forEach(Mockito.any(Consumer.class));

请注意，"defender"方法实际上将请求一个Iterator来遍历集合，因此将使用您在测试中模拟的Iterator。如果模拟集合没有提供Iterator，它就行不通，就像when(fruits.iterator()).thenReturn(fruitIterator)

(*)：从版本 2 开始，Mockito 增加了支持 Java 8 默认("防御者")方法的可能性 - 您可以在此处检查跟踪问题。