fix test

Signed-off-by: HunDunDM <[email protected]>
tikv · Jan 14, 2022 · 76519b2 · 76519b2
1 parent 401395c
commit 76519b2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/schedule/region_scatterer_test.go b/server/schedule/region_scatterer_test.go
@@ -484,7 +484,7 @@ func (s *testScatterRegionSuite) TestSelectedStores(c *C) {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 	opt := config.NewTestOptions()
-	tc := mockcluster.NewCluster(ctx, opt)
+	tc := mockcluster.NewCluster(opt)
 	// Add 4 stores.
 	for i := uint64(1); i <= 4; i++ {
 		tc.AddRegionStore(i, 0)
@@ -520,7 +520,7 @@ func isPeerCountChanged(op *operator.Operator) bool {
 	for i := 0; i < op.Len(); i++ {
 		step := op.Step(i)
 		switch step.(type) {
-		case operator.AddPeer, operator.AddLearner:
+		case operator.AddPeer, operator.AddLearner, operator.AddLightPeer, operator.AddLightLearner:
 			add++
 		case operator.RemovePeer:
 			remove++