executor: fix unstable TestMaxPreparedStmtCount (#40024)

ref pingcap/tidb#39735
This commit is contained in:
Yifan Xu
2022-12-19 15:12:54 +08:00
committed by GitHub
parent 08f6ea484f
commit 3edde02708

View File

@ -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")