From 0eac900fc4953fb36c6b37e7c0ee52184443efbd Mon Sep 17 00:00:00 2001 From: Abhinandan Prateek Date: Mon, 4 Jan 2016 16:28:42 +0530 Subject: [PATCH] QuotaStatementTest: fix the method invocation verification --- .../org/apache/cloudstack/quota/QuotaStatementTest.java | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/framework/quota/test/org/apache/cloudstack/quota/QuotaStatementTest.java b/framework/quota/test/org/apache/cloudstack/quota/QuotaStatementTest.java index f2a0deda3c7..d1cc7ace29d 100644 --- a/framework/quota/test/org/apache/cloudstack/quota/QuotaStatementTest.java +++ b/framework/quota/test/org/apache/cloudstack/quota/QuotaStatementTest.java @@ -22,7 +22,6 @@ import com.cloud.utils.db.TransactionLegacy; import junit.framework.TestCase; import org.apache.cloudstack.framework.config.dao.ConfigurationDao; 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.QuotaUsageDao; import org.apache.cloudstack.quota.vo.QuotaAccountVO; @@ -224,6 +223,7 @@ public class QuotaStatementTest extends TestCase { } + @Test public void testSendStatement() throws UnsupportedEncodingException, MessagingException { 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))) .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 - Mockito.doCallRealMethod().when(quotaStatement).sendStatement(); + quotaStatement.sendStatement(); Calendar period[] = quotaStatement.statementTime(date, STATEMENT_PERIODS.MONTHLY); if (period != null){ - Mockito.verify(alertManager, Mockito.times(1)).sendQuotaAlert(email); + Mockito.verify(alertManager, Mockito.times(1)).sendQuotaAlert(Mockito.any(QuotaAlertManagerImpl.DeferredQuotaEmail.class)); } }