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);
}
@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);
}
