*: avoid flaky test with high load scenario (#57016)

pingcap · Oct 31, 2024 · c240fdf · c240fdf
1 parent b314a9c
commit c240fdf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pkg/statistics/handle/cache/internal/lfu/lfu_cache_test.go b/pkg/statistics/handle/cache/internal/lfu/lfu_cache_test.go
@@ -310,6 +310,7 @@ func TestMemoryControlWithUpdate(t *testing.T) {
 		t1 := testutil.NewMockStatisticsTable(i, 1, true, false, false)
 		lfu.Put(1, t1)
 	}
-	time.Sleep(1 * time.Second)
-	require.Equal(t, int64(0), lfu.Cost())
+	require.Eventually(t, func() bool {
+		return int64(0) == lfu.Cost()
+	}, 5*time.Second, 100*time.Millisecond)
 }