protected void assertQuotaUnlimited(Quota quotaUnlimited) {
assertEquals(QuotaHelper.UNLIMITED, quotaUnlimited.getGlobalQuotaStorage().getStorageSizeGB());
assertEquals(
QuotaHelper.UNLIMITED.intValue(),
quotaUnlimited.getGlobalQuotaVdsGroup().getVirtualCpu().intValue());
assertEquals(QuotaHelper.UNLIMITED, quotaUnlimited.getGlobalQuotaVdsGroup().getMemSizeMB());
assertEquals(true, quotaUnlimited.getIsDefaultQuota());
}
@Test
public void testGetUnlimitedQuota() throws Exception {
Quota quotaUnlimited = quotaHelper.getUnlimitedQuota(mockStoragePool(), false);
assertEquals(QuotaHelper.UNLIMITED, quotaUnlimited.getGlobalQuotaStorage().getStorageSizeGB());
assertEquals(
QuotaHelper.UNLIMITED.intValue(),
quotaUnlimited.getGlobalQuotaVdsGroup().getVirtualCpu().intValue());
assertEquals(QuotaHelper.UNLIMITED, quotaUnlimited.getGlobalQuotaVdsGroup().getMemSizeMB());
}
private Quota mockUnlimitedQuota() {
Quota generalQuota = new Quota();
generalQuota.setDescription("New Quota to create");
generalQuota.setQuotaName(quotaName);
QuotaStorage storageQuota = new QuotaStorage();
storageQuota.setStorageSizeGB(QuotaHelper.UNLIMITED);
storageQuota.setStorageSizeGBUsage(0d);
generalQuota.setGlobalQuotaStorage(storageQuota);
QuotaVdsGroup vdsGroupQuota = new QuotaVdsGroup();
vdsGroupQuota.setVirtualCpu(QuotaHelper.UNLIMITED.intValue());
vdsGroupQuota.setVirtualCpuUsage(0);
vdsGroupQuota.setMemSizeMB(QuotaHelper.UNLIMITED);
vdsGroupQuota.setMemSizeMBUsage(0l);
generalQuota.setGlobalQuotaVdsGroup(vdsGroupQuota);
generalQuota.setId(Guid.NewGuid());
generalQuota.setStoragePoolId(storagePoolUUID);
generalQuota.setIsDefaultQuota(true);
return generalQuota;
}
