Spring Integration单元测试中模拟出站网关失败的问题

我来帮你搞定这个单元测试里模拟HTTP出站网关失败的问题——你之前的尝试方向是对的，但有些细节没处理到位，导致异常没按预期触发流程终止或重试。

问题根源 #

你第一次用handleNextAndReply返回401的ResponseEntity，但Spring Integration的HTTP出站网关默认不会把非2xx响应转换成异常，只会把状态码存入消息头，所以后续流程会继续执行。第二次改成抛出HttpClientErrorException是正确的思路，但可能还有配置细节需要调整。

正确配置步骤 #

1. 确保重试建议能捕获目标异常 #

首先检查你的retryAdvice()，它需要明确配置捕获HttpClientErrorException（或其父类RestClientException），否则即使抛出了异常，重试逻辑也不会触发，异常可能被框架吞掉后继续流程。示例配置：

public Advice retryAdvice() {
    RetryTemplate retryTemplate = new RetryTemplate();
    SimpleRetryPolicy retryPolicy = new SimpleRetryPolicy();
    retryPolicy.setMaxAttempts(3);
    // 配置只重试HttpClientErrorException类型的异常
    Map<Class<? extends Throwable>, Boolean> retryableExceptions = new HashMap<>();
    retryableExceptions.put(HttpClientErrorException.class, true);
    retryPolicy.setRetryableExceptions(retryableExceptions);
    
    retryTemplate.setRetryPolicy(retryPolicy);
    return RetryInterceptorBuilder.stateless()
            .retryOperations(retryTemplate)
            .build();
}

2. 正确使用Mock消息处理器抛出异常 #

你第二次的Mock写法是对的，但可以再明确异常信息，同时确保Mock被正确替换到上下文：

MessageHandler mockCognitiveAnalyze = mockMessageHandler()
    .handleNext(message -> {
        throw new HttpClientErrorException(HttpStatus.UNAUTHORIZED, "API Key无效");
    });
this.mockIntegrationContext.substituteMessageHandlerFor("cognitiveServicesReadEndpoint", mockCognitiveAnalyze);

或者用handleNextAndReply的方式抛出异常，效果是一样的：

MessageHandler mockCognitiveAnalyze = mockMessageHandler()
    .handleNextAndReply(message -> {
        throw new HttpClientErrorException(HttpStatus.UNAUTHORIZED);
    });

3. 验证异常传播与重试行为 #

在测试中添加断言，验证异常是否被正确抛出或重试是否触发：

// 验证直接抛出异常
assertThrows(HttpClientErrorException.class, () -> {
    Message<?> testMessage = MessageBuilder.withPayload("test")
            .setHeader("Ocp-Apim-Subscription-Key", "invalid-key")
            .build();
    this.yourInputChannel.send(testMessage);
});

// 如果要验证重试次数，可以用计数器或Mock来跟踪调用次数
AtomicInteger callCount = new AtomicInteger(0);
MessageHandler mockCognitiveAnalyze = mockMessageHandler()
    .handleNext(message -> {
        callCount.incrementAndGet();
        throw new HttpClientErrorException(HttpStatus.UNAUTHORIZED);
    });
this.mockIntegrationContext.substituteMessageHandlerFor("cognitiveServicesReadEndpoint", mockCognitiveAnalyze);

// 发送消息后断言调用次数等于重试次数+1
yourInputChannel.send(testMessage);
assertEquals(3, callCount.get()); // 假设重试次数是2次，总共调用3次

4. 检查Wire Tap的影响 #

注意到你的cognitiveServicesReadEndpoint是在Wire Tap里配置的，默认Wire Tap是同步执行的，但如果你的Wire Tap使用了异步通道，异常可能不会传播到主流程。如果是异步场景，需要给Wire Tap的通道配置错误处理器，或者在测试中捕获异步线程的异常。

额外注意事项 #

• 确保测试类上添加了@SpringIntegrationTest注解，这样mockIntegrationContext才能正确初始化并替换消息处理器。
• 如果使用Spring Boot测试，配合@SpringBootTest一起使用，保证上下文正确加载。

这样配置后，当cognitiveServicesReadEndpoint被调用时，Mock处理器会抛出HttpClientErrorException，触发重试建议，重试耗尽后异常会传播出去，终止后续的transform和第二个HTTP网关调用。

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