Add more test cases

Author: mike-evolved

java/src/test/java/org/rocksdb/FlushWALOptionsTest.java

@@ -26,6 +26,18 @@ public class FlushWALOptionsTest {
   public void newFlushWALOptions() {
     try (final FlushWALOptions flushWALOptions = new FlushWALOptions()) {
       assertThat(flushWALOptions).isNotNull();
+      // Verify defaults
+      assertThat(flushWALOptions.sync()).isFalse();
+      assertThat(flushWALOptions.rateLimiterPriority()).isEqualTo(IOPriority.IO_TOTAL);
+    }
+  }
+
+  @Test
+  public void newFlushWALOptionsWithSync() {
+    try (final FlushWALOptions flushWALOptions = new FlushWALOptions(true)) {
+      assertThat(flushWALOptions).isNotNull();
+      assertThat(flushWALOptions.sync()).isTrue();
+      assertThat(flushWALOptions.rateLimiterPriority()).isEqualTo(IOPriority.IO_TOTAL);
     }
   }
 
@@ -51,13 +63,35 @@ public void rateLimiterPriority() {
       // Default value should be IO_TOTAL
       assertThat(flushWALOptions.rateLimiterPriority()).isEqualTo(IOPriority.IO_TOTAL);
 
-      // Test setting to IO_HIGH
+      // Test all IOPriority values
+      flushWALOptions.setRateLimiterPriority(IOPriority.IO_LOW);
+      assertThat(flushWALOptions.rateLimiterPriority()).isEqualTo(IOPriority.IO_LOW);
+      
+      flushWALOptions.setRateLimiterPriority(IOPriority.IO_MID);
+      assertThat(flushWALOptions.rateLimiterPriority()).isEqualTo(IOPriority.IO_MID);
+      
       flushWALOptions.setRateLimiterPriority(IOPriority.IO_HIGH);
       assertThat(flushWALOptions.rateLimiterPriority()).isEqualTo(IOPriority.IO_HIGH);
 
-      // Test setting to IO_LOW
-      flushWALOptions.setRateLimiterPriority(IOPriority.IO_LOW);
-      assertThat(flushWALOptions.rateLimiterPriority()).isEqualTo(IOPriority.IO_LOW);
+      flushWALOptions.setRateLimiterPriority(IOPriority.IO_USER);
+      assertThat(flushWALOptions.rateLimiterPriority()).isEqualTo(IOPriority.IO_USER);
+      
+      flushWALOptions.setRateLimiterPriority(IOPriority.IO_TOTAL);
+      assertThat(flushWALOptions.rateLimiterPriority()).isEqualTo(IOPriority.IO_TOTAL);
+    }
+  }
+
+  @Test
+  public void fluentAPI() {
+    try (final FlushWALOptions flushWALOptions = new FlushWALOptions()) {
+      // Test fluent chaining
+      FlushWALOptions result = flushWALOptions
+          .setSync(true)
+          .setRateLimiterPriority(IOPriority.IO_HIGH);
+      
+      assertThat(result).isSameAs(flushWALOptions);
+      assertThat(flushWALOptions.sync()).isTrue();
+      assertThat(flushWALOptions.rateLimiterPriority()).isEqualTo(IOPriority.IO_HIGH);
     }
   }
 
@@ -109,6 +143,33 @@ public void flushWalWithRateLimiter() throws RocksDBException {
 
       // Verify data is accessible
       assertThat(new String(db.get("key1".getBytes()))).isEqualTo("value1");
+      
+      // Test with different priority
+      db.put("key2".getBytes(), "value2".getBytes());
+      flushWALOptions.setRateLimiterPriority(IOPriority.IO_LOW);
+      db.flushWal(flushWALOptions);
+      
+      assertThat(new String(db.get("key2".getBytes()))).isEqualTo("value2");
+    }
+  }
+
+  @Test
+  public void backwardCompatibility() throws RocksDBException {
+    try (final Options options = new Options()
+                                     .setCreateIfMissing(true)
+                                     .setManualWalFlush(true);
+         final RocksDB db = RocksDB.open(options, dbFolder.getRoot().getAbsolutePath())) {
+      
+      // Test old flushWal(boolean) still works
+      db.put("key1".getBytes(), "value1".getBytes());
+      db.flushWal(false);
+      
+      db.put("key2".getBytes(), "value2".getBytes());
+      db.flushWal(true);
+      
+      // Verify data is accessible
+      assertThat(new String(db.get("key1".getBytes()))).isEqualTo("value1");
+      assertThat(new String(db.get("key2".getBytes()))).isEqualTo("value2");
     }
   }
 }