java - Mockito - 方法内的模拟对象

Question

我有一个返回车辆列表的方法。像这样:

 public List<Vehicle> getVehicles() {

        List<Vehicle> vehicles=vehicleDAO.getAllVehicles();

        for (Vehicle v : vehicles){//NullPointerException
            //some bussines logic...
        }       
        return vehicles;

}

这是我的测试:

@Test
public void testShowVehicles() {
    when(vehicleDAO.getAllVehicles()).thenReturn(listVehiclesMock);
    List<Vehicle> vehicles= service.getVehicles();//NullPointerException
    assertEquals(listVehicleMock, vehicles);
}

当我运行它时，我得到 NullPointerException 因为车辆不存在。当我使用旧式 for 循环时，它通过了测试，但现在我用 forEach 循环替换，我在测试中遇到错误。那么我该如何模拟 Vehicle 对象呢？

Answer 1

对于每个循环，都使用给定Iterable 的iterator() 方法。然后使用以这种方式获得的迭代器来迭代集合。不幸的是，模拟列表的这个方法返回 null，这会导致 NullPointerException。要使用每个循环语法，您还必须将 iterator() 方法绑定(bind)到结果。