cherry pick 40024 to fix unstable TestMaxPreparedStmtCount

pingcap · Apr 14, 2023 · 2c062e8 · 2c062e8
1 parent a706515
commit 2c062e8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/executor/prepared_test.go b/executor/prepared_test.go
@@ -18,6 +18,7 @@ import (
 	"fmt"
 	"strconv"
 	"strings"
+	"sync/atomic"
 	"testing"
 
 	"github.com/pingcap/tidb/parser/auth"
@@ -1254,6 +1255,11 @@ func TestIssue31141(t *testing.T) {
 }
 
 func TestMaxPreparedStmtCount(t *testing.T) {
+	oldVal := atomic.LoadInt64(&variable.PreparedStmtCount)
+	atomic.StoreInt64(&variable.PreparedStmtCount, 0)
+	defer func() {
+		atomic.StoreInt64(&variable.PreparedStmtCount, oldVal)
+	}()
 	store := testkit.CreateMockStore(t)
 	tk := testkit.NewTestKit(t, store)
 	tk.MustExec("set @@global.max_prepared_stmt_count = 2")