mirror of
https://github.com/gogs/gogs.git
synced 2025-05-28 02:02:09 +00:00
db: simplify GORM logger init in tests (#6444)
This commit is contained in:
parent
c875950c43
commit
72af17bbbe
@ -7,8 +7,6 @@ package db
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
stdlog "log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
@ -20,14 +18,12 @@ import (
|
||||
log "unknwon.dev/clog/v2"
|
||||
|
||||
"gogs.io/gogs/internal/conf"
|
||||
"gogs.io/gogs/internal/dbutil"
|
||||
"gogs.io/gogs/internal/testutil"
|
||||
)
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
flag.Parse()
|
||||
|
||||
var w logger.Writer
|
||||
level := logger.Silent
|
||||
if !testing.Verbose() {
|
||||
// Remove the primary logger and register a noop logger.
|
||||
@ -37,18 +33,12 @@ func TestMain(m *testing.M) {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
w = &dbutil.Logger{Writer: ioutil.Discard}
|
||||
} else {
|
||||
w = stdlog.New(os.Stdout, "\r\n", stdlog.LstdFlags)
|
||||
level = logger.Info
|
||||
}
|
||||
|
||||
// NOTE: AutoMigrate does not respect logger passed in gorm.Config.
|
||||
logger.Default = logger.New(w, logger.Config{
|
||||
SlowThreshold: 100 * time.Millisecond,
|
||||
LogLevel: level,
|
||||
})
|
||||
logger.Default = logger.Default.LogMode(level)
|
||||
|
||||
os.Exit(m.Run())
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user