diff --git a/src/main/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormat.java b/src/main/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormat.java index 79d9703..7877866 100644 --- a/src/main/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormat.java +++ b/src/main/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormat.java @@ -132,4 +132,13 @@ Lucene95CustomCodec.Mode getMode() { public int getCompressionLevel() { return compressionLevel; } + + + public CompressionMode getZstdCompressionMode() { + return zstdCompressionMode; + } + + public CompressionMode getZstdNoDictCompressionMode() { + return zstdNoDictCompressionMode; + } } diff --git a/src/test/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormatTests.java b/src/test/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormatTests.java index e87fb56..57d2672 100644 --- a/src/test/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormatTests.java +++ b/src/test/java/org/opensearch/index/codec/customcodecs/Lucene95CustomStoredFieldsFormatTests.java @@ -44,4 +44,9 @@ public void testZstdNoDictLucene95CustomCodecModeWithCompressionLevel() { assertEquals(randomCompressionLevel, lucene95CustomStoredFieldsFormat.getCompressionLevel()); } + public void testCompressionModes(){ + Lucene95CustomStoredFieldsFormat lucene95CustomStoredFieldsFormat = new Lucene95CustomStoredFieldsFormat(); + assertTrue(lucene95CustomStoredFieldsFormat.getZstdCompressionMode() instanceof ZstdCompressionMode); + assertTrue(lucene95CustomStoredFieldsFormat.getZstdNoDictCompressionMode() instanceof ZstdNoDictCompressionMode); + } }