private void mockToUpdateDiskVm(List<VM> vms) { for (VM vm : vms) { if (vm.getId().equals(command.getParameters().getVmId())) { when(vmDao.get(command.getParameters().getVmId())).thenReturn(vm); break; } } }
private void mockNullVm() { mockGetForDisk((VM) null); mockGetVmsListForDisk(null); when(vmDao.get(command.getParameters().getVmId())).thenReturn(null); }