package logx import ( "encoding/json" "fmt" "io" "io/ioutil" "log" "os" "runtime" "strings" "sync" "sync/atomic" "testing" "time" "github.com/stretchr/testify/assert" ) var ( s = []byte("Sending #11 notification (id: 1451875113812010473) in #1 connection") pool = make(chan []byte, 1) ) type mockWriter struct { lock sync.Mutex builder strings.Builder } func (mw *mockWriter) Write(data []byte) (int, error) { mw.lock.Lock() defer mw.lock.Unlock() return mw.builder.Write(data) } func (mw *mockWriter) Close() error { return nil } func (mw *mockWriter) Contains(text string) bool { mw.lock.Lock() defer mw.lock.Unlock() return strings.Contains(mw.builder.String(), text) } func (mw *mockWriter) Reset() { mw.lock.Lock() defer mw.lock.Unlock() mw.builder.Reset() } func (mw *mockWriter) String() string { mw.lock.Lock() defer mw.lock.Unlock() return mw.builder.String() } func TestFileLineFileMode(t *testing.T) { writer := new(mockWriter) errorLog = writer atomic.StoreUint32(&initialized, 1) file, line := getFileLine() Error("anything") assert.True(t, writer.Contains(fmt.Sprintf("%s:%d", file, line+1))) writer.Reset() file, line = getFileLine() Errorf("anything %s", "format") assert.True(t, writer.Contains(fmt.Sprintf("%s:%d", file, line+1))) } func TestFileLineConsoleMode(t *testing.T) { writer := new(mockWriter) writeConsole = true errorLog = newLogWriter(log.New(writer, "[ERROR] ", flags)) atomic.StoreUint32(&initialized, 1) file, line := getFileLine() Error("anything") assert.True(t, writer.Contains(fmt.Sprintf("%s:%d", file, line+1))) writer.Reset() file, line = getFileLine() Errorf("anything %s", "format") assert.True(t, writer.Contains(fmt.Sprintf("%s:%d", file, line+1))) } func TestStructedLogAlert(t *testing.T) { doTestStructedLog(t, levelAlert, func(writer io.WriteCloser) { errorLog = writer }, func(v ...interface{}) { Alert(fmt.Sprint(v...)) }) } func TestStructedLogInfo(t *testing.T) { doTestStructedLog(t, levelInfo, func(writer io.WriteCloser) { infoLog = writer }, func(v ...interface{}) { Info(v...) }) } func TestStructedLogSlow(t *testing.T) { doTestStructedLog(t, levelSlow, func(writer io.WriteCloser) { slowLog = writer }, func(v ...interface{}) { Slow(v...) }) } func TestStructedLogSlowf(t *testing.T) { doTestStructedLog(t, levelSlow, func(writer io.WriteCloser) { slowLog = writer }, func(v ...interface{}) { Slowf(fmt.Sprint(v...)) }) } func TestStructedLogStat(t *testing.T) { doTestStructedLog(t, levelStat, func(writer io.WriteCloser) { statLog = writer }, func(v ...interface{}) { Stat(v...) }) } func TestStructedLogStatf(t *testing.T) { doTestStructedLog(t, levelStat, func(writer io.WriteCloser) { statLog = writer }, func(v ...interface{}) { Statf(fmt.Sprint(v...)) }) } func TestStructedLogSevere(t *testing.T) { doTestStructedLog(t, levelSevere, func(writer io.WriteCloser) { severeLog = writer }, func(v ...interface{}) { Severe(v...) }) } func TestStructedLogSeveref(t *testing.T) { doTestStructedLog(t, levelSevere, func(writer io.WriteCloser) { severeLog = writer }, func(v ...interface{}) { Severef(fmt.Sprint(v...)) }) } func TestStructedLogWithDuration(t *testing.T) { const message = "hello there" writer := new(mockWriter) infoLog = writer atomic.StoreUint32(&initialized, 1) WithDuration(time.Second).Info(message) var entry logEntry if err := json.Unmarshal([]byte(writer.builder.String()), &entry); err != nil { t.Error(err) } assert.Equal(t, levelInfo, entry.Level) assert.Equal(t, message, entry.Content) assert.Equal(t, "1000.0ms", entry.Duration) } func TestSetLevel(t *testing.T) { SetLevel(ErrorLevel) const message = "hello there" writer := new(mockWriter) infoLog = writer atomic.StoreUint32(&initialized, 1) Info(message) assert.Equal(t, 0, writer.builder.Len()) } func TestSetLevelTwiceWithMode(t *testing.T) { testModes := []string{ "mode", "console", "volumn", } for _, mode := range testModes { testSetLevelTwiceWithMode(t, mode) } } func TestSetLevelWithDuration(t *testing.T) { SetLevel(ErrorLevel) const message = "hello there" writer := new(mockWriter) infoLog = writer atomic.StoreUint32(&initialized, 1) WithDuration(time.Second).Info(message) assert.Equal(t, 0, writer.builder.Len()) } func TestMustNil(t *testing.T) { Must(nil) } func TestSetup(t *testing.T) { MustSetup(LogConf{ ServiceName: "any", Mode: "console", }) MustSetup(LogConf{ ServiceName: "any", Mode: "file", Path: os.TempDir(), }) MustSetup(LogConf{ ServiceName: "any", Mode: "volume", Path: os.TempDir(), }) assert.NotNil(t, setupWithVolume(LogConf{})) assert.NotNil(t, setupWithFiles(LogConf{})) assert.Nil(t, setupWithFiles(LogConf{ ServiceName: "any", Path: os.TempDir(), Compress: true, KeepDays: 1, })) setupLogLevel(LogConf{ Level: levelInfo, }) setupLogLevel(LogConf{ Level: levelError, }) setupLogLevel(LogConf{ Level: levelSevere, }) _, err := createOutput("") assert.NotNil(t, err) Disable() } func TestDisable(t *testing.T) { Disable() var opt logOptions WithKeepDays(1)(&opt) WithGzip()(&opt) assert.Nil(t, Close()) writeConsole = false assert.Nil(t, Close()) } func TestWithGzip(t *testing.T) { fn := WithGzip() var opt logOptions fn(&opt) assert.True(t, opt.gzipEnabled) } func TestWithKeepDays(t *testing.T) { fn := WithKeepDays(1) var opt logOptions fn(&opt) assert.Equal(t, 1, opt.keepDays) } func BenchmarkCopyByteSliceAppend(b *testing.B) { for i := 0; i < b.N; i++ { var buf []byte buf = append(buf, getTimestamp()...) buf = append(buf, ' ') buf = append(buf, s...) _ = buf } } func BenchmarkCopyByteSliceAllocExactly(b *testing.B) { for i := 0; i < b.N; i++ { now := []byte(getTimestamp()) buf := make([]byte, len(now)+1+len(s)) n := copy(buf, now) buf[n] = ' ' copy(buf[n+1:], s) } } func BenchmarkCopyByteSlice(b *testing.B) { var buf []byte for i := 0; i < b.N; i++ { buf = make([]byte, len(s)) copy(buf, s) } fmt.Fprint(ioutil.Discard, buf) } func BenchmarkCopyOnWriteByteSlice(b *testing.B) { var buf []byte for i := 0; i < b.N; i++ { size := len(s) buf = s[:size:size] } fmt.Fprint(ioutil.Discard, buf) } func BenchmarkCacheByteSlice(b *testing.B) { for i := 0; i < b.N; i++ { dup := fetch() copy(dup, s) put(dup) } } func BenchmarkLogs(b *testing.B) { b.ReportAllocs() log.SetOutput(ioutil.Discard) for i := 0; i < b.N; i++ { Info(i) } } func fetch() []byte { select { case b := <-pool: return b default: } return make([]byte, 4096) } func getFileLine() (string, int) { _, file, line, _ := runtime.Caller(1) short := file for i := len(file) - 1; i > 0; i-- { if file[i] == '/' { short = file[i+1:] break } } return short, line } func put(b []byte) { select { case pool <- b: default: } } func doTestStructedLog(t *testing.T, level string, setup func(writer io.WriteCloser), write func(...interface{})) { const message = "hello there" writer := new(mockWriter) setup(writer) atomic.StoreUint32(&initialized, 1) write(message) var entry logEntry if err := json.Unmarshal([]byte(writer.builder.String()), &entry); err != nil { t.Error(err) } assert.Equal(t, level, entry.Level) assert.True(t, strings.Contains(entry.Content, message)) } func testSetLevelTwiceWithMode(t *testing.T, mode string) { SetUp(LogConf{ Mode: mode, Level: "error", Path: "/dev/null", }) SetUp(LogConf{ Mode: mode, Level: "info", Path: "/dev/null", }) const message = "hello there" writer := new(mockWriter) infoLog = writer atomic.StoreUint32(&initialized, 1) Info(message) assert.Equal(t, 0, writer.builder.Len()) Infof(message) assert.Equal(t, 0, writer.builder.Len()) ErrorStack(message) assert.Equal(t, 0, writer.builder.Len()) ErrorStackf(message) assert.Equal(t, 0, writer.builder.Len()) }