如何模拟来自KafkaTemplate的结果

import org.apache.kafka.clients.producer.ProducerRecord;
import org.apache.kafka.clients.producer.RecordMetadata;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import org.springframework.kafka.core.KafkaTemplate;
import org.springframework.kafka.support.SendResult;
import org.springframework.util.concurrent.ListenableFuture;

import java.util.concurrent.ExecutionException;

import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.when;

public class KafkaTemplateMockTest {

    @Test
    public void testKafkaTemplate() throws ExecutionException, InterruptedException {
        // 创建一个Mock的KafkaTemplate对象
        KafkaTemplate<String, String> kafkaTemplate = Mockito.mock(KafkaTemplate.class);

        // 创建一个Mock的ListenableFuture对象
        ListenableFuture<SendResult<String, String>> future = Mockito.mock(ListenableFuture.class);

        // 创建一个Mock的RecordMetadata对象
        RecordMetadata recordMetadata = Mockito.mock(RecordMetadata.class);

        // 设置Mock的ListenableFuture对象返回的结果
        when(future.get()).thenReturn(new SendResult<>(new ProducerRecord<>("topic", "key", "value"), recordMetadata));

        // 设置Mock的KafkaTemplate对象发送消息时返回的ListenableFuture对象
        when(kafkaTemplate.send(any(String.class), any(String.class))).thenReturn(future);

        // 调用KafkaTemplate发送消息
        kafkaTemplate.send("topic", "key", "value");

        // 验证发送结果
        Mockito.verify(kafkaTemplate).send("topic", "key", "value");
    }
}