如何使用Mockito模拟HttpClient.send()方法

你的测试不生效主要有两个原因：

1. HttpClient.newHttpClient()是静态方法，Mockito默认无法模拟静态方法，所以你写的when(HttpClient.newHttpClient())不会起作用；
2. 测试中mock的HttpResponse<Object>和原方法返回的HttpResponse<String>类型不匹配，会导致类型转换问题。

下面提供两种可行的解决方案，优先推荐第一种：

方案一：重构代码，依赖注入HttpClient（推荐） #

把HttpClient的创建从方法内部抽离，通过构造方法或参数注入到类中，这样测试时可以直接传入mock实例，完全不需要处理静态方法。

修改原业务类代码 #

public class MyClass {
    private final HttpClient httpClient;

    // 构造方法注入HttpClient，生产环境使用时传入HttpClient.newHttpClient()即可
    public MyClass(HttpClient httpClient) {
        this.httpClient = httpClient;
    }

    public String myMethod(String url) throws IOException, InterruptedException, HttpTimeoutException {
        HttpRequest request = HttpRequest.newBuilder()
                .uri(URI.create(url))
                .header("accept", "application/json")
                .method("GET", HttpRequest.BodyPublishers.noBody())
                .timeout(Duration.ofSeconds(3))
                .build();
        // 使用注入的HttpClient，而非静态方法创建
        HttpResponse<String> response = httpClient.send(request, HttpResponse.BodyHandlers.ofString());

        return response.body();
    }
}

对应的测试代码 #

@Test
void checkGetEarningRulesByTypeReturnGoodValues() throws IOException, InterruptedException {
    String expectedRes = "...";
    String testUrl = "https://example.com/test";

    // Mock所需对象
    HttpClient mockHttpClient = mock(HttpClient.class);
    HttpResponse<String> mockResponse = mock(HttpResponse.class);

    // 设置mock行为
    when(mockResponse.body()).thenReturn(expectedRes);
    when(mockHttpClient.send(any(HttpRequest.class), any(HttpResponse.BodyHandler.class)))
            .thenReturn(mockResponse);

    // 传入mock的HttpClient创建业务类实例
    MyClass myClass = new MyClass(mockHttpClient);
    String actualRes = myClass.myMethod(testUrl);

    // 断言结果
    assertEquals(expectedRes, actualRes);
}

这种方式不仅解决了测试问题，还让代码的可扩展性、可维护性更强——后续如果需要更换HttpClient的配置（比如设置代理、自定义超时），只需要传入不同的实例即可，无需修改业务逻辑。

方案二：使用PowerMock模拟静态方法（不推荐） #

如果暂时无法重构代码，可以用PowerMock来模拟静态方法，但会增加测试的复杂度和维护成本。

第一步：添加PowerMock依赖 #

以Maven为例，在pom.xml中添加：

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-module-junit4</artifactId>
    <version>2.0.9</version>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-api-mockito2</artifactId>
    <version>2.0.9</version>
    <scope>test</scope>
</dependency>

第二步：编写测试代码 #

@RunWith(PowerMockRunner.class)
@PrepareForTest(HttpClient.class) // 告知PowerMock需要处理HttpClient的静态方法
public class MyClassTest {

    @Test
    void checkGetEarningRulesByTypeReturnGoodValues() throws IOException, InterruptedException {
        String expectedRes = "...";
        String testUrl = "https://example.com/test";

        HttpClient mockHttpClient = mock(HttpClient.class);
        HttpResponse<String> mockResponse = mock(HttpResponse.class);

        when(mockResponse.body()).thenReturn(expectedRes);
        when(mockHttpClient.send(any(HttpRequest.class), any(HttpResponse.BodyHandler.class)))
                .thenReturn(mockResponse);

        // 模拟静态方法返回mock实例
        PowerMockito.mockStatic(HttpClient.class);
        when(HttpClient.newHttpClient()).thenReturn(mockHttpClient);

        String actualRes = MyClass.myMethod(testUrl);

        assertEquals(expectedRes, actualRes);
    }
}

需要注意的是，PowerMock会修改类加载机制，可能和某些测试框架或工具兼容出问题，所以除非万不得已，优先选择方案一。

内容的提问来源于stack exchange，提问作者shr7