private void mockGetVmsListForDisk(List<VM> vms) {
List<Pair<VM, VmDevice>> vmsWithVmDevice = new ArrayList<>();
if (vms != null) {
for (VM vm : vms) {
VmDevice device = createVmDevice(diskImageGuid, vm.getId());
vmsWithVmDevice.add(new Pair<>(vm, device));
}
} else {
vms = Collections.emptyList();
}
when(vmDao.getVmsWithPlugInfo(diskImageGuid)).thenReturn(vmsWithVmDevice);
when(vmDao.getVmsListForDisk(diskImageGuid, true)).thenReturn(vms);
}
