From 678adb3e2f0ab54fc8b001dc379247ceb26d559a Mon Sep 17 00:00:00 2001
From: a631807682 <631807682@qq.com>
Date: Wed, 19 Oct 2022 14:27:35 +0800
Subject: [PATCH] test: fix test

---
 tests/prepared_stmt_test.go | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/tests/prepared_stmt_test.go b/tests/prepared_stmt_test.go
index 6c141851..64baa01b 100644
--- a/tests/prepared_stmt_test.go
+++ b/tests/prepared_stmt_test.go
@@ -171,16 +171,17 @@ func TestPreparedStmtInTransaction(t *testing.T) {
 
 func TestPreparedStmtReset(t *testing.T) {
 	tx := DB.Session(&gorm.Session{PrepareStmt: true})
+
+	user := *GetUser("prepared_stmt_reset", Config{})
+	tx = tx.Create(&user)
+
 	pdb, ok := tx.ConnPool.(*gorm.PreparedStmtDB)
 	if !ok {
 		t.Fatalf("should assign PreparedStatement Manager back to database when using PrepareStmt mode")
 	}
 
-	user := *GetUser("prepared_stmt_reset", Config{})
-	tx.Create(&user)
-
 	pdb.Mux.Lock()
-	if len(pdb.PreparedSQL) == 0 || len(pdb.Stmts) == 0 {
+	if len(pdb.Stmts) == 0 {
 		pdb.Mux.Unlock()
 		t.Fatalf("prepared stmt can not be empty")
 	}
@@ -189,7 +190,7 @@ func TestPreparedStmtReset(t *testing.T) {
 	pdb.Reset()
 	pdb.Mux.Lock()
 	defer pdb.Mux.Unlock()
-	if len(pdb.PreparedSQL) != 0 || len(pdb.Stmts) != 0 {
+	if len(pdb.Stmts) != 0 {
 		t.Fatalf("prepared stmt should be empty")
 	}
 }