fix ut

Signed-off-by: okjiang <[email protected]>

Author: okJiang

server/cluster/cluster.go

@@ -2451,10 +2451,6 @@ func (c *RaftCluster) SetAllStoresLimit(typ storelimit.Type, ratePerMin float64)
 
 // refreshStoreRateLimit applies the schedule config's store limit to the in-memory store limiter.
 func (c *RaftCluster) refreshStoreRateLimit(storeID uint64, limitType storelimit.Type) {
-	// Only v1 uses StoreRateLimit for AddPeer/RemovePeer.
-	if c.opt.GetStoreLimitVersion() != storelimit.VersionV1 {
-		return
-	}
 	store := c.GetStore(storeID)
 	if store == nil {
 		return

server/cluster/cluster_test.go

@@ -2905,8 +2905,14 @@ func TestCheckCache(t *testing.T) {
 
 	// case 2: operator cannot be created due to store limit restriction
 	oc.RemoveOperator(oc.GetOperator(1))
-	err := tc.SetStoreLimit(1, storelimit.AddPeer, 0)
+	err := tc.SetStoreLimit(1, storelimit.AddPeer, 0.0001)
 	re.NoError(err)
+	// StoreRateLimit treats rate=0 as unlimited. To simulate a throttled store,
+	// we set a small positive rate and exhaust the initial tokens.
+	store := tc.GetStore(1)
+	re.NotNil(store)
+	re.True(store.GetStoreLimit().Take(storelimit.RegionInfluence[storelimit.AddPeer], storelimit.AddPeer, constant.Low))
+	re.False(store.IsAvailable(storelimit.AddPeer, constant.Low))
 	checker.PatrolRegions()
 	re.Len(checker.GetPendingProcessedRegions(), 1)