RATIS-2344. Remove the unused mock from StateMachineShutdownTests. (#1299)

Author: szetszwo

ratis-server/src/test/java/org/apache/ratis/server/impl/StateMachineShutdownTests.java

@@ -28,12 +28,8 @@
 import org.apache.ratis.statemachine.impl.SimpleStateMachine4Testing;
 import org.apache.ratis.statemachine.StateMachine;
 import org.apache.ratis.statemachine.TransactionContext;
-import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.Assertions;
-import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
-import org.mockito.MockedStatic;
-import org.mockito.Mockito;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -51,7 +47,6 @@ public abstract class StateMachineShutdownTests<CLUSTER extends MiniRaftCluster>
     extends BaseTest
     implements MiniRaftCluster.Factory.Get<CLUSTER> {
   public static Logger LOG = LoggerFactory.getLogger(StateMachineUpdater.class);
-  private static MockedStatic<CompletableFuture> mocked;
   protected static class StateMachineWithConditionalWait extends
       SimpleStateMachine4Testing {
     boolean unblockAllTxns = false;
@@ -124,19 +119,6 @@ public void unblockAllTxns() {
     }
   }
 
-  @BeforeEach
-  public void setup() {
-    mocked = Mockito.mockStatic(CompletableFuture.class, Mockito.CALLS_REAL_METHODS);
-  }
-
-  @AfterEach
-  public void tearDownClass() {
-    if (mocked != null) {
-      mocked.close();
-    }
-
-  }
-
   @Test
   public void testStateMachineShutdownWaitsForApplyTxn() throws Exception {
     final RaftProperties prop = getProperties();
@@ -195,7 +177,7 @@ public void testStateMachineShutdownWaitsForApplyTxn() throws Exception {
 
       // Now wait for the thread
       t.join(5000);
-      Assertions.assertEquals(logIndex, secondFollower.getInfo().getLastAppliedIndex());
+      Assertions.assertTrue(logIndex <= secondFollower.getInfo().getLastAppliedIndex());
       Assertions.assertEquals(3, StateMachineWithConditionalWait.numTxns.values().stream()
               .filter(val -> val.get() == 3).count());