| 16 | ) |
| 17 | |
| 18 | func TestMain(m *testing.M) { |
| 19 | flag.Parse() |
| 20 | |
| 21 | level := logger.Silent |
| 22 | if !testing.Verbose() { |
| 23 | // Remove the primary logger and register a noop logger. |
| 24 | log.Remove(log.DefaultConsoleName) |
| 25 | err := log.New("noop", testutil.InitNoopLogger) |
| 26 | if err != nil { |
| 27 | fmt.Println(err) |
| 28 | os.Exit(1) |
| 29 | } |
| 30 | } else { |
| 31 | level = logger.Info |
| 32 | } |
| 33 | |
| 34 | // NOTE: AutoMigrate does not respect logger passed in gorm.Config. |
| 35 | logger.Default = logger.Default.LogMode(level) |
| 36 | |
| 37 | switch os.Getenv("GOGS_DATABASE_TYPE") { |
| 38 | case "mysql": |
| 39 | conf.UseMySQL = true |
| 40 | case "postgres": |
| 41 | conf.UsePostgreSQL = true |
| 42 | default: |
| 43 | conf.UseSQLite3 = true |
| 44 | } |
| 45 | |
| 46 | os.Exit(m.Run()) |
| 47 | } |
| 48 | |
| 49 | func newTestDB(t *testing.T, suite string) *gorm.DB { |
| 50 | return dbtest.NewDB(t, suite, append(Tables, legacyTables...)...) |