5252import org .junit .runner .RunWith ;
5353import org .mockito .InjectMocks ;
5454import org .mockito .Mock ;
55+ import org .mockito .MockedConstruction ;
5556import 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
6058import com .cloud .exception .InvalidParameterValueException ;
6159import com .cloud .user .Account ;
6462import com .cloud .user .dao .UserDao ;
6563
6664import junit .framework .TestCase ;
65+ import org .mockito .junit .MockitoJUnitRunner ;
6766
68- @ RunWith (PowerMockRunner .class )
67+ @ RunWith (MockitoJUnitRunner .class )
6968public 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 ));
0 commit comments