2626import static org .hamcrest .Matchers .notNullValue ;
2727import static org .hamcrest .Matchers .sameInstance ;
2828import static org .junit .Assert .fail ;
29- import static org .mockito .Matchers .eq ;
29+ import static org .mockito .ArgumentMatchers .eq ;
3030import static org .mockito .Mockito .mock ;
31+ import static org .mockito .Mockito .spy ;
32+ import static org .mockito .Mockito .verify ;
3133import static org .mockito .Mockito .when ;
32- import static org .powermock .api .mockito .PowerMockito .spy ;
33- import static org .powermock .api .mockito .PowerMockito .verifyPrivate ;
3434
3535import java .util .HashMap ;
3636import java .util .Map ;
4747import org .junit .BeforeClass ;
4848import org .junit .Test ;
4949import org .junit .runner .RunWith ;
50- import org .powermock .core .classloader .annotations .PowerMockIgnore ;
51- import org .powermock .core .classloader .annotations .PrepareForTest ;
52- import org .powermock .modules .junit4 .PowerMockRunner ;
50+ import org .mockito .junit .MockitoJUnitRunner ;
5351
5452import com .cloud .utils .rest .CloudstackRESTException ;
5553import com .cloud .utils .rest .HttpMethods ;
5654import com .cloud .utils .rest .HttpRequestMatcher ;
5755import com .cloud .utils .rest .HttpUriRequestBuilder ;
5856
59- @ RunWith (PowerMockRunner .class )
60- @ PrepareForTest (NiciraRestClient .class )
61- @ PowerMockIgnore ({"javax.xml.*" , "org.w3c.dom.*" , "org.apache.xerces.*" , "org.apache.log4j.*" })
57+ @ RunWith (MockitoJUnitRunner .class )
6258public class NiciraRestClientTest {
6359
6460 private static final int HTTPS_PORT = 443 ;
@@ -120,7 +116,7 @@ public void testExecuteSuccessAtFirstAttempt() throws Exception {
120116
121117 assertThat (response , notNullValue ());
122118 assertThat (response , sameInstance (mockResponse ));
123- verifyPrivate (client ).invoke ( " execute" , request , 0 );
119+ verify (client ).execute ( request , 0 );
124120 }
125121
126122 @ Test
@@ -139,9 +135,9 @@ public void testExecuteUnauthorizedThenSuccess() throws Exception {
139135
140136 assertThat (response , notNullValue ());
141137 assertThat (response , sameInstance (mockResponse ));
142- verifyPrivate (client ).invoke ( " execute" , HttpRequestMatcher .eq (request ), eq (0 ));
143- verifyPrivate (client ).invoke ( " execute" , HttpRequestMatcher .eq (loginRequest ), eq (401 ));
144- verifyPrivate (client ).invoke ( " execute" , HttpRequestMatcher .eq (request ), eq (200 ));
138+ verify (client ).execute (( HttpUriRequest ) HttpRequestMatcher .eq (request ), eq (0 ));
139+ verify (client ).execute (( HttpUriRequest ) HttpRequestMatcher .eq (request ), eq (200 ));
140+ verify (client ).execute (( HttpUriRequest ) HttpRequestMatcher .eq (loginRequest ), eq (401 ));
145141 }
146142
147143 @ Test
@@ -168,8 +164,8 @@ public void testExecuteTwoConsecutiveUnauthorizedExecutions() throws Exception {
168164 fail ("Expected CloudstackRESTException exception" );
169165 } catch (final CloudstackRESTException e ) {
170166 assertThat (e .getMessage (), not (isEmptyOrNullString ()));
171- verifyPrivate (client ).invoke ( " execute" , HttpRequestMatcher .eq (request ), eq (0 ));
172- verifyPrivate (client ).invoke ( " execute" , HttpRequestMatcher .eq (loginRequest ), eq (401 ));
167+ verify (client ).execute (( HttpUriRequest ) HttpRequestMatcher .eq (request ), eq (0 ));
168+ verify (client ).execute (( HttpUriRequest ) HttpRequestMatcher .eq (loginRequest ), eq (401 ));
173169 }
174170 }
175171
0 commit comments