Skip to content

Commit 84efda3

Browse files
committed
GH-1121: Fix backported test
1 parent d3f13ef commit 84efda3

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

spring-kafka-test/src/test/java/org/springframework/kafka/test/utils/KafkaTestUtilsTests.java

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -22,30 +22,33 @@
2222
import org.apache.kafka.clients.consumer.KafkaConsumer;
2323
import org.apache.kafka.clients.producer.KafkaProducer;
2424
import org.apache.kafka.clients.producer.ProducerRecord;
25-
import org.junit.jupiter.api.Test;
25+
import org.junit.ClassRule;
26+
import org.junit.Test;
2627

27-
import org.springframework.kafka.test.EmbeddedKafkaBroker;
28-
import org.springframework.kafka.test.context.EmbeddedKafka;
28+
import org.springframework.kafka.test.rule.EmbeddedKafkaRule;
2929

3030
/**
3131
* @author Gary Russell
3232
* @since 2.2.7
3333
*
3434
*/
35-
@EmbeddedKafka(topics = { "singleTopic1", "singleTopic2" })
3635
public class KafkaTestUtilsTests {
3736

37+
@ClassRule
38+
public static EmbeddedKafkaRule broker = new EmbeddedKafkaRule(1, true, 1, "singleTopic1", "singleTopic2");
39+
3840
@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());
4143
KafkaProducer<Integer, String> producer = new KafkaProducer<>(producerProps);
4244
producer.send(new ProducerRecord<>("singleTopic1", 1, "foo"));
4345
producer.send(new ProducerRecord<>("singleTopic2", 1, "foo"));
4446
producer.close();
45-
Map<String, Object> consumerProps = KafkaTestUtils.consumerProps("ktuTests", "false", broker);
47+
Map<String, Object> consumerProps = KafkaTestUtils.consumerProps("ktuTests", "false",
48+
broker.getEmbeddedKafka());
4649
consumerProps.put(ConsumerConfig.AUTO_OFFSET_RESET_CONFIG, "earliest");
4750
KafkaConsumer<Integer, String> consumer = new KafkaConsumer<>(consumerProps);
48-
broker.consumeFromAllEmbeddedTopics(consumer);
51+
broker.getEmbeddedKafka().consumeFromAllEmbeddedTopics(consumer);
4952
KafkaTestUtils.getSingleRecord(consumer, "singleTopic1");
5053
KafkaTestUtils.getSingleRecord(consumer, "singleTopic2");
5154
consumer.close();

0 commit comments

Comments
 (0)