小编fee*_*kie的帖子

如何使用 Mockito 模拟 forEach 行为

我想做以下工作,但我不知道如何正确模拟 forEach 行为。(该代码取自一个相关问题Testing Java Enhanced for behavior with Mockito

@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());
}
Run Code Online (Sandbox Code Playgroud)

任何帮助将不胜感激。

java mocking mockito

6
推荐指数
2
解决办法
2万
查看次数

标签 统计

java ×1

mocking ×1

mockito ×1