Fixing flaky test #9776

Signed-off-by: Dharmesh 💤 <sdharms@amazon.com>
opensearch-project · Sep 6, 2023 · 8c54a86 · 8c54a86
1 parent 899ae5d
commit 8c54a86
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/...t/java/org/opensearch/common/blobstore/stream/read/listener/ReadContextListenerTests.java b/...t/java/org/opensearch/common/blobstore/stream/read/listener/ReadContextListenerTests.java
@@ -70,7 +70,7 @@ public void testReadContextListener() throws InterruptedException, IOException {
         assertEquals(NUMBER_OF_PARTS * PART_SIZE, Files.size(fileLocation));
     }
 
-    public void testReadContextListenerFailure() throws InterruptedException {
+    public void testReadContextListenerFailure() throws Exception {
         Path fileLocation = path.resolve(UUID.randomUUID().toString());
         List<InputStreamContainer> blobPartStreams = initializeBlobPartStreams();
         CountDownLatch countDownLatch = new CountDownLatch(1);
@@ -99,8 +99,7 @@ public int available() {
         readContextListener.onResponse(readContext);
 
         countDownLatch.await();
-
-        assertFalse(Files.exists(fileLocation));
+        assertBusy(() -> { assertFalse(Files.exists(fileLocation)); });
     }
 
     public void testReadContextListenerException() {