Extend existing tests

Author: mike-evolved

java/src/test/java/org/rocksdb/DBOptionsTest.java

@@ -453,6 +453,26 @@ public void setWriteBufferManagerWithZeroBufferSize() throws RocksDBException {
     }
   }
 
+  @Test
+  public void writeBufferManagerDynamicSettings() throws RocksDBException {
+    try (final DBOptions opt = new DBOptions(); final Cache cache = new LRUCache(8 * 1024 * 1024);
+         final WriteBufferManager writeBufferManager = new WriteBufferManager(4 * 1024 * 1024, cache, false)) {
+      opt.setWriteBufferManager(writeBufferManager);
+      
+      // Test initial state
+      assertThat(writeBufferManager.enabled()).isTrue();
+      assertThat(writeBufferManager.bufferSize()).isEqualTo(4 * 1024 * 1024);
+      assertThat(writeBufferManager.allowStall()).isFalse();
+      
+      // Test dynamic updates
+      writeBufferManager.setBufferSize(8 * 1024 * 1024);
+      assertThat(writeBufferManager.bufferSize()).isEqualTo(8 * 1024 * 1024);
+      
+      writeBufferManager.setAllowStall(true);
+      assertThat(writeBufferManager.allowStall()).isTrue();
+    }
+  }
+
   @Test
   public void compactionReadaheadSize() {
     try(final DBOptions opt = new DBOptions()) {

java/src/test/java/org/rocksdb/OptionsTest.java

@@ -699,6 +699,32 @@ public void setWriteBufferManagerWithAllowStall() throws RocksDBException {
     }
   }
 
+  @Test
+  public void writeBufferManagerMonitoring() throws RocksDBException {
+    try (final Options opt = new Options(); final Cache cache = new LRUCache(1024 * 1024);
+         final WriteBufferManager writeBufferManager = new WriteBufferManager(8 * 1024 * 1024, cache, true)) {
+      opt.setWriteBufferManager(writeBufferManager);
+      
+      // Test monitoring methods
+      assertThat(writeBufferManager.enabled()).isTrue();
+      assertThat(writeBufferManager.costToCache()).isTrue();
+      assertThat(writeBufferManager.bufferSize()).isEqualTo(8 * 1024 * 1024);
+      assertThat(writeBufferManager.memoryUsage()).isEqualTo(0L);
+      assertThat(writeBufferManager.mutableMemtableMemoryUsage()).isEqualTo(0L);
+      assertThat(writeBufferManager.dummyEntriesInCacheUsage()).isEqualTo(0L);
+      assertThat(writeBufferManager.isStallActive()).isFalse();
+      assertThat(writeBufferManager.isStallThresholdExceeded()).isFalse();
+      
+      // Test dynamic buffer size change
+      writeBufferManager.setBufferSize(16 * 1024 * 1024);
+      assertThat(writeBufferManager.bufferSize()).isEqualTo(16 * 1024 * 1024);
+      
+      // Test dynamic allowStall change
+      writeBufferManager.setAllowStall(false);
+      assertThat(writeBufferManager.allowStall()).isFalse();
+    }
+  }
+
   @Test
   public void compactionReadaheadSize() {
     try (final Options opt = new Options()) {