add test

Signed-off-by: Cabinfever_B <[email protected]>
tikv · Mar 3, 2022 · 32023ef · 32023ef
1 parent 2d81e5c
commit 32023ef
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/server/schedule/placement/region_rule_cache_test.go b/server/schedule/placement/region_rule_cache_test.go
@@ -177,6 +177,10 @@ func (s *testRuleSuite) TestRegionRuleFitCache(c *C) {
 		c.Log(testcase.name)
 		c.Assert(cache.IsUnchanged(testcase.region, testcase.rules, mockStores(3)), Equals, testcase.unchanged)
 	}
+	for _, testcase := range testcases {
+		c.Log(testcase.name)
+		c.Assert(cache.IsUnchanged(testcase.region, testcase.rules, mockStoresNoHeartbeat(3)), Equals, false)
+	}
 	// Invalid Input4
 	c.Assert(cache.IsUnchanged(mockRegion(3, 0), addExtraRules(0), nil), IsFalse)
 	// Invalid Input5
@@ -207,3 +211,11 @@ func mockStores(num int) []*core.StoreInfo {
 	}
 	return stores
 }
+
+func mockStoresNoHeartbeat(num int) []*core.StoreInfo {
+	stores := make([]*core.StoreInfo, 0, num)
+	for i := 1; i <= num; i++ {
+		stores = append(stores, core.NewStoreInfo(&metapb.Store{Id: uint64(i)}))
+	}
+	return stores
+}