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.
105 lines
2.6 KiB
105 lines
2.6 KiB
package log
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/bilibili/kratos/pkg/net/metadata"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func initStdout() {
|
|
conf := &Config{
|
|
Stdout: true,
|
|
}
|
|
Init(conf)
|
|
}
|
|
|
|
func initFile() {
|
|
conf := &Config{
|
|
Dir: "/tmp",
|
|
// VLevel: 2,
|
|
Module: map[string]int32{"log_test": 1},
|
|
}
|
|
Init(conf)
|
|
}
|
|
|
|
type TestLog struct {
|
|
A string
|
|
B int
|
|
C string
|
|
D string
|
|
}
|
|
|
|
func testLog(t *testing.T) {
|
|
t.Run("Error", func(t *testing.T) {
|
|
Error("hello %s", "world")
|
|
Errorv(context.Background(), KV("key", 2222222), KV("test2", "test"))
|
|
Errorc(context.Background(), "keys: %s %s...", "key1", "key2")
|
|
})
|
|
t.Run("Warn", func(t *testing.T) {
|
|
Warn("hello %s", "world")
|
|
Warnv(context.Background(), KV("key", 2222222), KV("test2", "test"))
|
|
Warnc(context.Background(), "keys: %s %s...", "key1", "key2")
|
|
})
|
|
t.Run("Info", func(t *testing.T) {
|
|
Info("hello %s", "world")
|
|
Infov(context.Background(), KV("key", 2222222), KV("test2", "test"))
|
|
Infoc(context.Background(), "keys: %s %s...", "key1", "key2")
|
|
})
|
|
}
|
|
|
|
func TestFile(t *testing.T) {
|
|
initFile()
|
|
testLog(t)
|
|
assert.Equal(t, nil, Close())
|
|
}
|
|
|
|
func TestStdout(t *testing.T) {
|
|
initStdout()
|
|
testLog(t)
|
|
assert.Equal(t, nil, Close())
|
|
}
|
|
|
|
func TestLogW(t *testing.T) {
|
|
D := logw([]interface{}{"i", "like", "a", "dog"})
|
|
if len(D) != 2 || D[0].Key != "i" || D[0].Value != "like" || D[1].Key != "a" || D[1].Value != "dog" {
|
|
t.Fatalf("logw out put should be ' {i like} {a dog}'")
|
|
}
|
|
D = logw([]interface{}{"i", "like", "dog"})
|
|
if len(D) != 1 || D[0].Key != "i" || D[0].Value != "like" {
|
|
t.Fatalf("logw out put should be ' {i like}'")
|
|
}
|
|
}
|
|
|
|
func TestLogWithMirror(t *testing.T) {
|
|
Info("test log")
|
|
mdcontext := metadata.NewContext(context.Background(), metadata.MD{metadata.Mirror: "true"})
|
|
Infov(mdcontext, KV("key1", "val1"), KV("key2", ""), KV("log", "log content"), KV("msg", "msg content"))
|
|
|
|
mdcontext = metadata.NewContext(context.Background(), metadata.MD{metadata.Mirror: "***"})
|
|
Infov(mdcontext, KV("key1", "val1"), KV("key2", ""), KV("log", "log content"), KV("msg", "msg content"))
|
|
|
|
Infov(context.Background(), KV("key1", "val1"), KV("key2", ""), KV("log", "log content"), KV("msg", "msg content"))
|
|
|
|
}
|
|
|
|
func TestOverwriteSouce(t *testing.T) {
|
|
ctx := context.Background()
|
|
t.Run("test source kv string", func(t *testing.T) {
|
|
Infov(ctx, KVString("source", "test"))
|
|
})
|
|
t.Run("test source kv string", func(t *testing.T) {
|
|
Infov(ctx, KV("source", "test"))
|
|
})
|
|
}
|
|
|
|
func BenchmarkLog(b *testing.B) {
|
|
ctx := context.Background()
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
Infov(ctx, KVString("test", "hello"), KV("int", 34), KV("hhh", "hhhh"))
|
|
}
|
|
})
|
|
}
|
|
|