topsql: make unit test stable

util/topsql/topsql_test.go

@@ -124,7 +124,6 @@ func TestTopSQLReporter(t *testing.T) {
 	defer wg.Wait()
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
-
 	sqlMap := make(map[string]string)
 	sql2plan := make(map[string]string)
 	for _, req := range reqs {
@@ -143,29 +142,35 @@ func TestTopSQLReporter(t *testing.T) {
 			}
 		})
 	}
-	server.WaitCollectCnt(1, time.Second*5)
-	records := server.GetLatestRecords()
 	checkSQLPlanMap := map[string]struct{}{}
-	for _, req := range records {
-		require.Greater(t, len(req.Items), 0)
-		require.Greater(t, req.Items[0].CpuTimeMs, uint32(0))
-		sqlMeta, exist := server.GetSQLMetaByDigestBlocking(req.SqlDigest, time.Second)
-		require.True(t, exist)
-		expectedNormalizedSQL, exist := sqlMap[string(req.SqlDigest)]
-		require.True(t, exist)
-		require.Equal(t, expectedNormalizedSQL, sqlMeta.NormalizedSql)
-
-		expectedNormalizedPlan := sql2plan[expectedNormalizedSQL]
-		if expectedNormalizedPlan == "" || len(req.PlanDigest) == 0 {
-			require.Len(t, req.PlanDigest, 0)
-			continue
+	for retry := 0; retry < 5; retry++ {
+		server.WaitCollectCnt(1, time.Second*5)
+		records := server.GetLatestRecords()
+		for _, req := range records {
+			require.Greater(t, len(req.Items), 0)
+			require.Greater(t, req.Items[0].CpuTimeMs, uint32(0))
+			sqlMeta, exist := server.GetSQLMetaByDigestBlocking(req.SqlDigest, time.Second)
+			require.True(t, exist)
+			expectedNormalizedSQL, exist := sqlMap[string(req.SqlDigest)]
+			require.True(t, exist)
+			require.Equal(t, expectedNormalizedSQL, sqlMeta.NormalizedSql)
+
+			expectedNormalizedPlan := sql2plan[expectedNormalizedSQL]
+			if expectedNormalizedPlan == "" || len(req.PlanDigest) == 0 {
+				require.Len(t, req.PlanDigest, 0)
+				checkSQLPlanMap[expectedNormalizedSQL] = struct{}{}
+				continue
+			}
+			normalizedPlan, exist := server.GetPlanMetaByDigestBlocking(req.PlanDigest, time.Second)
+			require.True(t, exist)
+			require.Equal(t, expectedNormalizedPlan, normalizedPlan)
+			checkSQLPlanMap[expectedNormalizedSQL] = struct{}{}
+		}
+		if len(checkSQLPlanMap) == len(reqs) {
+			break
 		}
-		normalizedPlan, exist := server.GetPlanMetaByDigestBlocking(req.PlanDigest, time.Second)
-		require.True(t, exist)
-		require.Equal(t, expectedNormalizedPlan, normalizedPlan)
-		checkSQLPlanMap[expectedNormalizedSQL] = struct{}{}
 	}
-	require.Equal(t, 2, len(checkSQLPlanMap))
+	require.Equal(t, len(reqs), len(checkSQLPlanMap))
 }
 
 func TestMaxSQLAndPlanTest(t *testing.T) {