92 lines
1.5 KiB
Go
92 lines
1.5 KiB
Go
package flog
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetOutput(t *testing.T) {
|
|
assert.Equal(t, os.Stdout, getOutput())
|
|
|
|
b := bytes.Buffer{}
|
|
Output = &b
|
|
assert.Equal(t, &b, getOutput())
|
|
|
|
Output = nil
|
|
}
|
|
|
|
func TestLogFormat(t *testing.T) {
|
|
b := bytes.Buffer{}
|
|
MinLevel, Output = LevelDebug, &b
|
|
|
|
Debug(`Testtt`)
|
|
assert.True(t, strings.HasPrefix(b.String(), `debug [`))
|
|
b.Reset()
|
|
|
|
Info(`Testtt`)
|
|
assert.True(t, strings.HasPrefix(b.String(), `info [`))
|
|
|
|
MinLevel, Output = LevelInfo, nil
|
|
}
|
|
|
|
func TestLogLevel(t *testing.T) {
|
|
b := bytes.Buffer{}
|
|
MinLevel, Output = LevelWarning, &b
|
|
|
|
Info(`Testtt`)
|
|
assert.True(t, b.Len() == 0)
|
|
b.Reset()
|
|
|
|
Warning(`Testtt`)
|
|
assert.False(t, b.Len() == 0)
|
|
b.Reset()
|
|
|
|
MinLevel = LevelError
|
|
|
|
Warning(`Testtt`)
|
|
assert.True(t, b.Len() == 0)
|
|
b.Reset()
|
|
|
|
MinLevel, Output = LevelInfo, nil
|
|
}
|
|
|
|
func TestStackTrace(t *testing.T) {
|
|
b := bytes.Buffer{}
|
|
Output = &b
|
|
|
|
Error(`Testtt`)
|
|
checkStackTrace(t, &b, `git.fuyu.moe/Fuyu/flog.TestStackTrace`)
|
|
|
|
b.Reset()
|
|
|
|
func() {
|
|
defer func() {
|
|
v := recover()
|
|
if v != nil {
|
|
Critical(v)
|
|
}
|
|
}()
|
|
panic(`Testtt`)
|
|
}()
|
|
checkStackTrace(t, &b, `git.fuyu.moe/Fuyu/flog.TestStackTrace.func1`)
|
|
|
|
Output = nil
|
|
}
|
|
|
|
func checkStackTrace(t *testing.T, b *bytes.Buffer, f string) {
|
|
rdr := bufio.NewReader(b)
|
|
_, _ = rdr.ReadBytes('\n')
|
|
|
|
data, err := rdr.ReadBytes('\n')
|
|
if err != nil {
|
|
t.Fatal(`Got error while reading:`, err)
|
|
}
|
|
|
|
parts := strings.Split(strings.TrimSpace(string(data)), ` `)
|
|
assert.Equal(t, f, parts[0])
|
|
}
|