use of org.ovirt.engine.core.common.businessentities.storage.UnregisteredDiskId in project ovirt-engine by oVirt.
the class UnregisteredDisksDaoTest method initUnregisteredDisks.
private UnregisteredDisk initUnregisteredDisks(ArrayList<VmBase> vms) {
DiskImage diskImage = new DiskImage();
diskImage.setId(Guid.newGuid());
diskImage.setDiskAlias("Disk Alias");
diskImage.setDiskDescription("Disk Description");
diskImage.setStorageIds(new ArrayList<>(Collections.singletonList(FixturesTool.STORAGE_DOMAIN_NFS2_1)));
UnregisteredDiskId id = new UnregisteredDiskId(diskImage.getId(), diskImage.getStorageIds().get(0));
return new UnregisteredDisk(id, diskImage, vms);
}
use of org.ovirt.engine.core.common.businessentities.storage.UnregisteredDiskId in project ovirt-engine by oVirt.
the class UnregisteredDisksDaoTest method testSaveAttachedDiskWithoutAliasAndDescription.
@Test
public void testSaveAttachedDiskWithoutAliasAndDescription() {
VmBase vm1 = new VmBase();
vm1.setId(Guid.newGuid());
vm1.setName("First VM");
ArrayList<VmBase> vms = new ArrayList<>();
vms.add(vm1);
// Set new disk image.
DiskImage diskImage = new DiskImage();
diskImage.setId(Guid.newGuid());
diskImage.setStorageIds(new ArrayList<>(Collections.singletonList(FixturesTool.STORAGE_DOMAIN_NFS2_1)));
UnregisteredDiskId id = new UnregisteredDiskId(diskImage.getId(), diskImage.getStorageIds().get(0));
UnregisteredDisk unregDisk = new UnregisteredDisk(id, diskImage, vms);
dao.saveUnregisteredDisk(unregDisk);
List<UnregisteredDisk> fetchedUnregisteredDisk = dao.getByDiskIdAndStorageDomainId(unregDisk.getDiskId(), FixturesTool.STORAGE_DOMAIN_NFS2_1);
assertTrue("Disk should exists in the UnregisteredDisks table", !fetchedUnregisteredDisk.isEmpty());
assertEquals("Disk should have vm attached", 1, fetchedUnregisteredDisk.get(0).getVms().size());
assertTrue("Disk alias should not be initialized", fetchedUnregisteredDisk.get(0).getDiskAlias().isEmpty());
assertTrue("Disk description should not be initialized", StringUtils.isEmpty(fetchedUnregisteredDisk.get(0).getDiskDescription()));
assertEquals("Storage Domain id should be the same as initialized", 0, fetchedUnregisteredDisk.get(0).getStorageDomainId().compareTo(FixturesTool.STORAGE_DOMAIN_NFS2_1));
}
use of org.ovirt.engine.core.common.businessentities.storage.UnregisteredDiskId in project ovirt-engine by oVirt.
the class GetUnregisteredDisksFromDBQueryTest method setUpQueryEntities.
private void setUpQueryEntities() throws OvfReaderException {
// Set up the expected result
UnregisteredDisk unregistedDisk = new UnregisteredDisk();
unregistedDisk.setId(new UnregisteredDiskId(newDiskId, storageDomainId));
unregistedDisk.setDiskAlias("DiskAlias");
unregistedDisk.setDescription("DiskDescription");
ArrayList<VmBase> vms = new ArrayList<>();
VmBase vm = new VmBase();
vm.setId(Guid.newGuid());
vm.setName("FirstVM");
vms.add(vm);
unregistedDisk.setVms(vms);
UnregisteredDisk unregistedDisk2 = new UnregisteredDisk();
unregistedDisk2.setId(new UnregisteredDiskId(newDiskId2, storageDomainId));
unregistedDisk2.setDiskAlias("DiskAlias2");
unregistedDisk2.setDescription("DiskDescription2");
List<UnregisteredDisk> expectedResultFromDB = new ArrayList<>();
expectedResultFromDB.add(unregistedDisk);
expectedResultFromDB.add(unregistedDisk2);
// Mock the Daos
when(unregisteredDisksDaoMock.getByDiskIdAndStorageDomainId(null, storageDomainId)).thenReturn(expectedResultFromDB);
}
use of org.ovirt.engine.core.common.businessentities.storage.UnregisteredDiskId in project ovirt-engine by oVirt.
the class ScanStorageForUnregisteredDisksCommand method castDiskImagesToUnregisteredDisks.
protected void castDiskImagesToUnregisteredDisks(List<DiskImage> disksFromStorage) {
if (disksFromStorage != null) {
for (DiskImage disk : disksFromStorage) {
disk.getStorageIds().set(0, getStorageDomainId());
UnregisteredDiskId id = new UnregisteredDiskId(disk.getId(), disk.getStorageIds().get(0));
UnregisteredDisk unregisteredDisk = new UnregisteredDisk(id, disk, new ArrayList<>());
unregisteredDisks.add(unregisteredDisk);
}
}
}
use of org.ovirt.engine.core.common.businessentities.storage.UnregisteredDiskId in project ovirt-engine by oVirt.
the class UnregisteredDisksDaoImpl method getByDiskIdAndStorageDomainId.
@Override
public List<UnregisteredDisk> getByDiskIdAndStorageDomainId(Guid diskId, Guid storageDomainId) {
List<UnregisteredDisk> unregisteredDisks = getCallsHandler().executeReadList("GetDiskByDiskIdAndStorageDomainId", unregisteredDiskRowMapper, createIdParameterMapper(new UnregisteredDiskId(diskId, storageDomainId)));
for (UnregisteredDisk unregDisk : unregisteredDisks) {
List<VmBase> vms = getCallsHandler().executeReadList("GetEntitiesByDiskId", vmsForUnregisteredDiskRowMapper, getCustomMapSqlParameterSource().addValue("disk_id", unregDisk.getDiskId()));
unregDisk.getVms().addAll(vms);
}
return unregisteredDisks;
}
Aggregations