Spring Kafka单元测试触发监听器，但该方法无法使用consumer.poll获取消息

import org.apache.kafka.clients.consumer.ConsumerRecord;
import org.apache.kafka.clients.consumer.MockConsumer;
import org.apache.kafka.common.TopicPartition;
import org.junit.jupiter.api.Test;
import org.springframework.kafka.listener.MessageListenerContainer;

import java.util.Collections;

public class KafkaListenerTest {

    @Test
    public void testListener() {
        // 创建MockConsumer对象
        MockConsumer<String, String> consumer = new MockConsumer<>();
        
        // 创建监听器
        MyMessageListener listener = new MyMessageListener();
        
        // 创建MessageListenerContainer对象，并设置MockConsumer作为消费者
        MessageListenerContainer container = new ConcurrentMessageListenerContainer<>(consumer);
        container.setupMessageListener(listener);
        
        // 创建一个主题分区
        TopicPartition topicPartition = new TopicPartition("topic", 0);
        
        // 订阅主题分区
        consumer.assign(Collections.singletonList(topicPartition));
        
        // 发送模拟消息
        ConsumerRecord<String, String> record = new ConsumerRecord<>("topic", 0, 0, "key", "value");
        consumer.addRecord(record);
        
        // 触发监听器
        container.poll();
        
        // 验证监听器的行为是否符合预期
        // ...
    }
}