fix ci fail

redis/redis-console/src/test/java/com/ctrip/xpipe/redis/console/migration/status/migration/MigrationPartialSuccessStateTest.java

@@ -8,6 +8,7 @@
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.mockito.Mockito;
 import org.mockito.junit.MockitoJUnitRunner;
 
 import java.util.LinkedList;
@@ -82,8 +83,8 @@ public void testSingleThreadProcess() throws Exception {
         partialSuccessState.getStateActionState().tryAction();
 
         waitConditionUntilTimeOut(() -> retryCnt.get() >= failCnt);
-        verify(migrationCluster).updateStat(isA(MigrationPublishState.class));
-        verify(migrationCluster).process();
+        verify(migrationCluster, Mockito.timeout(3000).times(1)).updateStat(isA(MigrationPublishState.class));
+        verify(migrationCluster, Mockito.timeout(3000).times(1)).process();
     }
 
     @Test

redis/redis-keeper/src/test/java/com/ctrip/xpipe/redis/keeper/ratelimit/AllTests.java → redis/redis-keeper/src/test/java/com/ctrip/xpipe/redis/keeper/ratelimit/AllRateLimitTests.java

@@ -15,6 +15,6 @@
         FakeRedisRdbDumpLong.class,
         RateLimitTest.class,
 })
-public class AllTests {
+public class AllRateLimitTests {
 
 }

redis/redis-keeper/src/test/java/com/ctrip/xpipe/redis/keeper/ratelimit/RateLimitTest.java

@@ -84,10 +84,8 @@ public void testBackupDcActiveKeeper_DrMigration_ShouldLimit() throws Exception
         redisKeeperServer1.getRedisKeeperServerState().becomeActive(new DefaultEndPoint("127.0.0.1", redisKeeperServer2.getListeningPort()));
 
         waitRedisKeeperServerConnected(redisKeeperServer1);
-        waitConditionUntilTimeOut(()->assertSuccess(()->{
-            verify(leakyBucket, times(1)).tryAcquire();
-            verify(leakyBucket, times(1)).release();
-        }));
+        verify(leakyBucket, timeout(5000).times(1)).tryAcquire();
+        verify(leakyBucket, timeout(5000).times(1)).tryAcquire();
     }