/**
* Make Quota specific to be the same as Quota general and specific.
*
* @throws Exception
*/
@Test
public void testUpdateQuota() throws Exception {
Quota quotaGeneralToSpecific = dao.getById(FixturesTool.QUOTA_GENERAL);
// Save quotaName and vdsGroup list for future check.
String quotaName = "New Temporary name";
List<QuotaVdsGroup> quotaVdsGroupList =
getQuotaVdsGroup(getSpecificQuotaVdsGroup(quotaGeneralToSpecific.getId()));
Long newStorageLimit = new Long("2345");
// Check before the update, that the fields are not equal.
assertEquals(quotaName.equals(quotaGeneralToSpecific.getQuotaName()), false);
assertEquals(
quotaVdsGroupList.size() == quotaGeneralToSpecific.getQuotaVdsGroups().size(), false);
assertEquals(
quotaGeneralToSpecific.getGlobalQuotaStorage().getStorageSizeGB().equals(newStorageLimit),
false);
// Update
quotaGeneralToSpecific.setQuotaName(quotaName);
quotaGeneralToSpecific.getGlobalQuotaStorage().setStorageSizeGB(newStorageLimit);
quotaGeneralToSpecific.setQuotaVdsGroups(quotaVdsGroupList);
dao.update(quotaGeneralToSpecific);
quotaGeneralToSpecific = dao.getById(FixturesTool.QUOTA_GENERAL);
// Check after the update, that the fields are equal now.
assertEquals(quotaName.equals(quotaGeneralToSpecific.getQuotaName()), true);
assertEquals(
quotaVdsGroupList.size() == quotaGeneralToSpecific.getQuotaVdsGroups().size(), true);
assertEquals(
quotaGeneralToSpecific.getGlobalQuotaStorage().getStorageSizeGB().equals(newStorageLimit),
true);
}
@Test
public void testGeneralQuotaLimitations() throws Exception {
// Set new Quota definition.
Quota quota = new Quota();
Guid quotaId = Guid.NewGuid();
quota.setId(quotaId);
quota.setStoragePoolId(FixturesTool.STORAGE_POOL_NFS);
quota.setQuotaName("Watson");
quota.setDescription("General quota");
quota.setThresholdVdsGroupPercentage(
Config.<Integer>GetValue(ConfigValues.QuotaThresholdVdsGroup));
quota.setThresholdStoragePercentage(
Config.<Integer>GetValue(ConfigValues.QuotaThresholdStorage));
quota.setGraceVdsGroupPercentage(Config.<Integer>GetValue(ConfigValues.QuotaGraceVdsGroup));
quota.setGraceStoragePercentage(Config.<Integer>GetValue(ConfigValues.QuotaGraceStorage));
setQuotaGlobalLimitations(quota);
quota.setQuotaVdsGroups(getQuotaVdsGroup(null));
quota.setQuotaStorages(getQuotaStorage(null));
dao.save(quota);
Quota quotaEntity = dao.getById(quota.getId());
assertNotNull(quotaEntity);
assertEquals(quotaEntity, quota);
assertEquals(quotaEntity.getStoragePoolName(), "rhel6.NFS");
assertEquals(quotaEntity.getQuotaEnforcementType(), QuotaEnforcementTypeEnum.DISABLED);
}
@Test
public void testRemoveQuota() throws Exception {
Quota quota = dao.getById(FixturesTool.QUOTA_SPECIFIC_AND_GENERAL);
assertNotNull(quota);
dao.remove(FixturesTool.QUOTA_SPECIFIC_AND_GENERAL);
assertEquals(null, dao.getById(FixturesTool.QUOTA_SPECIFIC_AND_GENERAL));
assertEquals(
0, dao.getQuotaVdsGroupByQuotaGuid(FixturesTool.QUOTA_SPECIFIC_AND_GENERAL).size());
assertEquals(0, dao.getQuotaStorageByQuotaGuid(FixturesTool.QUOTA_SPECIFIC_AND_GENERAL).size());
}
/**
* Comparing the two previous test results.<br>
* Since there is only global limitation for specific quota that is being checked, the results for
* fetching per vds group or fetching per storage pool should be the same.
*
* @throws Exception
*/
@Test
public void testCompareFetchGlobalQuotaForSpecificAndForGlobalVdsGroup() throws Exception {
List<QuotaVdsGroup> quotaVdsGroupGlobalList =
dao.getQuotaVdsGroupByVdsGroupGuid(null, FixturesTool.QUOTA_GENERAL);
List<QuotaVdsGroup> quotaVdsGroupSpecificList =
dao.getQuotaVdsGroupByVdsGroupGuid(
FixturesTool.VDS_GROUP_RHEL6_ISCSI, FixturesTool.QUOTA_GENERAL);
// Check if the global variable returns when null is initialization.
assertEquals(true, quotaVdsGroupGlobalList.equals(quotaVdsGroupSpecificList));
}
/**
* Test get Quota by Name, with name of specific Quota.
*
* @throws Exception
*/
@Test
public void testGetQuotaByExistingName() throws Exception {
Quota quotaGeneralToSpecific = dao.getQuotaByQuotaName("Quota General");
assertEquals(
dao.getById(FixturesTool.QUOTA_GENERAL)
.getQuotaName()
.equals(quotaGeneralToSpecific.getQuotaName()),
true);
assertEquals(
dao.getById(FixturesTool.QUOTA_GENERAL).getId().equals(quotaGeneralToSpecific.getId()),
true);
}
/**
* Test {@link QuotaDAO#getAllRelevantQuotasForStorage(Guid)} fetching quota for user without
* privileges for quota.
*/
@Test
public void testGetRelevantStorageQuotaForUserWithoutPrivileges() throws Exception {
List<Quota> quotas =
dao.getAllRelevantQuotasForStorage(
FixturesTool.STORAGE_DOAMIN_NFS_MASTER, UNPRIVILEGED_USER_ID, true);
assertEquals("Unprivileged user is not allowed to fetch for quota", 0, quotas.size());
}
/**
* Asserts that when {@link QuotaDAO#getQuotaStorageByQuotaGuidWithGeneralDefault(Guid)} is called
* with an empty quota, no storage quotas are returned
*/
@Test
public void testQuotaStorageByQuotaGuidWitheGeneralDefaultWithEmpty() {
List<QuotaStorage> quotaStorageList =
dao.getQuotaStorageByQuotaGuidWithGeneralDefault(FixturesTool.QUOTA_EMPTY);
assertNotNull(quotaStorageList);
assertEquals("wrong number of quotas returned", 0, quotaStorageList.size());
}
/**
* Test {@link QuotaDAO#getAllRelevantQuotasForVdsGroup(Guid)} fetching quota for user without
* privileges for quota.
*/
@Test
public void testGetRelevantVdsGroupQuotaForUserWithoutPrivileges() throws Exception {
List<Quota> quotas =
dao.getAllRelevantQuotasForVdsGroup(
FixturesTool.VDS_GROUP_RHEL6_NFS, UNPRIVILEGED_USER_ID, true);
assertEquals("Unprivileged user is not allowed to fetch for quota", 0, quotas.size());
}
/**
* Test scenario when there is a specific limitation on the storage pool, and we check limitation
* on the entire storage pool.<br>
* The value that should be returned, is the specific limitations and the specific usage on the
* storage pool.
*
* @throws Exception
*/
@Test
public void testFetchSpecificQuotaUsageForGlobalVdsGroup() throws Exception {
List<QuotaVdsGroup> quotaVdsGroupList =
dao.getQuotaVdsGroupByVdsGroupGuid(null, FixturesTool.QUOTA_SPECIFIC);
QuotaVdsGroup quotaVdsGroup = quotaVdsGroupList.get(0);
assertEquals(true, quotaVdsGroupList.size() == 2);
assertNotNull(quotaVdsGroup);
}
/**
* Test get Quota by Name, with name of specific Quota.
*
* @throws Exception
*/
@Test
public void testGetQuotaByAdElementId() throws Exception {
List<Quota> quotaByAdElementIdList =
dao.getQuotaByAdElementId(FixturesTool.USER_EXISTING_ID, FixturesTool.STORAGE_POOL_NFS);
// Check if quota general has been fetched.
assertEquals(quotaByAdElementIdList.get(0).getQuotaName(), "Quota General");
}
@Test
public void testFetchSpecificAndGeneralQuotaForStorage() throws Exception {
List<QuotaStorage> quotaStorageList =
dao.getQuotaStorageByStorageGuid(null, FixturesTool.QUOTA_SPECIFIC_AND_GENERAL);
QuotaStorage quotaStorage = quotaStorageList.get(0);
assertEquals(true, quotaStorageList.size() == 1);
assertNotNull(quotaStorage);
// Check if the global variable returns when null is initialization.
assertEquals(true, quotaStorage.getStorageSizeGBUsage() > 0);
}
/**
* Asserts that when {@link QuotaDAO#getQuotaVdsGroupByQuotaGuidWithGeneralDefault(Guid)} is
* called with a non-specific quota, the general is returned
*/
@Test
public void testQuotaVdsGroupByQuotaGuidWithGeneralDefaultWithDefault() {
List<QuotaVdsGroup> quotaVdsGroupList =
dao.getQuotaVdsGroupByQuotaGuidWithGeneralDefault(FixturesTool.QUOTA_GENERAL);
assertNotNull(quotaVdsGroupList);
assertEquals("wrong number of quotas returned", 1, quotaVdsGroupList.size());
for (QuotaVdsGroup group : quotaVdsGroupList) {
assertEquals("VDS ID should be empty in general mode", Guid.Empty, group.getVdsGroupId());
assertNull("VDS name should be null in general mode", group.getVdsGroupName());
}
}
/**
* Asserts that when {@link QuotaDAO#getQuotaVdsGroupByQuotaGuidWithGeneralDefault(Guid)} is
* called with a specific quota, all the relevant VDSs are returned
*/
@Test
public void testQuotaVdsGroupByQuotaGuidWithGeneralDefaultNoDefault() {
List<QuotaVdsGroup> quotaVdsGroupList =
dao.getQuotaVdsGroupByQuotaGuidWithGeneralDefault(FixturesTool.QUOTA_SPECIFIC);
assertNotNull(quotaVdsGroupList);
assertEquals("wrong number of quotas returned", 2, quotaVdsGroupList.size());
for (QuotaVdsGroup group : quotaVdsGroupList) {
assertNotNull("VDS ID should not be null in specific mode", group.getVdsGroupId());
assertNotNull("VDS name should not be null in specific mode", group.getVdsGroupName());
}
}
/**
* Asserts that when {@link QuotaDAO#getQuotaStorageByQuotaGuidWithGeneralDefault(Guid)} is called
* with a specific quota, all the relevant storages are returned
*/
@Test
public void testQuotaStorageByQuotaGuidWithGeneralDefaultNoDefault() {
List<QuotaStorage> quotaStorageList =
dao.getQuotaStorageByQuotaGuidWithGeneralDefault(FixturesTool.QUOTA_SPECIFIC);
assertNotNull(quotaStorageList);
assertEquals("wrong number of quotas returned", 1, quotaStorageList.size());
for (QuotaStorage group : quotaStorageList) {
assertNotNull("Storage ID should not be null in specific mode", group.getStorageId());
assertNotNull("Storage name should not be null in specific mode", group.getStorageName());
}
}
/**
* Test scenario when there is a specific limitation on the storage pool, and we check if there is
* enough resources on a specific vds group.<br>
* The returned value from the query, should be the specific limitation and the specific usage on
* the vds group.
*
* @throws Exception
*/
@Test
public void testFetchSpecificQuotaUsageForSpecificVdsGroup() throws Exception {
List<QuotaVdsGroup> quotaVdsGroupList =
dao.getQuotaVdsGroupByVdsGroupGuid(
FixturesTool.VDS_GROUP_RHEL6_ISCSI, FixturesTool.QUOTA_SPECIFIC);
QuotaVdsGroup quotaVdsGroup = quotaVdsGroupList.get(0);
assertNotNull(quotaVdsGroup);
assertEquals(true, quotaVdsGroupList.size() == 1);
// Check if the global variable returns when null is initialization.
assertEquals(quotaVdsGroup.getVirtualCpu(), new Integer(10));
}
/**
* Test scenario when there is a global limitation on the storage pool, and we check limitation on
* the entire storage pool.<br>
* The value that should be returned, is the global limitation and the global usage on the storage
* pool.
*
* @throws Exception
*/
@Test
public void testFetchGlobalQuotaUsageForGlobalVdsGroup() throws Exception {
List<QuotaVdsGroup> quotaVdsGroupList =
dao.getQuotaVdsGroupByVdsGroupGuid(null, FixturesTool.QUOTA_GENERAL);
QuotaVdsGroup quotaVdsGroup = quotaVdsGroupList.get(0);
assertEquals(true, quotaVdsGroupList.size() == 1);
assertNotNull(quotaVdsGroup);
assertEquals(true, quotaVdsGroup.getMemSizeMBUsage() > 0);
// Check if the global variable returns when null is initialization.
assertEquals(new Integer(100), quotaVdsGroup.getVirtualCpu());
}
/**
* Asserts that when {@link QuotaDAO#getQuotaStorageByQuotaGuidWithGeneralDefault(Guid)} is called
* with a specific quota, all the relevant VDSs are returned
*/
@Test
public void testQuotaStorageByQuotaGuidWithGeneralDefaultWithDefault() {
List<QuotaStorage> quotaStorageList =
dao.getQuotaStorageByQuotaGuidWithGeneralDefault(FixturesTool.DEFAULT_QUOTA_GENERAL);
assertNotNull(quotaStorageList);
assertEquals("wrong number of quotas returned", 1, quotaStorageList.size());
for (QuotaStorage group : quotaStorageList) {
assertEquals(
"Storage ID should not be null in general mode", Guid.Empty, group.getStorageId());
assertNull("Storage name should not be null in general mode", group.getStorageName());
}
}
@Test
public void testSpecificQuotaLimitations() throws Exception {
// Set new Quota definition.
Quota quota = new Quota();
Guid quotaId = Guid.NewGuid();
quota.setId(quotaId);
quota.setStoragePoolId(FixturesTool.STORAGE_POOL_NFS);
quota.setQuotaName("Watson");
quota.setDescription("Specific quota");
quota.setThresholdVdsGroupPercentage(
Config.<Integer>GetValue(ConfigValues.QuotaThresholdVdsGroup));
quota.setThresholdStoragePercentage(
Config.<Integer>GetValue(ConfigValues.QuotaThresholdStorage));
quota.setGraceVdsGroupPercentage(Config.<Integer>GetValue(ConfigValues.QuotaGraceVdsGroup));
quota.setGraceStoragePercentage(Config.<Integer>GetValue(ConfigValues.QuotaGraceStorage));
quota.setQuotaVdsGroups(getQuotaVdsGroup(getSpecificQuotaVdsGroup(quotaId)));
quota.setQuotaStorages(getQuotaStorage(getSpecificQuotaStorage(quotaId)));
dao.save(quota);
Quota quotaEntity = dao.getById(quota.getId());
assertNotNull(quotaEntity);
assertEquals(quotaEntity, quota);
}
/**
* Test scenario when there is a specific limitation on a vds group, and we check if there is
* enough resources on it.<br>
* The returned value from the query, should be the specific limitation and the global usage on
* the storage pool.
*
* @throws Exception
*/
@Test
public void testFetchVdsGroupWithUnlimitedGlobalLimitation() throws Exception {
List<QuotaVdsGroup> quotaVdsGroupList =
dao.getQuotaVdsGroupByVdsGroupGuid(
FixturesTool.VDS_GROUP_RHEL6_ISCSI, FixturesTool.QUOTA_SPECIFIC);
// Should be two rows of specific vds group
assertEquals(true, quotaVdsGroupList.size() == 1);
// Get first specific vds group.
QuotaVdsGroup quotaVdsGroup = quotaVdsGroupList.get(0);
assertNotNull(quotaVdsGroup);
assertEquals(quotaVdsGroup.getMemSizeMB(), unlimited);
assertEquals(true, quotaVdsGroup.getVirtualCpu().equals(10));
}
@Test
public void testFetchAllVdsGroupForQuota() throws Exception {
List<QuotaVdsGroup> quotaVdsGroupList =
dao.getQuotaVdsGroupByVdsGroupGuid(null, FixturesTool.QUOTA_SPECIFIC);
assertNotNull(quotaVdsGroupList);
assertEquals(quotaVdsGroupList.size(), 2);
for (QuotaVdsGroup quotaVdsGroup : quotaVdsGroupList) {
if (quotaVdsGroup
.getQuotaVdsGroupId()
.equals(Guid.createGuidFromString("68c96e11-0aad-4e3a-9091-12897b7f2388"))) {
assertEquals(quotaVdsGroup.getVirtualCpu(), new Integer("10"));
assertEquals(quotaVdsGroup.getMemSizeMB(), unlimited);
} else if (quotaVdsGroup
.getQuotaVdsGroupId()
.equals(Guid.createGuidFromString("68c96e11-0aad-4e3a-9091-12897b7f2389"))) {
assertEquals(quotaVdsGroup.getVirtualCpu(), new Integer("1000"));
assertEquals(quotaVdsGroup.getMemSizeMB(), unlimited);
}
}
}
/** Asserts that {@link #expectedQuotas} are relevant for the given {@link #vdsGroupId} */
private void assertGetAllRelevantQuoatsForVdsGroup(Guid vdsGroupId, int expectedQuotas) {
List<Quota> quotas = dao.getAllRelevantQuotasForVdsGroup(vdsGroupId, null, false);
assertEquals("Wrong number of quotas retuend", expectedQuotas, quotas.size());
}
/**
* Test get all Quotas in the setup
*
* @throws Exception
*/
@Test
public void testGetFetchAllQuotaInTheSetup() throws Exception {
List<Quota> quotaList = dao.getQuotaByStoragePoolGuid(null);
assertEquals(5, quotaList.size());
}
/**
* Test get Quota by Name, with not existing name.
*
* @throws Exception
*/
@Test
public void testGetQuotaWithNoExistingName() throws Exception {
Quota quotaGeneralToSpecific = dao.getQuotaByQuotaName("Any name");
assertEquals(null, quotaGeneralToSpecific);
}
/**
* Test get Quota by Name, with name that does not exist for the storage pool.
*
* @throws Exception
*/
@Test
public void testGetQuotaByExistingNameWIthNoMatchingStoragePool() throws Exception {
Quota quotaGeneralToSpecific = dao.getQuotaByQuotaName("Quota General2");
assertEquals(null, quotaGeneralToSpecific);
}
/**
* Test get Quota by storage pool Id, for storage pool with no quotas in it.
*
* @throws Exception
*/
@Test
public void testFetchStoragePoolWithNoQuota() throws Exception {
List<Quota> quotaList = dao.getQuotaByStoragePoolGuid(Guid.NewGuid());
assertEquals(true, quotaList.size() == 0);
}
/**
* Test get Quota by storage pool Id
*
* @throws Exception
*/
@Test
public void testGetFetchForSpecificStoragePool() throws Exception {
List<Quota> quotaList = dao.getQuotaByStoragePoolGuid(FixturesTool.STORAGE_POOL_NFS);
assertEquals(5, quotaList.size());
}