private DiskImage mockDiskImage(DiskInterface iface) {
    disk.setImageId(diskImageGuid);
    ArrayList<Guid> storageIdList = new ArrayList<>();
    storageIdList.add(storageDomainId);
    disk.setStorageIds(storageIdList);
    disk.setStoragePoolId(storagePoolId);
    disk.setActive(true);
    disk.setId(Guid.newGuid());

    when(diskDao.get(diskImageGuid)).thenReturn(disk);
    DiskVmElement dve = new DiskVmElement(disk.getId(), vmId);
    dve.setDiskInterface(iface);
    when(diskVmElementDao.get(new VmDeviceId(disk.getId(), vmId))).thenReturn(dve);
    return disk;
  }
  private void setUpDaoMocks() {
    // Disk Image Dao
    List<Disk> returnArray = new ArrayList<>();
    returnArray.add(pluggedDisk);
    returnArray.add(unpluggedDisk);
    returnArray.add(pluggedDiskSnapshot);
    returnArray.add(unpluggedDiskSnapshot);

    when(diskDaoMock.getAllForVm(vmID, getUser().getId(), getQueryParameters().isFiltered()))
        .thenReturn(returnArray);

    when(diskVmElementDao.get(any(VmDeviceId.class)))
        .thenReturn(new DiskVmElement(new VmDeviceId()));

    // Snapshots
    doReturn(
            new ArrayList<>(
                Collections.nCopies(
                    NUM_DISKS_OF_EACH_KIND, createDiskSnapshot(pluggedDisk.getId()))))
        .when(diskImageDao)
        .getAllSnapshotsForLeaf(pluggedDisk.getImageId());
    doReturn(Collections.nCopies(NUM_DISKS_OF_EACH_KIND, createDiskSnapshot(unpluggedDisk.getId())))
        .when(diskImageDao)
        .getAllSnapshotsForLeaf(unpluggedDisk.getImageId());
    doReturn(
            new ArrayList<>(
                Collections.nCopies(
                    NUM_DISKS_OF_EACH_KIND, createDiskSnapshot(pluggedDiskSnapshot.getId()))))
        .when(diskImageDao)
        .getAllSnapshotsForLeaf(pluggedDiskSnapshot.getImageId());
    doReturn(
            Collections.nCopies(
                NUM_DISKS_OF_EACH_KIND, createDiskSnapshot(unpluggedDiskSnapshot.getId())))
        .when(diskImageDao)
        .getAllSnapshotsForLeaf(unpluggedDiskSnapshot.getImageId());
  }