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

hawkingrei · web-flow · commit c240fdf085ce · 2024-10-31T10:23:27.000Z
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)
 }

Original file line number	Diff line number	Diff line change
`@@ -310,6 +310,7 @@ func TestMemoryControlWithUpdate(t *testing.T) {`
`310`	`310`	`t1 := testutil.NewMockStatisticsTable(i, 1, true, false, false)`
`311`	`311`	`lfu.Put(1, t1)`
`312`	`312`	`}`
`313`		`- time.Sleep(1 * time.Second)`
`314`		`- require.Equal(t, int64(0), lfu.Cost())`
	`313`	`+ require.Eventually(t, func() bool {`
	`314`	`+ return int64(0) == lfu.Cost()`
	`315`	`+ }, 5time.Second, 100time.Millisecond)`
`315`	`316`	`}`