PowerMockito.doReturn返回null

Question

這是我的課程正在測試：

public class A {

public Integer callMethod(){
  return someMethod();
}


private Integer someMethod(){
  //Some Code
  HttpPost httpPost = new HttpPost(oAuthMessage.URL);
  //Some Code
  HttpClient httpClient = new DefaultHttpClient();
  HttpResponse httpResponse = httpClient.execute(httpPost); ------1
  Integer code = httpResponse.getStatusLine().getStatusCode(); ---2
  return code;
}

現在我要模擬第1和2行，並返回模擬HttpResponse和代碼。

我已經嘗試過但是失敗了：

@RunWith(PowerMockRunner.class)
@PowerMockIgnore("javax.crypto.*")
public class TestA {

//Spying some things here & Injecting them

@Test
public void testA() {


   DefaultHttpClient defaultHttpClientMock = PowerMockito.mock(DefaultHttpClient.class);
   HttpResponse httpResponse = PowerMockito.mock(HttpResponse.class, RETURNS_DEEP_STUBS);
   HttpClient httpClient = PowerMockito.mock(HttpClient.class);
   //HttpResponse httpResponseMock    PowerMockito.mock(HttpResponse.class);
   HttpPost httpPost = PowerMockito.mock(HttpPost.class);
   PowerMockito.whenNew(DefaultHttpClient.class).withNoArguments().thenReturn(defaultHttpClientMock);
   PowerMockito.doReturn(httpResponse).when(httpClient).execute(httpPost);      //Returns null. It never returns httpResponse.
   PowerMockito.when(httpResponse.getStatusLine().getStatusCode()).thenReturn(anyInt());
   //call the method
}

PowerMockito.doReturn(httpResponse).when(httpClient).execute(httpPost)始終返回null。 我希望它返回HttpResponse的模擬對象。 我已經閱讀了與此錯誤相關的其他帖子，但不確定該如何處理。 有人可以幫忙嗎？

Answer 1

代替

PowerMockito.doReturn(httpResponse).when(httpClient).execute(httpPost);

你應該使用

PowerMockito.when(httpResponse.execute(httpPost)).thenReturn(httpResponse);

您的測試中還存在一些問題：模擬構造器不正確，並且根本不需要httpResponse。

更新此代碼對我而言正常工作：

@RunWith(PowerMockRunner.class)
@PowerMockIgnore("javax.crypto.*")
@PrepareForTest({ HttpPost.class, DefaultHttpClient.class, A.class })
public class TestA {

    @Test
    public void testA() throws Exception {
        HttpPost httpPost = Mockito.mock(HttpPost.class);
        PowerMockito.whenNew(HttpPost.class).withArguments(oAuthMessage.URL).thenReturn(httpPost);

        DefaultHttpClient defaultHttpClientMock = PowerMockito.mock(DefaultHttpClient.class);
        HttpResponse httpResponse = PowerMockito.mock(HttpResponse.class);
        PowerMockito.whenNew(DefaultHttpClient.class).withNoArguments().thenReturn(defaultHttpClientMock);

        PowerMockito.when(defaultHttpClientMock.execute(httpPost)).thenReturn(httpResponse);

        StatusLine statusLine = PowerMockito.mock(StatusLine.class);

        PowerMockito.when(httpResponse.getStatusLine()).thenReturn(statusLine);
        Integer expected = new Integer(0);
        PowerMockito.when(statusLine.getStatusCode()).thenReturn(expected);

        A a = new A();
        Assert.assertEquals(expected, a.callMethod());
    }
}