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

…StreamingStressIT.testCloseClientStreamingRequest test case (opensearch-project#15959) Signed-off-by: Andriy Redko <[email protected]>
sachinpkale · Sep 19, 2024 · 53cf9dc · 53cf9dc
1 parent edaef91
commit 53cf9dc
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
@@ -8,6 +8,7 @@
 
 package org.opensearch.rest;
 
+import org.apache.hc.core5.http.ConnectionClosedException;
 import org.opensearch.client.Request;
 import org.opensearch.client.Response;
 import org.opensearch.client.StreamingRequest;
@@ -75,7 +76,7 @@ public void testCloseClientStreamingRequest() throws Exception {
                 }
             })
             .then(() -> scheduler.advanceTimeBy(delay))
-            .expectErrorMatches(t -> t instanceof InterruptedIOException)
+            .expectErrorMatches(t -> t instanceof InterruptedIOException || t instanceof ConnectionClosedException)
             .verify();
     }
 }