123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276 |
- // Copyright 2019 The Gitea Authors. All rights reserved.
- // Use of this source code is governed by a MIT-style
- // license that can be found in the LICENSE file.
-
- package log
-
- import (
- "fmt"
- "strings"
- "testing"
- "time"
-
- "github.com/stretchr/testify/assert"
- )
-
- type CallbackWriteCloser struct {
- callback func([]byte, bool)
- }
-
- func (c CallbackWriteCloser) Write(p []byte) (int, error) {
- c.callback(p, false)
- return len(p), nil
- }
-
- func (c CallbackWriteCloser) Close() error {
- c.callback(nil, true)
- return nil
- }
-
- func TestBaseLogger(t *testing.T) {
- var written []byte
- var closed bool
-
- c := CallbackWriteCloser{
- callback: func(p []byte, close bool) {
- written = p
- closed = close
- },
- }
- prefix := "TestPrefix "
- b := WriterLogger{
- out: c,
- Level: INFO,
- Flags: LstdFlags | LUTC,
- Prefix: prefix,
- }
- location, _ := time.LoadLocation("EST")
-
- date := time.Date(2019, time.January, 13, 22, 3, 30, 15, location)
-
- dateString := date.UTC().Format("2006/01/02 15:04:05")
-
- event := Event{
- level: INFO,
- msg: "TEST MSG",
- caller: "CALLER",
- filename: "FULL/FILENAME",
- line: 1,
- time: date,
- }
-
- assert.Equal(t, INFO, b.GetLevel())
-
- expected := fmt.Sprintf("%s%s %s:%d:%s [%c] %s\n", prefix, dateString, event.filename, event.line, event.caller, strings.ToUpper(event.level.String())[0], event.msg)
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- event.level = DEBUG
- expected = ""
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
-
- event.level = TRACE
- expected = ""
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
-
- event.level = WARN
- expected = fmt.Sprintf("%s%s %s:%d:%s [%c] %s\n", prefix, dateString, event.filename, event.line, event.caller, strings.ToUpper(event.level.String())[0], event.msg)
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- event.level = ERROR
- expected = fmt.Sprintf("%s%s %s:%d:%s [%c] %s\n", prefix, dateString, event.filename, event.line, event.caller, strings.ToUpper(event.level.String())[0], event.msg)
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- event.level = CRITICAL
- expected = fmt.Sprintf("%s%s %s:%d:%s [%c] %s\n", prefix, dateString, event.filename, event.line, event.caller, strings.ToUpper(event.level.String())[0], event.msg)
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- b.Close()
- assert.True(t, closed)
- }
-
- func TestBaseLoggerDated(t *testing.T) {
- var written []byte
- var closed bool
-
- c := CallbackWriteCloser{
- callback: func(p []byte, close bool) {
- written = p
- closed = close
- },
- }
- prefix := ""
- b := WriterLogger{
- out: c,
- Level: WARN,
- Flags: Ldate | Ltime | Lmicroseconds | Lshortfile | Llevel,
- Prefix: prefix,
- }
-
- location, _ := time.LoadLocation("EST")
-
- date := time.Date(2019, time.January, 13, 22, 3, 30, 115, location)
-
- dateString := date.Format("2006/01/02 15:04:05.000000")
-
- event := Event{
- level: WARN,
- msg: "TEST MESSAGE TEST\n",
- caller: "CALLER",
- filename: "FULL/FILENAME",
- line: 1,
- time: date,
- }
-
- assert.Equal(t, WARN, b.GetLevel())
-
- expected := fmt.Sprintf("%s%s %s:%d [%s] %s", prefix, dateString, "FILENAME", event.line, strings.ToUpper(event.level.String()), event.msg)
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- event.level = INFO
- expected = ""
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- event.level = ERROR
- expected = fmt.Sprintf("%s%s %s:%d [%s] %s", prefix, dateString, "FILENAME", event.line, strings.ToUpper(event.level.String()), event.msg)
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- event.level = DEBUG
- expected = ""
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- event.level = CRITICAL
- expected = fmt.Sprintf("%s%s %s:%d [%s] %s", prefix, dateString, "FILENAME", event.line, strings.ToUpper(event.level.String()), event.msg)
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- event.level = TRACE
- expected = ""
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- b.Close()
- assert.True(t, closed)
- }
-
- func TestBaseLoggerMultiLineNoFlagsRegexp(t *testing.T) {
- var written []byte
- var closed bool
-
- c := CallbackWriteCloser{
- callback: func(p []byte, close bool) {
- written = p
- closed = close
- },
- }
- prefix := ""
- b := WriterLogger{
- Level: DEBUG,
- StacktraceLevel: ERROR,
- Flags: -1,
- Prefix: prefix,
- Expression: "FILENAME",
- }
- b.NewWriterLogger(c)
-
- location, _ := time.LoadLocation("EST")
-
- date := time.Date(2019, time.January, 13, 22, 3, 30, 115, location)
-
- event := Event{
- level: DEBUG,
- msg: "TEST\nMESSAGE\nTEST",
- caller: "CALLER",
- filename: "FULL/FILENAME",
- line: 1,
- time: date,
- }
-
- assert.Equal(t, DEBUG, b.GetLevel())
-
- expected := "TEST\n\tMESSAGE\n\tTEST\n"
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- event.filename = "ELSEWHERE"
-
- b.LogEvent(&event)
- assert.Equal(t, "", string(written))
- assert.False(t, closed)
- written = written[:0]
-
- event.caller = "FILENAME"
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
-
- event = Event{
- level: DEBUG,
- msg: "TEST\nFILENAME\nTEST",
- caller: "CALLER",
- filename: "FULL/ELSEWHERE",
- line: 1,
- time: date,
- }
- expected = "TEST\n\tFILENAME\n\tTEST\n"
- b.LogEvent(&event)
- assert.Equal(t, expected, string(written))
- assert.False(t, closed)
- written = written[:0]
- }
-
- func TestBrokenRegexp(t *testing.T) {
- var closed bool
-
- c := CallbackWriteCloser{
- callback: func(p []byte, close bool) {
- closed = close
- },
- }
-
- b := WriterLogger{
- Level: DEBUG,
- StacktraceLevel: ERROR,
- Flags: -1,
- Prefix: prefix,
- Expression: "\\",
- }
- b.NewWriterLogger(c)
- assert.Empty(t, b.regexp)
- b.Close()
- assert.True(t, closed)
- }
|