fix: stabilize flaky issue #68051

Author: flaky-claw

pkg/statistics/handle/handletest/initstats/init_stats_test.go

@@ -95,6 +95,7 @@ func TestLiteInitStatsWithTableIDs(t *testing.T) {
 	withStatsLease(t, -1, func() {
 		dom, err = session.BootstrapSession(store)
 		require.NoError(t, err)
+		dom.InfoSyncer().SetSessionManager(&testkit.MockSessionManager{})
 		h := dom.StatsHandle()
 		_, ok := h.Get(tbl1.Meta().ID)
 		require.False(t, ok)
@@ -152,6 +153,7 @@ func TestNonLiteInitStatsWithTableIDs(t *testing.T) {
 	withStatsLease(t, -1, func() {
 		dom, err = session.BootstrapSession(store)
 		require.NoError(t, err)
+		dom.InfoSyncer().SetSessionManager(&testkit.MockSessionManager{})
 		is = dom.InfoSchema()
 		h := dom.StatsHandle()
 		_, ok := h.Get(tbl1.Meta().ID)
@@ -365,20 +367,27 @@ func TestNonLiteInitStatsAndCheckTheLastTableStats(t *testing.T) {
 	withStatsLease(t, -1, func() {
 		dom, err = session.BootstrapSession(store)
 		require.NoError(t, err)
+		dom.InfoSyncer().SetSessionManager(&testkit.MockSessionManager{})
 		is = dom.InfoSchema()
 		h := dom.StatsHandle()
 		_, ok := h.Get(tbl1.Meta().ID)
 		require.False(t, ok)
 		require.NoError(t, h.InitStats(context.Background(), is))
 		stats1, ok := h.Get(tbl1.Meta().ID)
 		require.True(t, ok)
-		require.True(t, stats1.GetIdx(1).IsFullLoad())
+		idx1 := stats1.GetIdx(tbl1.Meta().Indices[0].ID)
+		require.NotNil(t, idx1)
+		require.True(t, idx1.IsFullLoad())
 		stats2, ok := h.Get(tbl2.Meta().ID)
 		require.True(t, ok)
-		require.True(t, stats2.GetIdx(1).IsFullLoad())
+		idx2 := stats2.GetIdx(tbl2.Meta().Indices[0].ID)
+		require.NotNil(t, idx2)
+		require.True(t, idx2.IsFullLoad())
 		stats3, ok := h.Get(tbl3.Meta().ID)
 		require.True(t, ok)
-		require.True(t, stats3.GetIdx(1).IsFullLoad())
+		idx3 := stats3.GetIdx(tbl3.Meta().Indices[0].ID)
+		require.NotNil(t, idx3)
+		require.True(t, idx3.IsFullLoad())
 
 		dom.Close()
 	})