From 8310c23ac44c14cf22fbd424077446bc67b12c6d Mon Sep 17 00:00:00 2001
From: Andriy Redko <andriy.redko@aiven.io>
Date: Tue, 17 Sep 2024 15:44:32 -0400
Subject: [PATCH] Fix flaky terminaton conditions for
 org.opensearch.rest.ReactorNetty4StreamingStressIT.testCloseClientStreamingRequest
 test case (#15959)

Signed-off-by: Andriy Redko <andriy.redko@aiven.io>
(cherry picked from commit eb5b70370e548cd0f3b51721397df818b3050daa)
---
 .../org/opensearch/rest/ReactorNetty4StreamingStressIT.java    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/plugins/transport-reactor-netty4/src/javaRestTest/java/org/opensearch/rest/ReactorNetty4StreamingStressIT.java b/plugins/transport-reactor-netty4/src/javaRestTest/java/org/opensearch/rest/ReactorNetty4StreamingStressIT.java
index bb349425fd441..e8efe81e40f8f 100644
--- a/plugins/transport-reactor-netty4/src/javaRestTest/java/org/opensearch/rest/ReactorNetty4StreamingStressIT.java
+++ b/plugins/transport-reactor-netty4/src/javaRestTest/java/org/opensearch/rest/ReactorNetty4StreamingStressIT.java
@@ -17,6 +17,7 @@
 import org.junit.After;
 
 import java.io.IOException;
+import java.io.InterruptedIOException;
 import java.io.UncheckedIOException;
 import java.nio.ByteBuffer;
 import java.nio.charset.StandardCharsets;
@@ -75,7 +76,7 @@ public void testCloseClientStreamingRequest() throws Exception {
                 }
             })
             .then(() -> scheduler.advanceTimeBy(delay))
-            .expectErrorMatches(t -> t instanceof ConnectionClosedException)
+            .expectErrorMatches(t -> t instanceof InterruptedIOException || t instanceof ConnectionClosedException)
             .verify();
     }
 }