如何从HTTP请求返回Mock响应对象？Java单元测试相关问题

你需要的场景无法通过WireMock实现，WireMock的作用是模拟真实HTTP服务端的报文返回，客户端收到响应后会由HTTP客户端框架自行构造Response实例，你无法直接干预该实例的方法行为。以下是两种可行的实现方案：

方案一：最小重构解耦（推荐） #

通过极小的代码调整提升可测试性，无需引入特殊测试框架：

1. 将Response的处理逻辑从原有方法中提取为包级私有方法：

// 原有业务类示例调整后
public class CustomHttpClient {
    public <T> T doGet(String url, Class<T> type) {
        Request request = // 原有request构造逻辑
        Response response = request.get();
        return handleSuccessResponse(response, type);
    }

    // 包级私有，仅同包下的测试类可访问
    <T> T handleSuccessResponse(Response response, Class<T> type) {
        if (response.getStatusInfo().getFamily().equals(Response.Status.Family.SUCCESSFUL)) {
             return response.readEntity(type);
        }
        // 原有其他错误处理逻辑
        return null;
    }
}

2. 测试时使用Mockito的Spy功能部分mock被测类，构造抛出异常的场景：

import static org.mockito.Mockito.*;

@Test
void testReadEntityThrowException() {
    // 构造Spy对象，未指定mock的方法会调用真实逻辑
    CustomHttpClient client = spy(new CustomHttpClient());
    Response mockResp = mock(Response.class);
    
    // 定义mock行为
    when(mockResp.getStatusInfo()).thenReturn(Response.Status.OK);
    when(mockResp.readEntity(any(Class.class))).thenThrow(new RuntimeException("反序列化异常"));
    
    // 验证异常抛出符合预期
    assertThrows(RuntimeException.class, () -> client.handleSuccessResponse(mockResp, String.class));
}

方案二：使用PowerMock Mock局部变量（无需修改业务代码） #

如果业务代码不允许修改，可以用PowerMock拦截方法内部的request构造逻辑，直接返回自定义的Mock对象：

@RunWith(PowerMockRunner.class)
@PrepareForTest(CustomHttpClient.class)
public class CustomHttpClientTest {
    @Test
    void testReadEntityThrowException() {
        Request mockRequest = mock(Request.class);
        Response mockResp = mock(Response.class);

        // 定义mock行为
        when(mockResp.getStatusInfo()).thenReturn(Response.Status.OK);
        when(mockResp.readEntity(any(Class.class))).thenThrow(new RuntimeException("反序列化异常"));
        when(mockRequest.get()).thenReturn(mockResp);
        
        // 拦截方法内部request实例的创建逻辑，返回mock对象
        // 示例中如果request是通过WebTarget.request()创建，对应mock逻辑如下：
        WebTarget mockTarget = mock(WebTarget.class);
        when(mockTarget.request()).thenReturn(mockRequest);
        PowerMockito.whenNew(WebTarget.class).withAnyArguments().thenReturn(mockTarget);

        // 执行测试验证
        CustomHttpClient client = new CustomHttpClient();
        assertThrows(RuntimeException.class, () -> client.doGet("testUrl", String.class));
    }
}

方案选择建议：优先选择方案一，PowerMock对高版本JDK和高版本Mockito兼容性较差，且会增加单元测试的执行耗时，重构解耦的方式更符合可测试性代码设计规范。

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