package logx import ( "encoding/json" "log" "strings" "sync/atomic" "testing" "github.com/stretchr/testify/assert" ) const testlog = "Stay hungry, stay foolish." func TestCollectSysLog(t *testing.T) { CollectSysLog() content := getContent(captureOutput(func() { log.Print(testlog) })) assert.True(t, strings.Contains(content, testlog)) } func TestRedirector(t *testing.T) { var r redirector content := getContent(captureOutput(func() { r.Write([]byte(testlog)) })) assert.Equal(t, testlog, content) } func captureOutput(f func()) string { atomic.StoreUint32(&initialized, 1) writer := new(mockWriter) infoLog = writer prevLevel := atomic.LoadUint32(&logLevel) SetLevel(InfoLevel) f() SetLevel(prevLevel) return writer.builder.String() } func getContent(jsonStr string) string { var entry logEntry json.Unmarshal([]byte(jsonStr), &entry) return entry.Content }