如何测试@KafkaListener的方法是否被调用

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.kafka.core.KafkaTemplate;
import org.springframework.kafka.test.context.EmbeddedKafka;
import org.springframework.test.context.TestPropertySource;
import static org.assertj.core.api.Assertions.assertThat;

@SpringBootTest
@EmbeddedKafka // 使用内嵌的Kafka进行测试
@TestPropertySource(properties = { "spring.kafka.bootstrap-servers=${spring.embedded.kafka.brokers}" })
public class KafkaListenerTest {

    @Autowired
    private KafkaTemplate<String, String> kafkaTemplate;

    @Autowired
    private MyKafkaListener myKafkaListener;

    @Test
    public void testKafkaListenerMethod() throws InterruptedException {
        String topic = "test-topic";
        String message = "test-message";

        // 发送一条测试消息
        kafkaTemplate.send(topic, message);

        // 等待一段时间以确保消息被处理
        Thread.sleep(1000);

        // 验证方法是否被调用
        assertThat(myKafkaListener.isMethodCalled()).isTrue();
    }
}