(flink-connector-kafka) 02/02: [FLINK-37611] Deflake KafkaWriterFaultToleranceITCase.testWriteExceptionWhenKafkaUnavailable

[arvid]

This is an automated email from the ASF dual-hosted git repository.

arvid pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/flink-connector-kafka.git

commit d74a7bd8b95522152102f4a8a72a8c12a2dfb2e5
Author: Arvid Heise <ar...@apache.org>
AuthorDate: Fri Apr 4 10:05:27 2025 +0200

    [FLINK-37611] Deflake 
KafkaWriterFaultToleranceITCase.testWriteExceptionWhenKafkaUnavailable
    
    Apparently, with newer kafka-clients library, we may also see 
TimeoutExceptions in some occasions.
---
 .../flink/connector/kafka/sink/KafkaWriterFaultToleranceITCase.java | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git 
a/flink-connector-kafka/src/test/java/org/apache/flink/connector/kafka/sink/KafkaWriterFaultToleranceITCase.java
 
b/flink-connector-kafka/src/test/java/org/apache/flink/connector/kafka/sink/KafkaWriterFaultToleranceITCase.java
index cccd322f..a3a9bc9c 100644
--- 
a/flink-connector-kafka/src/test/java/org/apache/flink/connector/kafka/sink/KafkaWriterFaultToleranceITCase.java
+++ 
b/flink-connector-kafka/src/test/java/org/apache/flink/connector/kafka/sink/KafkaWriterFaultToleranceITCase.java
@@ -67,7 +67,11 @@ public class KafkaWriterFaultToleranceITCase extends 
KafkaWriterTestBase {
             try {
                 writer.getCurrentProducer().flush();
                 assertThatCode(() -> writer.write(1, SINK_WRITER_CONTEXT))
-                        .hasRootCauseExactlyInstanceOf(NetworkException.class);
+                        .rootCause()
+                        .matches(
+                                e ->
+                                        e instanceof NetworkException
+                                                || e instanceof 
TimeoutException);
             } finally {
                 KAFKA_CONTAINER.start();
             }