如何让JUnit支持任意Lambda表达式？SpringAMQP生产者测试求助

嘿，针对你这段Spring AMQP生产者代码里的Lambda测试需求，我来给你梳理几个实用的方案：

1. 先确保用JUnit 5（Jupiter） #

JUnit 4对Lambda的支持比较有限，而JUnit 5是原生支持Lambda表达式的，所以第一步要确认你的测试依赖是JUnit 5：
如果用Maven，在pom.xml里添加：

<dependency>
    <groupId>org.junit.jupiter</groupId>
    <artifactId>junit-jupiter-api</artifactId>
    <version>5.9.2</version>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.junit.jupiter</groupId>
    <artifactId>junit-jupiter-engine</artifactId>
    <version>5.9.2</version>
    <scope>test</scope>
</dependency>

2. 用Mockito验证Lambda里的业务逻辑 #

你代码里的Lambda是MessagePostProcessor接口的实现，核心是给消息头添加x-death-reason，要测试这个逻辑，得配合Mockito来捕获并验证Lambda的行为：

步骤拆解： #

• 首先引入Mockito的JUnit 5依赖：

<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-core</artifactId>
    <version>4.11.0</version>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-junit-jupiter</artifactId>
    <version>4.11.0</version>
    <scope>test</scope>
</dependency>

• 编写测试类：

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.ArgumentCaptor;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessagePostProcessor;
import org.springframework.amqp.core.MessageProperties;
import org.springframework.amqp.rabbit.core.AmqpTemplate;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.verify;

@ExtendWith(MockitoExtension.class)
public class YourProducerTest {

    @Mock
    private AmqpTemplate amqpTemplate;

    @InjectMocks
    private YourProducerClass producer; // 替换成你的生产者类名

    @Test
    void testSendMethodAddsDeathReasonHeader() {
        // 准备测试数据
        Message testMessage = new Message("test payload".getBytes());
        Throwable testError = new RuntimeException("Connection timed out");
        String testQueue = "test-queue";
        String testRoutingKey = "test-key";

        // 调用目标方法
        producer.send(testMessage, testError, testQueue, testRoutingKey);

        // 捕获MessagePostProcessor类型的Lambda参数
        ArgumentCaptor<MessagePostProcessor> captor = ArgumentCaptor.forClass(MessagePostProcessor.class);

        // 验证convertAndSend方法被调用，并捕获Lambda参数
        verify(amqpTemplate).convertAndSend(
                eq(RabbitConfiguration.ERROR_EXCHANGE),
                eq(RabbitConfiguration.ERROR_ROUTING_KEY),
                any(Message.class),
                captor.capture()
        );

        // 获取捕获的Lambda并执行，检查消息头
        MessagePostProcessor processor = captor.getValue();
        Message processedMessage = processor.postProcessMessage(testMessage);
        MessageProperties properties = processedMessage.getMessageProperties();

        String deathReason = (String) properties.getHeaders().get("x-death-reason");
        assertEquals(testError.getMessage(), deathReason);
    }
}

3. 用JUnit 5的动态测试简化多场景验证 #

如果要测试多种错误消息的场景，JUnit 5支持用Lambda编写动态测试，让代码更简洁：

import org.junit.jupiter.api.DynamicTest;
import org.junit.jupiter.api.TestFactory;
import java.util.List;
import java.util.stream.Stream;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.verify;

@ExtendWith(MockitoExtension.class)
public class YourProducerDynamicTest {

    @Mock
    private AmqpTemplate amqpTemplate;

    @InjectMocks
    private YourProducerClass producer;

    @TestFactory
    Stream<DynamicTest> testSendWithDifferentErrorReasons() {
        List<String> errorReasons = List.of("Connection failed", "Queue full", "Message expired");

        return DynamicTest.stream(
                errorReasons.stream(),
                reason -> "Testing death reason: " + reason,
                reason -> {
                    // 执行测试
                    producer.send(new Message("test".getBytes()), new RuntimeException(reason), "queue", "key");

                    // 捕获并验证
                    ArgumentCaptor<MessagePostProcessor> captor = ArgumentCaptor.forClass(MessagePostProcessor.class);
                    verify(amqpTemplate).convertAndSend(
                            eq(RabbitConfiguration.ERROR_EXCHANGE),
                            eq(RabbitConfiguration.ERROR_ROUTING_KEY),
                            any(Message.class),
                            captor.capture()
                    );

                    Message processed = captor.getValue().postProcessMessage(new Message("test".getBytes()));
                    assertEquals(reason, processed.getMessageProperties().getHeaders().get("x-death-reason"));
                }
        );
    }
}

简单来说，JUnit 5本身就支持Lambda作为测试逻辑，而要验证代码里的Lambda行为，Mockito的参数捕获是最直接的方案，这样就能完全覆盖你这段生产者代码的测试啦～

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