如何让KafkaProducer使用模拟模式注册表进行测试？

<dependency>
    <groupId>org.apache.kafka</groupId>
    <artifactId>kafka-clients</artifactId>
    <version>2.8.0</version>
    <scope>test</scope>
</dependency>

import org.apache.kafka.clients.producer.KafkaProducer;
import org.mockito.Mockito;

// 创建模拟的KafkaProducer对象
KafkaProducer<String, String> mockProducer = Mockito.mock(KafkaProducer.class);

import org.apache.kafka.clients.producer.MockProducer;
import org.apache.kafka.clients.producer.ProducerRecord;

// 创建模拟的注册表
MockProducer<String, String> mockRegistry = new MockProducer<>();

// 设置模拟的注册表返回的结果
mockRegistry.completeNext();

// 将模拟的注册表设置给模拟的KafkaProducer
Mockito.when(mockProducer.send(Mockito.any(ProducerRecord.class))).thenReturn(mockRegistry.future());

import org.apache.kafka.clients.producer.ProducerRecord;
import org.apache.kafka.clients.producer.RecordMetadata;

// 创建要发送的消息
ProducerRecord<String, String> record = new ProducerRecord<>("topic", "key", "value");

// 发送消息并获取发送结果
RecordMetadata metadata = mockProducer.send(record).get();

// 验证发送结果
Assert.assertNotNull(metadata);