test(framework,plugins): fix flaky tests in bandwidth, stats and db-move

- BandWidthRuntimeTest: deploy contract before setDebug(true) to avoid
  CPU timeout triggered by debug mode during contract deployment
- TronStatsManagerTest: capture P2pStats snapshot before invoking work()
  and compare against snapshot instead of hardcoded 0, preventing
  failures when another test starts the real P2pService with non-zero stats
- DbMoveTest: add @after cleanup of output-directory-toolkit so LevelDB
  and RocksDB tests don't share the same relative destination path
- PeerStatusCheckMockTest: replace 5-second sleep with mock executor;
  verify scheduleWithFixedDelay args and run the captured Runnable directly

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: Little-Peony

framework/src/test/java/org/tron/common/runtime/vm/BandWidthRuntimeTest.java

@@ -24,6 +24,7 @@
 import org.junit.BeforeClass;
 import org.junit.Test;
 import org.tron.common.BaseTest;
+import org.tron.common.parameter.CommonParameter;
 import org.tron.common.runtime.RuntimeImpl;
 import org.tron.common.runtime.TvmTestUtils;
 import org.tron.common.utils.Commons;
@@ -153,8 +154,10 @@ public void testSuccess() {
 
   @Test
   public void testSuccessNoBandd() {
+    boolean originalDebug = CommonParameter.getInstance().isDebug();
     try {
       byte[] contractAddress = createContract();
+      CommonParameter.getInstance().setDebug(true);
       TriggerSmartContract triggerContract = TvmTestUtils.createTriggerContract(contractAddress,
           "setCoin(uint256)", "50", false,
           0, Commons.decodeFromBase58Check(TriggerOwnerTwoAddress));
@@ -185,6 +188,8 @@ public void testSuccessNoBandd() {
           balance);
     } catch (TronException e) {
       Assert.assertNotNull(e);
+    } finally {
+      CommonParameter.getInstance().setDebug(originalDebug);
     }
   }
 
@@ -254,4 +259,4 @@ public void testMaxContractResultSize() {
     }
     Assert.assertEquals(2, maxSize);
   }
-}
+}

framework/src/test/java/org/tron/core/net/peer/PeerStatusCheckMockTest.java

@@ -1,12 +1,18 @@
 package org.tron.core.net.peer;
 
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.spy;
 
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
 import org.junit.After;
 import org.junit.Assert;
 import org.junit.Test;
 import org.mockito.Mockito;
+import org.tron.common.utils.ReflectUtils;
 
 public class PeerStatusCheckMockTest {
   @After
@@ -15,17 +21,26 @@ public void  clearMocks() {
   }
 
   @Test
-  public void testInitException() throws InterruptedException {
+  public void testInitException() {
     PeerStatusCheck peerStatusCheck = spy(new PeerStatusCheck());
+    ScheduledExecutorService executor = mock(ScheduledExecutorService.class);
+    ReflectUtils.setFieldValue(peerStatusCheck, "peerStatusCheckExecutor", executor);
     doThrow(new RuntimeException("test exception")).when(peerStatusCheck).statusCheck();
+
     peerStatusCheck.init();
 
-    // the initialDelay of scheduleWithFixedDelay is 5s; wait for at least one execution
-    Thread.sleep(5000L);
+    Mockito.verify(executor).scheduleWithFixedDelay(any(Runnable.class), eq(5L), eq(2L),
+        eq(TimeUnit.SECONDS));
+    Runnable scheduledTask = Mockito.mockingDetails(executor).getInvocations().stream()
+        .filter(invocation -> invocation.getMethod().getName().equals("scheduleWithFixedDelay"))
+        .map(invocation -> (Runnable) invocation.getArgument(0))
+        .findFirst()
+        .orElseThrow(() -> new AssertionError("scheduled task was not registered"));
+
+    scheduledTask.run();
 
-    // Verify statusCheck() was invoked by the scheduler and the exception was handled gracefully
-    Mockito.verify(peerStatusCheck, Mockito.atLeastOnce()).statusCheck();
+    Mockito.verify(peerStatusCheck).statusCheck();
     Assert.assertNotNull(peerStatusCheck);
   }
 
-}
+}

framework/src/test/java/org/tron/core/net/services/TronStatsManagerTest.java

@@ -7,8 +7,10 @@
 
 import org.junit.Assert;
 import org.junit.Test;
+import org.tron.core.net.TronNetService;
 import org.tron.core.net.service.statistics.NodeStatistics;
 import org.tron.core.net.service.statistics.TronStatsManager;
+import org.tron.p2p.stats.P2pStats;
 import org.tron.protos.Protocol;
 
 public class TronStatsManagerTest {
@@ -50,14 +52,20 @@ public void testWork() throws Exception {
     Assert.assertEquals(field3.get(manager), 1L);
     Assert.assertEquals(field4.get(manager), 1L);
 
+    P2pStats statsSnapshot = TronNetService.getP2pService().getP2pStats();
+    long expectedTcpIn = statsSnapshot.getTcpInSize();
+    long expectedTcpOut = statsSnapshot.getTcpOutSize();
+    long expectedUdpIn = statsSnapshot.getUdpInSize();
+    long expectedUdpOut = statsSnapshot.getUdpOutSize();
+
     Method method = manager.getClass().getDeclaredMethod("work");
     method.setAccessible(true);
     method.invoke(manager);
 
-    Assert.assertEquals(field1.get(manager), 0L);
-    Assert.assertEquals(field2.get(manager), 0L);
-    Assert.assertEquals(field3.get(manager), 0L);
-    Assert.assertEquals(field4.get(manager), 0L);
+    Assert.assertEquals(expectedTcpIn, (long) field1.get(manager));
+    Assert.assertEquals(expectedTcpOut, (long) field2.get(manager));
+    Assert.assertEquals(expectedUdpIn, (long) field3.get(manager));
+    Assert.assertEquals(expectedUdpOut, (long) field4.get(manager));
   }
 
 }

plugins/src/test/java/org/tron/plugins/DbMoveTest.java

@@ -5,6 +5,7 @@
 import java.net.URL;
 import java.nio.file.Paths;
 import lombok.extern.slf4j.Slf4j;
+import org.junit.After;
 import org.junit.Assert;
 import org.junit.Rule;
 import org.junit.Test;
@@ -17,6 +18,8 @@
 @Slf4j
 public class DbMoveTest {
 
+  private static final String OUTPUT_DIRECTORY = "output-directory-toolkit";
+
   @Rule
   public TemporaryFolder temporaryFolder = new TemporaryFolder();
 
@@ -30,6 +33,29 @@ private void init(DbTool.DbType dbType, String path) throws IOException, RocksDB
     DbTool.getDB(path, TRANS, dbType).close();
   }
 
+  @After
+  public void destroy() {
+    deleteDir(new File(OUTPUT_DIRECTORY));
+  }
+
+  /**
+   * delete directory.
+   */
+  private static boolean deleteDir(File dir) {
+    if (dir.isDirectory()) {
+      String[] children = dir.list();
+      assert children != null;
+      for (String child : children) {
+        boolean success = deleteDir(new File(dir, child));
+        if (!success) {
+          logger.warn("can't delete dir:" + dir);
+          return false;
+        }
+      }
+    }
+    return dir.delete();
+  }
+
   private static String getConfig(String config) {
     URL path = DbMoveTest.class.getClassLoader().getResource(config);
     return path == null ? null : path.getPath();