103 lines
2.2 KiB
Go
103 lines
2.2 KiB
Go
|
package log
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"go-common/library/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)
|
||
|
|
}
|
||
|
|
|
||
|
|
func initAgent() {
|
||
|
|
conf := &Config{
|
||
|
|
Agent: &AgentConfig{
|
||
|
|
TaskID: "000003",
|
||
|
|
Addr: "172.16.0.204:514",
|
||
|
|
Proto: "tcp",
|
||
|
|
Chan: 1024,
|
||
|
|
Buffer: 10,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
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"))
|
||
|
|
})
|
||
|
|
t.Run("Warn", func(t *testing.T) {
|
||
|
|
Warn("hello %s", "world")
|
||
|
|
Warnv(context.Background(), KV("key", 2222222), KV("test2", "test"))
|
||
|
|
})
|
||
|
|
t.Run("Info", func(t *testing.T) {
|
||
|
|
Info("hello %s", "world")
|
||
|
|
Infov(context.Background(), KV("key", 2222222), KV("test2", "test"))
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLogAgent(t *testing.T) {
|
||
|
|
initAgent()
|
||
|
|
testLog(t)
|
||
|
|
assert.Equal(t, nil, Close())
|
||
|
|
}
|
||
|
|
|
||
|
|
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"))
|
||
|
|
|
||
|
|
}
|