KafkaTestUtils.getRecords()只返回第一个发送的记录

以下是使用KafkaTestUtils.getRecords()返回所有发送的记录的解决方法的代码示例：

import org.apache.kafka.clients.consumer.ConsumerRecord;
import org.apache.kafka.common.serialization.StringDeserializer;
import org.junit.jupiter.api.Test;
import org.springframework.kafka.test.context.EmbeddedKafka;
import org.springframework.kafka.test.utils.KafkaTestUtils;
import org.springframework.test.context.junit.jupiter.SpringJUnitConfig;

import java.util.List;

import static org.junit.jupiter.api.Assertions.assertEquals;

@SpringJUnitConfig
@EmbeddedKafka(partitions = 1, topics = "test-topic")
public class KafkaTest {

    @Test
    public void testKafkaConsumer() throws InterruptedException {
        String topic = "test-topic";
        String message1 = "Message 1";
        String message2 = "Message 2";

        // Produce messages
        KafkaTestUtils.produceRecords(KafkaTestUtils.producerProps(embeddedKafkaBroker()), topic, message1, message2);

        // Create a Kafka consumer
        Map<String, Object> consumerProps = KafkaTestUtils.consumerProps("test-group", "true", embeddedKafkaBroker());
        consumerProps.put(ConsumerConfig.KEY_DESERIALIZER_CLASS_CONFIG, StringDeserializer.class);
        consumerProps.put(ConsumerConfig.VALUE_DESERIALIZER_CLASS_CONFIG, StringDeserializer.class);
        DefaultKafkaConsumerFactory<String, String> consumerFactory = new DefaultKafkaConsumerFactory<>(consumerProps);
        KafkaConsumer<String, String> consumer = consumerFactory.createConsumer();
        consumer.subscribe(Collections.singleton(topic));

        // Consume records
        ConsumerRecords<String, String> records;
        List<ConsumerRecord<String, String>> allRecords = new ArrayList<>();

        do {
            records = KafkaTestUtils.getRecords(consumer);
            allRecords.addAll(records.records(topic));
            consumer.commitSync();
        } while (!records.isEmpty());

        // Assert that all messages were consumed
        assertEquals(2, allRecords.size());
        assertEquals(message1, allRecords.get(0).value());
        assertEquals(message2, allRecords.get(1).value());
    }
}

在上述示例中，我们首先使用KafkaTestUtils.produceRecords()方法发送两条消息到名为"test-topic"的主题。然后，我们创建一个Kafka消费者，并订阅该主题。

为了获取所有发送的记录，我们使用一个循环来重复调用KafkaTestUtils.getRecords()方法，直到返回的记录为空。在每次循环中，我们将返回的记录添加到一个列表中。最后，我们可以使用断言来验证所有的消息是否都被消费，并且它们的值与预期的值相匹配。