KAFKA-14938: Fixing flaky test org.apache.kafka.connect.integration.E…

…xactlyOnceSourceIntegrationTest#testConnectorBoundary
apache · Apr 27, 2023 · 512fd6e · 512fd6e
1 parent 007c0d3
commit 512fd6e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.../src/test/java/org/apache/kafka/connect/integration/ExactlyOnceSourceIntegrationTest.java b/.../src/test/java/org/apache/kafka/connect/integration/ExactlyOnceSourceIntegrationTest.java
@@ -412,10 +412,10 @@ public void testConnectorBoundary() throws Exception {
         Map<String, Object> consumerProps = new HashMap<>();
         consumerProps.put(ConsumerConfig.ISOLATION_LEVEL_CONFIG, "read_committed");
         // consume all records from the source topic or fail, to ensure that they were correctly produced
-        ConsumerRecords<byte[], byte[]> sourceRecords = connect.kafka().consumeAll(
+        ConsumerRecords<byte[], byte[]> sourceRecords = connect.kafka().consume(
+                MINIMUM_MESSAGES,
                 CONSUME_RECORDS_TIMEOUT_MS,
                 Collections.singletonMap(ConsumerConfig.ISOLATION_LEVEL_CONFIG, "read_committed"),
-                null,
                 topic
         );
         assertTrue("Not enough records produced by source connector. Expected at least: " + MINIMUM_MESSAGES + " + but got " + sourceRecords.count(),