84 lines
1.5 KiB
Go
84 lines
1.5 KiB
Go
package logawl
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var logger = New()
|
|
|
|
func TestLogawl(t *testing.T) {
|
|
|
|
assert.Equal(t, Level(2), logger.Level) //cast 2 (int) to 2 (level)
|
|
|
|
//Validate setting and getting levels from memory works
|
|
for i := range AllLevels {
|
|
logger.SetLevel(Level(i))
|
|
assert.Equal(t, Level(i), logger.GetLevel())
|
|
}
|
|
|
|
}
|
|
|
|
func TestUnmarshalLevels(t *testing.T) {
|
|
m := make(map[int]string)
|
|
var err error
|
|
//Fill map with unmarshalled level info
|
|
for i := range AllLevels {
|
|
m[i], err = logger.UnMarshalLevel(Level(i))
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
//iterate over map and assert equal
|
|
for i := range AllLevels {
|
|
lv, err := logger.UnMarshalLevel(Level(i))
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, m[i], lv)
|
|
}
|
|
|
|
lv, err := logger.UnMarshalLevel(Level(9001))
|
|
assert.NotNil(t, err)
|
|
assert.Equal(t, "", lv)
|
|
assert.ErrorContains(t, err, "invalid log level choice")
|
|
}
|
|
|
|
func TestLogger(t *testing.T) {
|
|
|
|
for i := range AllLevels {
|
|
// only test non-exiting log levels
|
|
switch i {
|
|
case 1:
|
|
fn := func() {
|
|
logger.Info("")
|
|
}
|
|
var buffer bytes.Buffer
|
|
logger.Out = &buffer
|
|
fn()
|
|
case 2:
|
|
fn := func() {
|
|
logger.Info("Test")
|
|
}
|
|
var buffer bytes.Buffer
|
|
logger.Out = &buffer
|
|
fn()
|
|
case 3:
|
|
fn := func() {
|
|
logger.Debug("Test")
|
|
}
|
|
var buffer bytes.Buffer
|
|
logger.Out = &buffer
|
|
fn()
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func TestFmt(t *testing.T) {
|
|
ti := time.Now()
|
|
test := []byte("test")
|
|
assert.NotNil(t, logger.formatHeader(&test, ti, 0, Level(9001))) //make sure error is error
|
|
|
|
}
|