Fix flaky test in InputStreamStreamMessageTest (#5702)

Close #5695
line · May 27, 2024 · 1b28c8e · 1b28c8e
1 parent 2414099
commit 1b28c8e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/core/src/test/java/com/linecorp/armeria/common/stream/InputStreamStreamMessageTest.java b/core/src/test/java/com/linecorp/armeria/common/stream/InputStreamStreamMessageTest.java
@@ -407,7 +407,7 @@ void emptyInputStream() {
         StepVerifier.create(byteStreamMessage)
                     .verifyComplete();
         assertThat(byteStreamMessage.isEmpty()).isTrue();
-        assertThat(byteStreamMessage.isOpen()).isFalse();
+        await().untilAsserted(() -> assertThat(byteStreamMessage.isOpen()).isFalse());
     }
 
     @Test
@@ -420,7 +420,7 @@ void closedInputStream() throws IOException {
         assertThatThrownBy(() -> byteStreamMessage.subscribe().join())
                 .hasCauseInstanceOf(IOException.class);
         assertThat(byteStreamMessage.isEmpty()).isTrue();
-        assertThat(byteStreamMessage.isOpen()).isFalse();
+        await().untilAsserted(() -> assertThat(byteStreamMessage.isOpen()).isFalse());
     }
 
     @Test
@@ -447,7 +447,7 @@ public int read() throws IOException {
                     .expectNext(HttpData.wrap(new byte[] { 4 }))
                     .verifyError(IOException.class);
         assertThat(byteStreamMessage.isEmpty()).isFalse();
-        assertThat(byteStreamMessage.isOpen()).isFalse();
+        await().untilAsserted(() -> assertThat(byteStreamMessage.isOpen()).isFalse());
     }
 
     @Test
@@ -463,7 +463,7 @@ void aborted() {
                     .expectNext(HttpData.wrap(new byte[] { 1, 2, 3 }))
                     .then(byteStreamMessage::abort)
                     .verifyError(AbortedStreamException.class);
-        assertThat(byteStreamMessage.isOpen()).isFalse();
+        await().untilAsserted(() -> assertThat(byteStreamMessage.isOpen()).isFalse());
     }
 
     @Test