Examples with Disk org.ovirt.engine.core.common.businessentities.storage.Disk used on opensource projects

Example 56 with Disk

use of org.ovirt.engine.core.common.businessentities.storage.Disk in project ovirt-engine by oVirt.

the class GetAllDisksByVmIdQueryTest method setUpDaoMocks.

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())).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());
}

Example 57 with Disk

use of org.ovirt.engine.core.common.businessentities.storage.Disk in project ovirt-engine by oVirt.

the class GetDiskAndSnapshotsByDiskIdQueryTest method testQueryWithLunDisk.

@Test
public void testQueryWithLunDisk() {
    Disk disk = executeQuery(lunDisk);
    assertTrue("disk should be from type LunDisk", disk instanceof LunDisk);
}

Example 58 with Disk

use of org.ovirt.engine.core.common.businessentities.storage.Disk in project ovirt-engine by oVirt.

the class GetNextAvailableDiskAliasNameByVMIdQueryTest method populateVmDiskMap.

/**
 * Populates the VM disk map with the amount of disks specified, each with a default disk alias
 */
private static void populateVmDiskMap(VM vm, int numOfDisks) {
    Map<Guid, Disk> diskMap = vm.getDiskMap();
    for (Integer i = 0; i < numOfDisks; i++) {
        DiskImage diskImage = new DiskImage();
        diskImage.setId(Guid.newGuid());
        diskImage.setDiskAlias(ImagesHandler.getDefaultDiskAlias(vm.getName(), Integer.toString(i + 1)));
        diskMap.put(diskImage.getId(), diskImage);
    }
}

Example 59 with Disk

use of org.ovirt.engine.core.common.businessentities.storage.Disk in project ovirt-engine by oVirt.

the class RestoreAllSnapshotsCommandTest method mockDiskImageDao.

/**
 * Mock disk image Dao.
 */
private void mockDiskImageDao() {
    List<Disk> diskImageList = new ArrayList<>();
    DiskImage diskImage = new DiskImage();
    diskImage.setStorageIds(new ArrayList<>(Collections.singletonList(Guid.newGuid())));
    diskImageList.add(diskImage);
    when(diskDao.getAllForVm(vmId)).thenReturn(diskImageList);
}

Example 60 with Disk

use of org.ovirt.engine.core.common.businessentities.storage.Disk in project ovirt-engine by oVirt.

the class AddVmAndCloneImageCommand method buildAndCheckDestStorageDomains.

@Override
protected boolean buildAndCheckDestStorageDomains() {
    if (diskInfoDestinationMap.isEmpty()) {
        List<StorageDomain> domains = storageDomainDao.getAllForStoragePool(getStoragePoolId());
        Map<Guid, StorageDomain> storageDomainsMap = new HashMap<>();
        for (StorageDomain storageDomain : domains) {
            StorageDomainValidator validator = new StorageDomainValidator(storageDomain);
            if (validate(validator.isDomainExistAndActive()) && validate(validator.domainIsValidDestination())) {
                storageDomainsMap.put(storageDomain.getId(), storageDomain);
            }
        }
        for (Disk disk : getDiskImagesToBeCloned()) {
            DiskImage image = (DiskImage) disk;
            for (Guid storageId : image.getStorageIds()) {
                if (storageDomainsMap.containsKey(storageId)) {
                    diskInfoDestinationMap.put(image.getId(), image);
                    break;
                }
            }
        }
        if (getDiskImagesToBeCloned().size() != diskInfoDestinationMap.size()) {
            logErrorOneOrMoreActiveDomainsAreMissing();
            return false;
        }
        List<Guid> storageDomainDest = new ArrayList<>();
        for (DiskImage diskImage : diskInfoDestinationMap.values()) {
            Guid storageDomainId = diskImage.getStorageIds().get(0);
            if (storageDomainDest.contains(storageDomainId)) {
                destStorages.put(storageDomainId, storageDomainsMap.get(storageDomainId));
            }
            storageDomainDest.add(storageDomainId);
        }
        return true;
    }
    return super.buildAndCheckDestStorageDomains();
}