You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.6 KiB
62 lines
1.6 KiB
3 years ago
|
package log
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestFilterAll(t *testing.T) {
|
||
|
|
logger := With(DefaultLogger, "ts", DefaultTimestamp, "caller", DefaultCaller)
|
||
|
|
log := NewFilter(logger,
|
||
|
|
FilterLevel(LevelDebug),
|
||
|
|
FilterKey("username"),
|
||
|
|
FilterValue("hello"),
|
||
|
|
FilterFunc(testFilterFunc),
|
||
|
|
)
|
||
|
|
|
||
|
|
log.Log(LevelDebug, "msg", "test debug")
|
||
|
|
log.Info("hello")
|
||
|
|
log.Infow("password", "123456")
|
||
|
|
log.Infow("username", "test debug")
|
||
|
|
log.Warn("warn log")
|
||
|
|
}
|
||
|
|
func TestFilterLevel(t *testing.T) {
|
||
|
|
logger := With(DefaultLogger, "ts", DefaultTimestamp, "caller", DefaultCaller)
|
||
|
|
log := NewFilter(logger, FilterLevel(LevelWarn))
|
||
|
|
log.Log(LevelDebug, "msg1", "te1st debug")
|
||
|
|
log.Debug("test debug")
|
||
|
|
log.Debugf("test %s", "debug")
|
||
|
|
log.Debugw("log", "test debug")
|
||
|
|
log.Warn("warn log")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFilterKey(t *testing.T) {
|
||
|
|
logger := With(DefaultLogger, "ts", DefaultTimestamp, "caller", DefaultCaller)
|
||
|
|
log := NewFilter(logger, FilterKey("password"))
|
||
|
|
log.Debugw("password", "123456")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFilterValue(t *testing.T) {
|
||
|
|
logger := With(DefaultLogger, "ts", DefaultTimestamp, "caller", DefaultCaller)
|
||
|
|
log := NewFilter(logger, FilterValue("debug"))
|
||
|
|
log.Debugf("test %s", "debug")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFilterFunc(t *testing.T) {
|
||
|
|
logger := With(DefaultLogger, "ts", DefaultTimestamp, "caller", DefaultCaller)
|
||
|
|
log := NewFilter(logger, FilterFunc(testFilterFunc))
|
||
|
|
log.Debug("debug level")
|
||
|
|
log.Infow("password", "123456")
|
||
|
|
}
|
||
|
|
|
||
|
|
func testFilterFunc(level Level, keyvals ...interface{}) bool {
|
||
|
|
if level == LevelWarn {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
for i := 0; i < len(keyvals); i++ {
|
||
|
|
if keyvals[i] == "password" {
|
||
|
|
keyvals[i+1] = "***"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|