hoteas
332c2414bc
- WebConnectLog 访问日志新增 ua 字段（User-Agent 按 rune 截断 200 字符），便于按设备归因 - SetSeqWriter 挂 Seq 后业务控制台由 Warn+ 放宽为 Info+，Debug/SQL 仍只进 Seq 与文件 - 新增 Logger.SetConsoleMinLevel，访问日志控制台单独保持 Warn+ Co-authored-by: Cursor <cursoragent@cursor.com>
88 lines
2.9 KiB
Go
88 lines
2.9 KiB
Go
package log
|
||
|
||
import (
|
||
"bytes"
|
||
"testing"
|
||
|
||
"github.com/rs/zerolog"
|
||
)
|
||
|
||
// TestSetSeqWriter_ConsoleMinLevelBecomesInfo 验证挂 Seq 后控制台门槛降到 Info（而非 Warn）
|
||
func TestSetSeqWriter_ConsoleMinLevelBecomesInfo(t *testing.T) {
|
||
l := NewLogger(1, "", 0)
|
||
if got := zerolog.Level(l.console.minLevel.Load()); got != zerolog.DebugLevel {
|
||
t.Fatalf("初始 console minLevel=%v want Debug", got)
|
||
}
|
||
// seqUrl 不必可达：SeqWriter 异步 channel 推送，Write 不阻塞
|
||
l.SetSeqWriter("http://127.0.0.1:1", "", "test:level")
|
||
defer l.CloseSeq()
|
||
if got := zerolog.Level(l.console.minLevel.Load()); got != zerolog.InfoLevel {
|
||
t.Fatalf("挂 Seq 后 console minLevel=%v want Info", got)
|
||
}
|
||
}
|
||
|
||
// TestSetConsoleMinLevel_OverridesToWarn 验证导出方法可将 console 门槛单独收紧到 Warn，
|
||
// 用于访问日志等高噪声 Logger 在挂 Seq 后仍保持 Warn+。
|
||
func TestSetConsoleMinLevel_OverridesToWarn(t *testing.T) {
|
||
l := NewLogger(1, "", 0)
|
||
l.SetSeqWriter("http://127.0.0.1:1", "", "test:override")
|
||
defer l.CloseSeq()
|
||
|
||
l.SetConsoleMinLevel(zerolog.WarnLevel)
|
||
if got := zerolog.Level(l.console.minLevel.Load()); got != zerolog.WarnLevel {
|
||
t.Fatalf("SetConsoleMinLevel 后 minLevel=%v want Warn", got)
|
||
}
|
||
|
||
var buf bytes.Buffer
|
||
l.console.inner = &buf
|
||
if _, err := l.console.WriteLevel(zerolog.InfoLevel, []byte("info-line\n")); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if buf.Len() != 0 {
|
||
t.Fatalf("Info 应被 Warn 门槛丢弃，实际写入: %q", buf.String())
|
||
}
|
||
if _, err := l.console.WriteLevel(zerolog.WarnLevel, []byte("warn-line\n")); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if !bytes.Contains(buf.Bytes(), []byte("warn-line")) {
|
||
t.Fatalf("Warn 应放行，实际: %q", buf.String())
|
||
}
|
||
}
|
||
|
||
// TestLevelFilterWriter_WriteLevel_DropAndPass 直接验证 levelFilterWriter.WriteLevel
|
||
// 对低于门槛的级别丢弃（但仍返回成功），达到门槛的放行。
|
||
func TestLevelFilterWriter_WriteLevel_DropAndPass(t *testing.T) {
|
||
var buf bytes.Buffer
|
||
w := newLevelFilterWriter(&buf, zerolog.WarnLevel)
|
||
|
||
n, err := w.WriteLevel(zerolog.DebugLevel, []byte("debug-msg"))
|
||
if err != nil || n != len("debug-msg") {
|
||
t.Fatalf("Debug 丢弃返回值异常: n=%d err=%v", n, err)
|
||
}
|
||
if buf.Len() != 0 {
|
||
t.Fatalf("Debug 不应写入 inner: %q", buf.String())
|
||
}
|
||
|
||
n, err = w.WriteLevel(zerolog.InfoLevel, []byte("info-msg"))
|
||
if err != nil || n != len("info-msg") {
|
||
t.Fatalf("Info 丢弃返回值异常: n=%d err=%v", n, err)
|
||
}
|
||
if buf.Len() != 0 {
|
||
t.Fatalf("Info 不应写入 inner: %q", buf.String())
|
||
}
|
||
|
||
if _, err := w.WriteLevel(zerolog.WarnLevel, []byte("warn-msg")); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if !bytes.Contains(buf.Bytes(), []byte("warn-msg")) {
|
||
t.Fatalf("Warn 应放行: %q", buf.String())
|
||
}
|
||
|
||
if _, err := w.WriteLevel(zerolog.ErrorLevel, []byte("error-msg")); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if !bytes.Contains(buf.Bytes(), []byte("error-msg")) {
|
||
t.Fatalf("Error 应放行: %q", buf.String())
|
||
}
|
||
}
|