mleku/gogs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

dbtest: properly close test connection (#7598)

Browse Source
This commit is contained in:
Joe Chen
2023-11-12 19:09:50 -05:00
committed by GitHub
parent b644d797b4
commit 31a0964e12
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/dbtest/dbtest.go
View File

@@ -55,7 +55,12 @@ func NewDB(t *testing.T, suite string, tables ...any) *gorm.DB {
dbOpts.Name = dbName dbOpts.Name = dbName
cleanup = func(_ *gorm.DB) { cleanup = func(db *gorm.DB) {
testDB, err := db.DB()
if err == nil {
_ = testDB.Close()
}
_, _ = sqlDB.Exec(fmt.Sprintf("DROP DATABASE `%s`", dbName)) _, _ = sqlDB.Exec(fmt.Sprintf("DROP DATABASE `%s`", dbName))
_ = sqlDB.Close() _ = sqlDB.Close()
} }
@@ -86,7 +91,12 @@ func NewDB(t *testing.T, suite string, tables ...any) *gorm.DB {
dbOpts.Name = dbName dbOpts.Name = dbName
cleanup = func(_ *gorm.DB) { cleanup = func(db *gorm.DB) {
testDB, err := db.DB()
if err == nil {
_ = testDB.Close()
}
_, _ = sqlDB.Exec(fmt.Sprintf(`DROP DATABASE %q`, dbName)) _, _ = sqlDB.Exec(fmt.Sprintf(`DROP DATABASE %q`, dbName))
_ = sqlDB.Close() _ = sqlDB.Close()
} }