apache/cloudstack
1
0
Fork 0
mirror of https://github.com/apache/cloudstack.git synced 2025-10-26 08:42:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

QuotaStatementTest: fix the method invocation verification

Browse Source
This commit is contained in:
Abhinandan Prateek 2016-01-04 16:28:42 +05:30 committed by Rohit Yadav
parent 9be76b350f
commit 0eac900fc4
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
framework/quota/test/org/apache/cloudstack/quota/QuotaStatementTest.java
View File

@ -22,7 +22,6 @@ import com.cloud.utils.db.TransactionLegacy;
import junit.framework.TestCase; import junit.framework.TestCase;
import org.apache.cloudstack.framework.config.dao.ConfigurationDao; import org.apache.cloudstack.framework.config.dao.ConfigurationDao;
import org.apache.cloudstack.quota.QuotaStatementImpl.STATEMENT_PERIODS; import org.apache.cloudstack.quota.QuotaStatementImpl.STATEMENT_PERIODS;
import org.apache.cloudstack.quota.constant.QuotaConfig;
import org.apache.cloudstack.quota.dao.QuotaAccountDao; import org.apache.cloudstack.quota.dao.QuotaAccountDao;
import org.apache.cloudstack.quota.dao.QuotaUsageDao; import org.apache.cloudstack.quota.dao.QuotaUsageDao;
import org.apache.cloudstack.quota.vo.QuotaAccountVO; import org.apache.cloudstack.quota.vo.QuotaAccountVO;
@ -224,6 +223,7 @@ public class QuotaStatementTest extends TestCase {
} }
@Test @Test
public void testSendStatement() throws UnsupportedEncodingException, MessagingException { public void testSendStatement() throws UnsupportedEncodingException, MessagingException {
Calendar date = Calendar.getInstance(); Calendar date = Calendar.getInstance();
@ -242,13 +242,11 @@ public class QuotaStatementTest extends TestCase {
Mockito.when(quotaUsage.findTotalQuotaUsage(Mockito.anyLong(), Mockito.anyLong(), Mockito.anyInt(), Mockito.any(Date.class), Mockito.any(Date.class))) Mockito.when(quotaUsage.findTotalQuotaUsage(Mockito.anyLong(), Mockito.anyLong(), Mockito.anyInt(), Mockito.any(Date.class), Mockito.any(Date.class)))
.thenReturn(new BigDecimal(100)); .thenReturn(new BigDecimal(100));
QuotaAlertManagerImpl.DeferredQuotaEmail email = new QuotaAlertManagerImpl.DeferredQuotaEmail(accountVO, acc, new BigDecimal(100),
QuotaConfig.QuotaEmailTemplateTypes.QUOTA_LOW);
// call real method on send monthly statement // call real method on send monthly statement
Mockito.doCallRealMethod().when(quotaStatement).sendStatement(); quotaStatement.sendStatement();
Calendar period[] = quotaStatement.statementTime(date, STATEMENT_PERIODS.MONTHLY); Calendar period[] = quotaStatement.statementTime(date, STATEMENT_PERIODS.MONTHLY);
if (period != null){ if (period != null){
Mockito.verify(alertManager, Mockito.times(1)).sendQuotaAlert(email); Mockito.verify(alertManager, Mockito.times(1)).sendQuotaAlert(Mockito.any(QuotaAlertManagerImpl.DeferredQuotaEmail.class));
} }
} }