test(db): add doClose() test to verify dbSource closes despite writeOptions failure

Author: warku123

framework/src/test/java/org/tron/core/db/TronDatabaseTest.java

@@ -1,7 +1,13 @@
 package org.tron.core.db;
 
+import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.spy;
+import static org.mockito.Mockito.verify;
+
 import com.google.protobuf.InvalidProtocolBufferException;
 import java.io.IOException;
+import java.lang.reflect.Field;
 import org.junit.AfterClass;
 import org.junit.Assert;
 import org.junit.BeforeClass;
@@ -12,7 +18,9 @@
 import org.junit.rules.TemporaryFolder;
 import org.rocksdb.RocksDB;
 import org.tron.common.TestConstants;
+import org.tron.common.storage.WriteOptionsWrapper;
 import org.tron.core.config.args.Args;
+import org.tron.core.db.common.DbSourceInter;
 import org.tron.core.exception.BadItemException;
 import org.tron.core.exception.ItemNotFoundException;
 
@@ -99,4 +107,34 @@ public void TestGetFromRoot() throws
     Assert.assertEquals(db.getFromRoot("test".getBytes()),
         "test");
   }
+
+  @Test
+  public void testDoCloseDbSourceCalledWhenWriteOptionsThrows() throws Exception {
+    TronDatabase<String> db = new TronDatabase<String>("test-do-close") {
+      @Override public void put(byte[] key, String item) {}
+      @Override public void delete(byte[] key) {}
+      @Override public String get(byte[] key) { return null; }
+      @Override public boolean has(byte[] key) { return false; }
+    };
+
+    Field writeOptionsField = TronDatabase.class.getDeclaredField("writeOptions");
+    writeOptionsField.setAccessible(true);
+    WriteOptionsWrapper spyWriteOptions = spy((WriteOptionsWrapper) writeOptionsField.get(db));
+    doThrow(new RuntimeException("simulated writeOptions failure")).when(spyWriteOptions).close();
+    writeOptionsField.set(db, spyWriteOptions);
+
+    Field dbSourceField = TronDatabase.class.getDeclaredField("dbSource");
+    dbSourceField.setAccessible(true);
+    DbSourceInter<byte[]> originalDbSource = (DbSourceInter<byte[]>) dbSourceField.get(db);
+    DbSourceInter<byte[]> mockDbSource = mock(DbSourceInter.class);
+    dbSourceField.set(db, mockDbSource);
+
+    try {
+      db.doClose();
+      verify(spyWriteOptions).close();
+      verify(mockDbSource).closeDB();
+    } finally {
+      originalDbSource.closeDB();
+    }
+  }
 }