Skip to content

Commit 0d9292c

Browse files
authored
Remove powermock from database/quota plugin (#7795)
1 parent d51d8a4 commit 0d9292c

2 files changed

Lines changed: 10 additions & 12 deletions

File tree

plugins/database/quota/src/test/java/org/apache/cloudstack/api/response/QuotaResponseBuilderImplTest.java

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -52,10 +52,8 @@
5252
import org.junit.runner.RunWith;
5353
import org.mockito.InjectMocks;
5454
import org.mockito.Mock;
55+
import org.mockito.MockedConstruction;
5556
import org.mockito.Mockito;
56-
import org.powermock.api.mockito.PowerMockito;
57-
import org.powermock.core.classloader.annotations.PrepareForTest;
58-
import org.powermock.modules.junit4.PowerMockRunner;
5957

6058
import com.cloud.exception.InvalidParameterValueException;
6159
import com.cloud.user.Account;
@@ -64,8 +62,9 @@
6462
import com.cloud.user.dao.UserDao;
6563

6664
import junit.framework.TestCase;
65+
import org.mockito.junit.MockitoJUnitRunner;
6766

68-
@RunWith(PowerMockRunner.class)
67+
@RunWith(MockitoJUnitRunner.class)
6968
public class QuotaResponseBuilderImplTest extends TestCase {
7069

7170
@Mock
@@ -322,12 +321,13 @@ public void validateEndDateOnCreatingNewQuotaTariffTestSetValidEndDate() {
322321
}
323322

324323
@Test
325-
@PrepareForTest(QuotaResponseBuilderImpl.class)
326324
public void getNewQuotaTariffObjectTestCreateFromCurrentQuotaTariff() throws Exception {
327-
PowerMockito.whenNew(QuotaTariffVO.class).withArguments(Mockito.any(QuotaTariffVO.class)).thenReturn(quotaTariffVoMock);
328-
329-
quotaResponseBuilderSpy.getNewQuotaTariffObject(quotaTariffVoMock, "", 0);
330-
PowerMockito.verifyNew(QuotaTariffVO.class).withArguments(Mockito.any(QuotaTariffVO.class));
325+
try (MockedConstruction<QuotaTariffVO> quotaTariffVOMockedConstruction = Mockito.mockConstruction(QuotaTariffVO.class, (mock,
326+
context) -> {
327+
})) {
328+
QuotaTariffVO result = quotaResponseBuilderSpy.getNewQuotaTariffObject(quotaTariffVoMock, "", 0);
329+
Assert.assertEquals(quotaTariffVOMockedConstruction.constructed().get(0), result);
330+
}
331331
}
332332

333333
@Test (expected = InvalidParameterValueException.class)
@@ -360,7 +360,6 @@ public void persistNewQuotaTariffTestpersistNewQuotaTariff() {
360360

361361
@Test (expected = ServerApiException.class)
362362
public void deleteQuotaTariffTestQuotaDoesNotExistThrowsServerApiException() {
363-
Mockito.doReturn(null).when(quotaTariffDaoMock).findById(Mockito.anyLong());
364363
quotaResponseBuilderSpy.deleteQuotaTariff("");
365364
}
366365

@@ -380,7 +379,6 @@ public void getQuotaSummaryResponseTestAccountIsNotNullQuotaIsDisabledShouldRetu
380379
Calendar[] period = createPeriodForQuotaSummary();
381380
overrideDefaultQuotaEnabledConfigValue("false");
382381

383-
Mockito.doReturn(accountMock).when(accountDaoMock).findActiveAccount(Mockito.anyString(), Mockito.anyLong());
384382
Mockito.doReturn(period).when(quotaStatementMock).getCurrentStatementTime();
385383
Mockito.doReturn(domainVOMock).when(domainDaoMock).findById(Mockito.anyLong());
386384
Mockito.doReturn(BigDecimal.ZERO).when(quotaBalanceDaoMock).lastQuotaBalance(Mockito.anyLong(), Mockito.anyLong(), Mockito.any(Date.class));
@@ -396,7 +394,6 @@ public void getQuotaSummaryResponseTestAccountIsNotNullQuotaIsEnabledShouldRetur
396394
Calendar[] period = createPeriodForQuotaSummary();
397395
overrideDefaultQuotaEnabledConfigValue("true");
398396

399-
Mockito.doReturn(accountMock).when(accountDaoMock).findActiveAccount(Mockito.anyString(), Mockito.anyLong());
400397
Mockito.doReturn(period).when(quotaStatementMock).getCurrentStatementTime();
401398
Mockito.doReturn(domainVOMock).when(domainDaoMock).findById(Mockito.anyLong());
402399
Mockito.doReturn(BigDecimal.ZERO).when(quotaBalanceDaoMock).lastQuotaBalance(Mockito.anyLong(), Mockito.anyLong(), Mockito.any(Date.class));
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
mock-maker-inline

0 commit comments

Comments
 (0)