Fix flaky terminaton conditions for org.opensearch.rest.ReactorNetty4…

…StreamingStressIT.testCloseClientStreamingRequest test case (opensearch-project#15959) Signed-off-by: Andriy Redko <[email protected]> (cherry picked from commit eb5b703)
reta · Sep 17, 2024 · 8310c23 · 8310c23
1 parent bfdc30c
commit 8310c23
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...ctor-netty4/src/javaRestTest/java/org/opensearch/rest/ReactorNetty4StreamingStressIT.java b/...ctor-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();
     }
 }