diff options
| author | Joe Chen <jc@unknwon.io> | 2023-11-12 19:09:50 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-12 19:09:50 -0500 |
| commit | 31a0964e124e7e370fd8091f4a367e52e35a692c (patch) | |
| tree | 7f484875f12640e31c8fa77a59bdb6c0bfcccbb6 | |
| parent | b644d797b46c941ba2dadafb2f66cc9e96d50141 (diff) | |
dbtest: properly close test connection (#7598)
| -rw-r--r-- | internal/dbtest/dbtest.go | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/internal/dbtest/dbtest.go b/internal/dbtest/dbtest.go index 4e6b7e6a..5fab7b13 100644 --- a/internal/dbtest/dbtest.go +++ b/internal/dbtest/dbtest.go @@ -55,7 +55,12 @@ func NewDB(t *testing.T, suite string, tables ...any) *gorm.DB { 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.Close() } @@ -86,7 +91,12 @@ func NewDB(t *testing.T, suite string, tables ...any) *gorm.DB { 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.Close() } |