Examples with UsageVO com.cloud.usage.UsageVO used on opensource projects

Example 31 with UsageVO

use of com.cloud.usage.UsageVO in project cloudstack by apache.

the class QuotaManagerImplTest method testAggregatePendingQuotaRecordsForAccount.

@Test
public void testAggregatePendingQuotaRecordsForAccount() {
    AccountVO accountVO = new AccountVO();
    accountVO.setId(2L);
    accountVO.setDomainId(1L);
    accountVO.setType(Account.ACCOUNT_TYPE_NORMAL);
    UsageVO usageVO = new UsageVO();
    usageVO.setQuotaCalculated(0);
    usageVO.setUsageType(UsageTypes.ALLOCATED_VM);
    List<UsageVO> usageVOList = new ArrayList<UsageVO>();
    usageVOList.add(usageVO);
    Pair<List<? extends UsageVO>, Integer> usageRecords = new Pair<List<? extends UsageVO>, Integer>(usageVOList, usageVOList.size());
    QuotaUsageVO quotaUsageVO = new QuotaUsageVO();
    quotaUsageVO.setAccountId(2L);
    Mockito.doReturn(quotaUsageVO).when(quotaManager).updateQuotaAllocatedVMUsage(Mockito.eq(usageVO));
    assertTrue(quotaManager.aggregatePendingQuotaRecordsForAccount(accountVO, new Pair<List<? extends UsageVO>, Integer>(null, 0)).size() == 0);
    assertTrue(quotaManager.aggregatePendingQuotaRecordsForAccount(accountVO, usageRecords).size() == 1);
}