fix test

Signed-off-by: yisaer <disxiaofei@163.com>
tikv · May 12, 2021 · bce4c3e · bce4c3e
1 parent 42c2c92
commit bce4c3e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/server/schedulers/hot_test.go b/server/schedulers/hot_test.go
@@ -672,12 +672,12 @@ func (s *testHotReadRegionSchedulerSuite) TestByteRateOnly(c *C) {
 	c.Assert(tc.IsRegionHot(tc.GetRegion(1)), IsTrue)
 	c.Assert(tc.IsRegionHot(tc.GetRegion(11)), IsFalse)
 	// check randomly pick hot region
-	r := tc.RandHotRegionFromStore(2, statistics.ReadFlow)
+	r := tc.HotRegionsFromStore(2, statistics.ReadFlow)
 	c.Assert(r, NotNil)
-	c.Assert(r.GetID(), Equals, uint64(2))
+	c.Assert(len(r), Equals, 3)
 	// check hot items
 	stats := tc.HotCache.RegionStats(statistics.ReadFlow, 0)
-	c.Assert(len(stats), Equals, 2)
+	c.Assert(len(stats), Equals, 3)
 	for _, ss := range stats {
 		for _, s := range ss {
 			c.Assert(s.GetLoad(statistics.RegionReadBytes), Equals, 512.0*KB)