executor: fix unstable TestMaxPreparedStmtCount (#40024)

ref #39735
pingcap · Dec 19, 2022 · 3edde02 · 3edde02
1 parent 08f6ea4
commit 3edde02
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"
@@ -1258,6 +1259,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")