如何模拟 HttpClient 类以返回不同的响应

Question

如何让 Mockito 返回不同的 HttpEntity 值以响应不同的 URI？

该测试将使用 mockHttpClient 发出多个 HTTP 请求(所有请求都是 POST 请求(。

HttpEntity httpEntity = EntityBuilder.create().setText(response).build();
PowerMockito.when(response, "getEntity").thenReturn(httpEntity);

测试本身是这样设计的：

CloseableHttpClient client = mockHttpClient(200, HTTP_ENTITY);
runTest();
Mockito.verify(client, Mockito.times(2)).execute(Mockito.any());

为了使上述测试返回不同的HTTP实体，我尝试了以下方法：

CloseableHttpClient client = Mockito.mock(CloseableHttpClient.class);
    Mockito.when(client.execute(new HttpPost("http://127.0.0.1:8000/new/a"))).thenReturn(resp1);
    Mockito.when(client.execute(new HttpPost("http://127.0.0.1:8000/new/a/b"))).thenReturn(resp2);

但是我无法构建不同的 http 实体并根据请求中的 URI 获得响应。

Answer 1

首先，在这种情况下不需要Powermock。你不需要嘲笑静态方法、构造函数或其他任何令人讨厌的东西。

我准备了一个小示例测试，它在这里开箱即用 Mockito 2.23.4 ，您可以看到断言isSameAs ("验证实际值是否与给定的值相同，即使用 == 比较POST。

public class CloseableHttpClientTest {
    private final CloseableHttpClient client = mock(CloseableHttpClient.class);
    @Test
    public void mockClient() throws IOException {
        // given
        CloseableHttpResponse resp1 = mock(CloseableHttpResponse.class);
        CloseableHttpResponse resp2 = mock(CloseableHttpResponse.class);
        HttpPost post1 = new HttpPost("http://127.0.0.1:8000/new/a");
        HttpPost post2 = new HttpPost("http://127.0.0.1:8000/new/a/b");
        when(client.execute(post1)).thenReturn(resp1);
        when(client.execute(post2)).thenReturn(resp2);
        // when
        CloseableHttpResponse returnedResp1 = client.execute(post1);
        CloseableHttpResponse returnedResp2 = client.execute(post2);
        // then
        assertThat(returnedResp1).isSameAs(resp1);
        assertThat(returnedResp2).isSameAs(resp2);
        verify(client).execute(post1);
        verify(client).execute(post2);
    }
}

Answer 2

谢谢你

。您的回答帮助我了解如何更好地组织测试。但是在这种情况下，我正在尝试在没有任何重组的情况下修改一个。

我想我会在这里发布我的答案，以防其他人正在寻找一个。

Mockito.doAnswer(new Answer<CloseableHttpResponse>() {
  @Override
  public CloseableHttpResponse answer(InvocationOnMock invocation) {
    CloseableHttpResponse httpResponse = Mockito.mock(CloseableHttpResponse.class);
    // No change to status code based on endpoints
    Mockito.when(status.getStatusCode()).thenReturn(withResponseCode);
    Mockito.when(httpResponse.getStatusLine()).thenReturn(status);
    HttpEntity entity = Mockito.mock(HttpEntity.class);
    Object[] args = invocation.getArguments();
    String endpoint = ((HttpPost) args[0]).getURI().getPath();
    if (endpoint.contains("/a/b")) {
      entity = EntityBuilder.create().setText("something").build();
      Mockito.when(httpResponse.getEntity()).thenReturn(entity);
    } else {
      entity = EntityBuilder.create().setText("something else").build();
      Mockito.when(httpResponse.getEntity()).thenReturn(entity);
    }
    return httpResponse;
  }
}).when(client).execute(Mockito.any(HttpUriRequest.class)); 

集成测试不会测试其余客户端本身，我也不需要旋钮和多个模拟客户端来执行不同的响应。以上内容可帮助我根据接收请求的端点返回不同的响应。