add test

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

pkg/mcs/scheduling/server/cluster.go

@@ -94,6 +94,12 @@ func (c *Cluster) GetHotStat() *statistics.HotStat {
 	return c.hotStat
 }
 
+// GetStoresStats returns stores' statistics from cluster.
+// And it will be unnecessary to filter unhealthy store, because it has been solved in process heartbeat
+func (c *Cluster) GetStoresStats() *statistics.StoresStats {
+	return c.hotStat.StoresStats
+}
+
 // GetRegionStats gets region statistics.
 func (c *Cluster) GetRegionStats() *statistics.RegionStatistics {
 	return c.regionStats

pkg/schedule/config/config_provider.go

@@ -119,6 +119,7 @@ type SharedConfigProvider interface {
 	GetHotRegionCacheHitsThreshold() int
 
 	// for test purpose
+	SetPlacementRuleEnabled(bool)
 	SetPlacementRulesCacheEnabled(bool)
 	SetEnableWitness(bool)
 }

tests/integrations/mcs/scheduling/api_test.go

@@ -13,6 +13,7 @@ import (
 	_ "github.com/tikv/pd/pkg/mcs/scheduling/server/apis/v1"
 	"github.com/tikv/pd/pkg/schedule/handler"
 	"github.com/tikv/pd/pkg/statistics"
+	"github.com/tikv/pd/pkg/storage"
 	"github.com/tikv/pd/pkg/utils/apiutil"
 	"github.com/tikv/pd/pkg/utils/tempurl"
 	"github.com/tikv/pd/pkg/utils/testutil"
@@ -212,4 +213,8 @@ func (suite *apiTestSuite) TestAPIForward() {
 	err = testutil.ReadGetJSON(re, testDialClient, fmt.Sprintf("%s/%s", urlPrefix, "/hotspot/buckets"), &buckets,
 		testutil.WithHeader(re, apiutil.ForwardToMicroServiceHeader, "true"))
 	re.NoError(err)
+	var history storage.HistoryHotRegions
+	err = testutil.ReadGetJSON(re, testDialClient, fmt.Sprintf("%s/%s", urlPrefix, "/hotspot/regions/history"), &history,
+		testutil.WithHeader(re, apiutil.ForwardToMicroServiceHeader, "true"))
+	re.NoError(err)
 }