USEFULL
/
gogs
mirror of https://github.com/gogs/gogs.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

dbtest: properly close test connection ()

pull/7607/head
Joe Chen 2023-11-12 19:09:50 -05:00 committed by GitHub
parent b644d797b4
commit 31a0964e12
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 2 deletions
internal/dbtest

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()
} }