如何编写单元测试验证RestTemplate的配置参数？

当然可以编写单元测试来验证RestTemplate的超时时间、连接池参数等配置，下面是针对你提供的WebConfiguration类的具体实现：

测试思路 #

通过Spring测试框架加载配置类，获取实例化后的RestTemplate，逐层拆解其内部组件，直接断言配置的参数值是否符合预期：

• 从RestTemplate提取HttpComponentsClientHttpRequestFactory
• 从工厂类获取CloseableHttpClient实例
• 从HttpClient中拿到连接池管理器PoolingHttpClientConnectionManager，验证最大总连接数、单路由最大连接数
• 从HttpClient中获取默认RequestConfig，验证连接超时、请求超时、Socket超时时间

单元测试代码 #

import org.apache.http.client.config.RequestConfig;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
import org.springframework.web.client.RestTemplate;

import static org.junit.jupiter.api.Assertions.assertEquals;

// 指定加载目标配置类，避免加载整个应用上下文
@SpringBootTest(classes = WebConfiguration.class)
class WebConfigurationTest {

    @Autowired
    private RestTemplate restTemplate;

    @Test
    void restTemplate_ShouldHaveCorrectConnectionPoolConfig() {
        // 提取请求工厂
        HttpComponentsClientHttpRequestFactory requestFactory = 
            (HttpComponentsClientHttpRequestFactory) restTemplate.getRequestFactory();
        CloseableHttpClient httpClient = (CloseableHttpClient) requestFactory.getHttpClient();

        // 验证连接池参数
        PoolingHttpClientConnectionManager connectionManager = 
            (PoolingHttpClientConnectionManager) httpClient.getConnectionManager();
        assertEquals(200, connectionManager.getMaxTotal());
        assertEquals(100, connectionManager.getDefaultMaxPerRoute());
    }

    @Test
    void restTemplate_ShouldHaveCorrectTimeoutConfig() {
        HttpComponentsClientHttpRequestFactory requestFactory = 
            (HttpComponentsClientHttpRequestFactory) restTemplate.getRequestFactory();
        CloseableHttpClient httpClient = (CloseableHttpClient) requestFactory.getHttpClient();

        // 验证超时参数
        RequestConfig requestConfig = httpClient.getDefaultRequestConfig();
        assertEquals(10000, requestConfig.getConnectTimeout());
        assertEquals(10000, requestConfig.getConnectionRequestTimeout());
        assertEquals(10000, requestConfig.getSocketTimeout());
    }
}

注意事项 #

• 使用@SpringBootTest(classes = WebConfiguration.class)仅加载目标配置类，加快测试执行速度
• 类型转换时确保配置类中确实使用了HttpComponentsClientHttpRequestFactory，避免类型转换异常
• 所有断言的数值要和配置类中设置的保持一致

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