|
22 | 22 | import org.apache.kafka.clients.consumer.KafkaConsumer; |
23 | 23 | import org.apache.kafka.clients.producer.KafkaProducer; |
24 | 24 | import org.apache.kafka.clients.producer.ProducerRecord; |
25 | | -import org.junit.jupiter.api.Test; |
| 25 | +import org.junit.ClassRule; |
| 26 | +import org.junit.Test; |
26 | 27 |
|
27 | | -import org.springframework.kafka.test.EmbeddedKafkaBroker; |
28 | | -import org.springframework.kafka.test.context.EmbeddedKafka; |
| 28 | +import org.springframework.kafka.test.rule.EmbeddedKafkaRule; |
29 | 29 |
|
30 | 30 | /** |
31 | 31 | * @author Gary Russell |
32 | 32 | * @since 2.2.7 |
33 | 33 | * |
34 | 34 | */ |
35 | | -@EmbeddedKafka(topics = { "singleTopic1", "singleTopic2" }) |
36 | 35 | public class KafkaTestUtilsTests { |
37 | 36 |
|
| 37 | + @ClassRule |
| 38 | + public static EmbeddedKafkaRule broker = new EmbeddedKafkaRule(1, true, 1, "singleTopic1", "singleTopic2"); |
| 39 | + |
38 | 40 | @Test |
39 | | - void testGetSingleWithMoreThatOneTopic(EmbeddedKafkaBroker broker) { |
40 | | - Map<String, Object> producerProps = KafkaTestUtils.producerProps(broker); |
| 41 | + public void testGetSingleWithMoreThatOneTopic() { |
| 42 | + Map<String, Object> producerProps = KafkaTestUtils.producerProps(broker.getEmbeddedKafka()); |
41 | 43 | KafkaProducer<Integer, String> producer = new KafkaProducer<>(producerProps); |
42 | 44 | producer.send(new ProducerRecord<>("singleTopic1", 1, "foo")); |
43 | 45 | producer.send(new ProducerRecord<>("singleTopic2", 1, "foo")); |
44 | 46 | producer.close(); |
45 | | - Map<String, Object> consumerProps = KafkaTestUtils.consumerProps("ktuTests", "false", broker); |
| 47 | + Map<String, Object> consumerProps = KafkaTestUtils.consumerProps("ktuTests", "false", |
| 48 | + broker.getEmbeddedKafka()); |
46 | 49 | consumerProps.put(ConsumerConfig.AUTO_OFFSET_RESET_CONFIG, "earliest"); |
47 | 50 | KafkaConsumer<Integer, String> consumer = new KafkaConsumer<>(consumerProps); |
48 | | - broker.consumeFromAllEmbeddedTopics(consumer); |
| 51 | + broker.getEmbeddedKafka().consumeFromAllEmbeddedTopics(consumer); |
49 | 52 | KafkaTestUtils.getSingleRecord(consumer, "singleTopic1"); |
50 | 53 | KafkaTestUtils.getSingleRecord(consumer, "singleTopic2"); |
51 | 54 | consumer.close(); |
|
0 commit comments