main_test.go 620 B

1234567891011121314151617181920212223242526272829303132333435
  1. package migrations
  2. import (
  3. "flag"
  4. "fmt"
  5. "os"
  6. "testing"
  7. "gorm.io/gorm/logger"
  8. log "unknwon.dev/clog/v2"
  9. "gogs.io/gogs/internal/testutil"
  10. )
  11. func TestMain(m *testing.M) {
  12. flag.Parse()
  13. level := logger.Silent
  14. if !testing.Verbose() {
  15. // Remove the primary logger and register a noop logger.
  16. log.Remove(log.DefaultConsoleName)
  17. err := log.New("noop", testutil.InitNoopLogger)
  18. if err != nil {
  19. fmt.Println(err)
  20. os.Exit(1)
  21. }
  22. } else {
  23. level = logger.Info
  24. }
  25. // NOTE: AutoMigrate does not respect logger passed in gorm.Config.
  26. logger.Default = logger.Default.LogMode(level)
  27. os.Exit(m.Run())
  28. }