diff --git a/pkg/schedule/schedulers/hot_region_test.go b/pkg/schedule/schedulers/hot_region_test.go index 401090d099f..641f40b94f8 100644 --- a/pkg/schedule/schedulers/hot_region_test.go +++ b/pkg/schedule/schedulers/hot_region_test.go @@ -781,8 +781,10 @@ func TestHotWriteRegionScheduleByteRateOnlyWithTiFlash(t *testing.T) { allowLeaderTiKVCount := aliveTiKVCount - 1 // store 5 with evict leader aliveTiFlashCount := float64(aliveTiFlashLastID - aliveTiFlashStartID + 1) tc.ObserveRegionsStats() - ops, _ := hb.Schedule(tc, false) - re.NotEmpty(ops) + testutil.Eventually(re, func() bool { + ops, _ := hb.Schedule(tc, false) + return len(ops) != 0 + }) re.True( loadsEqual( hb.stLoadInfos[writeLeader][1].LoadPred.Expect.Loads, @@ -801,8 +803,10 @@ func TestHotWriteRegionScheduleByteRateOnlyWithTiFlash(t *testing.T) { pdServerCfg.FlowRoundByDigit = 8 tc.SetPDServerConfig(pdServerCfg) clearPendingInfluence(hb) - ops, _ = hb.Schedule(tc, false) - re.NotEmpty(ops) + testutil.Eventually(re, func() bool { + ops, _ := hb.Schedule(tc, false) + return len(ops) != 0 + }) re.True( loadsEqual( hb.stLoadInfos[writePeer][8].LoadPred.Expect.Loads, diff --git a/pkg/schedule/schedulers/scheduler_test.go b/pkg/schedule/schedulers/scheduler_test.go index 90f14d90fb0..d503931f0b2 100644 --- a/pkg/schedule/schedulers/scheduler_test.go +++ b/pkg/schedule/schedulers/scheduler_test.go @@ -317,6 +317,7 @@ func TestShuffleRegionRole(t *testing.T) { } func TestSpecialUseHotRegion(t *testing.T) { + schedulePeerPr = 1.0 re := require.New(t) cancel, _, tc, oc := prepareSchedulersTest() defer cancel()