如何在 java 中返回迭代器的同一类中模拟方法

Question

我正在尝试对一个方法getFruitItemMap()进行单元测试，该方法调用另一个返回类型为Iterator在同一类中getAllItemsBelongingToFruit的方法。我正在尝试使用 Mockito.spy() 来模拟此方法，但不确定如何返回迭代器。我在这里检查了堆栈溢出的其他答案，但看起来我在这里遗漏了一些东西。

class FruitItemImpl {
Map<String, Fruit> getFruitItemMap() {
        Map<String, Fruit> fruitMap = new HashMap<>();
        Iterator<Fruit> items = getAllItemsBelongingToFruit("Apple");
while (items.hasNext()) {
            Fruit fruitItem = items.next();
            fruitMap.put(fruitItem.getID(), fruitItem);
        }
        return fruitMap;
    }
public Iterator<Fruit> getAllItemsBelongingToFruit(String fruit) {
      //some logic that returns an iterator
}

这是单元测试：

    @Test
    public void testGetFruitItemMap() {
        Map<String, Fruit> fruitItemMap = new HashMap<>();
        FruitItemImpl doa1 = Mockito.spy(dao);
   Mockito.doReturn(**new Iterator<Fruit>**).when(doa1).getAllItemsBelongingToFruit("Apple") //Here
        Assert.assertEquals(fruitItemMap.size(), doa1.getFruitItemMap().size());
    }

由于我是Mockito和单元测试世界的新手，因此试图绕过它。在这种情况下，我不确定如何使 mockito 返回迭代器Mockito.doReturn()。

Answer 1

最简单的方法是拥有相应Fruit对象的集合，并从中返回一个Iterator。在您的情况下，您可以在测试中使用的fruitItemMap values()返回一个迭代器：

@Test
public void testGetFruitItemMap() {
    Map<String, Fruit> fruitItemMap = new HashMap<>();
    FruitItemImpl doa1 = Mockito.spy(dao);
    Mockito.doReturn(fruitItemMap.values().iterator()) // Here!
           .when(doa1).getAllItemsBelongingToApple("Apple");
    Assert.assertEquals(fruitItemMap.size(), doa1.getFruitItemMap().size());
}