fix: fix the error handle in tests_test

This commit is contained in:
liweiting.wt 2021-12-28 18:22:17 +08:00
parent 2c3fc2db28
commit b9667cb747
1 changed files with 11 additions and 4 deletions

View File

@ -25,12 +25,15 @@ func init() {
os.Exit(1)
} else {
sqlDB, err := DB.DB()
if err == nil {
err = sqlDB.Ping()
}
if err != nil {
log.Printf("failed to connect database, got error %v", err)
os.Exit(1)
}
err = sqlDB.Ping()
if err != nil {
log.Printf("failed to ping sqlDB, got error %v", err)
os.Exit(1)
}
RunMigrations()
@ -76,6 +79,10 @@ func OpenTestConnection() (db *gorm.DB, err error) {
db, err = gorm.Open(sqlite.Open(filepath.Join(os.TempDir(), "gorm.db")), &gorm.Config{})
}
if err != nil {
return
}
if debug := os.Getenv("DEBUG"); debug == "true" {
db.Logger = db.Logger.LogMode(logger.Info)
} else if debug == "false" {