diff --git a/server/src/test/java/org/opensearch/index/translog/LocalTranslogTests.java b/server/src/test/java/org/opensearch/index/translog/LocalTranslogTests.java index d61b63e6ff53b..ec015a3049d38 100644 --- a/server/src/test/java/org/opensearch/index/translog/LocalTranslogTests.java +++ b/server/src/test/java/org/opensearch/index/translog/LocalTranslogTests.java @@ -3368,8 +3368,8 @@ public void testWithRandomException() throws IOException { localCheckpointOfSafeCommit = failableTLog.getDeletionPolicy().getLocalCheckpointOfSafeCommit(); IOUtils.closeWhileHandlingException(failableTLog); } - } catch (TranslogException | MockDirectoryWrapper.FakeIOException ex) { - // failed - that's ok, we didn't even create it + } catch (TranslogException | MockDirectoryWrapper.FakeIOException | TranslogCorruptedException ex) { + // failed - that's ok, we didn't even create it or it was corrupted from partial writes } catch (IOException ex) { assertEquals(ex.getMessage(), "__FAKE__ no space left on device"); } @@ -3379,8 +3379,8 @@ public void testWithRandomException() throws IOException { TranslogDeletionPolicy deletionPolicy = createTranslogDeletionPolicy(); deletionPolicy.setLocalCheckpointOfSafeCommit(localCheckpointOfSafeCommit); IOUtils.close(getFailableTranslog(fail, config, randomBoolean(), false, generationUUID, deletionPolicy)); - } catch (TranslogException | MockDirectoryWrapper.FakeIOException ex) { - // failed - that's ok, we didn't even create it + } catch (TranslogException | MockDirectoryWrapper.FakeIOException | TranslogCorruptedException ex) { + // failed - that's ok, we didn't even create it or it was corrupted from partial writes } catch (IOException ex) { assertEquals(ex.getMessage(), "__FAKE__ no space left on device"); }