tests: reduce mcs test duration (#7435)

ref #4399

Signed-off-by: lhy1024 <admin@liudos.us>

Co-authored-by: ti-chi-bot[bot] <108142056+ti-chi-bot[bot]@users.noreply.github.com>

pkg/schedule/operator/operator_controller.go

@@ -799,6 +799,12 @@ func (oc *Controller) GetFastOpInfluence(cluster *core.BasicCluster, influence O
 	}
 }
 
+// CleanAllOpRecords removes all operators' records.
+// It is used in tests only.
+func (oc *Controller) CleanAllOpRecords() {
+	oc.records.ttl.Clear()
+}
+
 // AddOpInfluence add operator influence for cluster
 func AddOpInfluence(op *Operator, influence OpInfluence, cluster *core.BasicCluster) {
 	region := cluster.GetRegion(op.RegionID())

pkg/statistics/hot_cache.go

@@ -205,3 +205,10 @@ func (w *HotCache) GetThresholds(kind utils.RWType, storeID uint64) []float64 {
 	}
 	return nil
 }
+
+// CleanCache cleans the cache.
+// This is used for test purpose.
+func (w *HotCache) CleanCache() {
+	w.writeCache.removeAllItem()
+	w.readCache.removeAllItem()
+}

pkg/statistics/hot_peer_cache.go

@@ -544,6 +544,18 @@ func (f *hotPeerCache) removeItem(item *HotPeerStat) {
 	}
 }
 
+// removeAllItem removes all items of the cache.
+// It is used for test.
+func (f *hotPeerCache) removeAllItem() {
+	for _, peers := range f.peersOfStore {
+		for _, peer := range peers.GetAll() {
+			item := peer.(*HotPeerStat)
+			item.actionType = utils.Remove
+			f.updateStat(item)
+		}
+	}
+}
+
 func (f *hotPeerCache) coldItem(newItem, oldItem *HotPeerStat) {
 	newItem.HotDegree = oldItem.HotDegree - 1
 	newItem.AntiCount = oldItem.AntiCount - 1