statstest: make test faster (#45772)

ref #44940
pingcap · Aug 3, 2023 · d43359a · d43359a
1 parent 41accbe
commit d43359a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/statistics/handle/handletest/statstest/BUILD.bazel b/statistics/handle/handletest/statstest/BUILD.bazel
@@ -9,7 +9,7 @@ go_test(
     ],
     flaky = True,
     race = "on",
-    shard_count = 8,
+    shard_count = 9,
     deps = [
         "//config",
         "//parser/model",

diff --git a/statistics/handle/handletest/statstest/stats_test.go b/statistics/handle/handletest/statstest/stats_test.go
@@ -186,15 +186,21 @@ func testInitStatsMemTrace(t *testing.T) {
 	require.Equal(t, h.GetMemConsumed(), memCostTot)
 }
 
-func TestInitStatsMemTrace(t *testing.T) {
+func TestInitStatsMemTraceWithLite(t *testing.T) {
+	testInitStatsMemTraceFunc(t, true)
+}
+
+func TestInitStatsMemTraceWithoutLite(t *testing.T) {
+	testInitStatsMemTraceFunc(t, false)
+}
+
+func testInitStatsMemTraceFunc(t *testing.T, liteInitStats bool) {
 	originValue := config.GetGlobalConfig().Performance.LiteInitStats
 	defer func() {
 		config.GetGlobalConfig().Performance.LiteInitStats = originValue
 	}()
-	for _, v := range []bool{false, true} {
-		config.GetGlobalConfig().Performance.LiteInitStats = v
-		testInitStatsMemTrace(t)
-	}
+	config.GetGlobalConfig().Performance.LiteInitStats = liteInitStats
+	testInitStatsMemTrace(t)
 }
 
 func TestInitStats(t *testing.T) {