tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: fix test

Browse Source
This commit is contained in:
a631807682 2022-10-19 14:27:35 +08:00
parent 6ae2f6b00a
commit 678adb3e2f
No known key found for this signature in database
GPG Key ID: 137D1D75522168AB
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/prepared_stmt_test.go
View File

@ -171,16 +171,17 @@ func TestPreparedStmtInTransaction(t *testing.T) {
func TestPreparedStmtReset(t *testing.T) { func TestPreparedStmtReset(t *testing.T) {
tx := DB.Session(&gorm.Session{PrepareStmt: true}) tx := DB.Session(&gorm.Session{PrepareStmt: true})
user := *GetUser("prepared_stmt_reset", Config{})
tx = tx.Create(&user)
pdb, ok := tx.ConnPool.(*gorm.PreparedStmtDB) pdb, ok := tx.ConnPool.(*gorm.PreparedStmtDB)
if !ok { if !ok {
t.Fatalf("should assign PreparedStatement Manager back to database when using PrepareStmt mode") 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() pdb.Mux.Lock()
if len(pdb.PreparedSQL) == 0 || len(pdb.Stmts) == 0 { if len(pdb.Stmts) == 0 {
pdb.Mux.Unlock() pdb.Mux.Unlock()
t.Fatalf("prepared stmt can not be empty") t.Fatalf("prepared stmt can not be empty")
} }
@ -189,7 +190,7 @@ func TestPreparedStmtReset(t *testing.T) {
pdb.Reset() pdb.Reset()
pdb.Mux.Lock() pdb.Mux.Lock()
defer pdb.Mux.Unlock() 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") t.Fatalf("prepared stmt should be empty")
} }
} }