private void mockStorageDomains() { // Set list of storages domain to have sum of available disk size enough for the quota // limitations. List<storage_domains> storageDomains = new ArrayList<storage_domains>(); firstStorageDomains.setused_disk_size(100); firstStorageDomains.setavailable_disk_size(50); secondStorageDomains.setused_disk_size(100); secondStorageDomains.setavailable_disk_size(50); storageDomains.add(firstStorageDomains); storageDomains.add(secondStorageDomains); when(storageDomainDAO.getAllForStoragePool(storagePoolUUID)).thenReturn(storageDomains); }
@Override public storage_domains getStorageDomain() { storage_domains sd = new storage_domains(); sd.setstatus(StorageDomainStatus.Active); sd.setavailable_disk_size(10); return sd; }
@Test public void testValidSpecificStorageLimitation() throws Exception { List<String> messages = new ArrayList<String>(); firstStorageDomains.setused_disk_size(0); firstStorageDomains.setavailable_disk_size(100); Quota quota = mockGeneralStorageQuota(); quota.setIsDefaultQuota(false); when(storageDomainDAO.getForStoragePool(firstStorageDomainUUID, storagePoolUUID)) .thenReturn(firstStorageDomains); boolean isQuotaValid = quotaHelper.checkQuotaValidationForAdd(quota, messages); assertTrue(isQuotaValid); isQuotaValid = quotaHelper.checkQuotaValidationForEdit(quota, messages); assertTrue(isQuotaValid); }