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); }
@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(!fetchedUnregisteredDisk.isEmpty(), "Disk should exists in the UnregisteredDisks table"); assertEquals(1, fetchedUnregisteredDisk.get(0).getVms().size(), "Disk should have vm attached"); assertTrue(fetchedUnregisteredDisk.get(0).getDiskAlias().isEmpty(), "Disk alias should not be initialized"); assertTrue(StringUtils.isEmpty(fetchedUnregisteredDisk.get(0).getDiskDescription()), "Disk description should not be initialized"); assertEquals(0, fetchedUnregisteredDisk.get(0).getStorageDomainId().compareTo(FixturesTool.STORAGE_DOMAIN_NFS2_1), "Storage Domain id should be the same as initialized"); } }