114 lines
2.8 KiB
Go
114 lines
2.8 KiB
Go
package log_test
|
||
|
||
import (
|
||
"os"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
|
||
"apigo.cc/go/log"
|
||
)
|
||
|
||
func TestSplitTag(t *testing.T) {
|
||
logFile := "test_split.log"
|
||
// 使用每分钟切分的标签，方便测试 (当然实际可能需要模拟时间，但这里我们可以直接写然后检查文件名)
|
||
splitTag := ".200601021504"
|
||
|
||
conf := log.Config{
|
||
Name: "test-split",
|
||
Level: "info",
|
||
File: logFile,
|
||
SplitTag: splitTag,
|
||
}
|
||
logger := log.NewLogger(conf)
|
||
|
||
logger.Info("split test message")
|
||
|
||
// 给异步写入一点时间
|
||
time.Sleep(300 * time.Millisecond)
|
||
|
||
// 预期文件名
|
||
expectedFile := logFile + "." + time.Now().Format(splitTag)
|
||
|
||
if _, err := os.Stat(expectedFile); os.IsNotExist(err) {
|
||
t.Errorf("Expected log file %s does not exist", expectedFile)
|
||
} else {
|
||
// 清理
|
||
os.Remove(expectedFile)
|
||
}
|
||
}
|
||
|
||
func TestSensitiveDetailed(t *testing.T) {
|
||
type SecretLog struct {
|
||
log.BaseLog
|
||
Password string
|
||
SecretKey string
|
||
SafeData string
|
||
}
|
||
|
||
entry := log.GetEntry[SecretLog]()
|
||
entry.BaseLog.LogType = "secret"
|
||
entry.Password = "my_password"
|
||
entry.SecretKey = "super_secret"
|
||
entry.SafeData = "hello"
|
||
|
||
// 直接测试 ToArrayBytes
|
||
// 注意：passed to ToArrayBytes 的 keys 应该是已经过 fixField 处理的
|
||
sensitiveKeys := []string{"password", "secretkey"}
|
||
buf := log.ToArrayBytes(entry, sensitiveKeys)
|
||
result := string(buf)
|
||
|
||
if strings.Contains(result, "my_password") {
|
||
t.Errorf("Sensitive data 'my_password' not masked in: %s", result)
|
||
}
|
||
if strings.Contains(result, "super_secret") {
|
||
t.Errorf("Sensitive data 'super_secret' not masked in: %s", result)
|
||
}
|
||
if !strings.Contains(result, "hello") {
|
||
t.Errorf("Safe data 'hello' should be present in: %s", result)
|
||
}
|
||
}
|
||
|
||
func TestDeepDesensitization(t *testing.T) {
|
||
type Nested struct {
|
||
Password string
|
||
Token string
|
||
}
|
||
type DeepLog struct {
|
||
log.BaseLog
|
||
Data map[string]any
|
||
User Nested
|
||
}
|
||
|
||
entry := log.GetEntry[DeepLog]()
|
||
entry.BaseLog.LogType = "deep"
|
||
entry.Data = map[string]any{
|
||
"password": "data_password",
|
||
"safe": "data_safe",
|
||
}
|
||
entry.User = Nested{
|
||
Password: "user_password",
|
||
Token: "user_token",
|
||
}
|
||
|
||
sensitiveKeys := []string{"password", "token"}
|
||
buf := log.ToArrayBytes(entry, sensitiveKeys)
|
||
result := string(buf)
|
||
|
||
// Check deep desensitization in map
|
||
if strings.Contains(result, "data_password") {
|
||
t.Errorf("Nested map data 'data_password' not masked in: %s", result)
|
||
}
|
||
// Check deep desensitization in struct
|
||
if strings.Contains(result, "user_password") {
|
||
t.Errorf("Nested struct data 'user_password' not masked in: %s", result)
|
||
}
|
||
if strings.Contains(result, "user_token") {
|
||
t.Errorf("Nested struct data 'user_token' not masked in: %s", result)
|
||
}
|
||
if !strings.Contains(result, "data_safe") {
|
||
t.Errorf("Safe data 'data_safe' should be present in: %s", result)
|
||
}
|
||
}
|
||
|