random/log/log_test.go

package log

import (
	"bytes"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestLog(t *testing.T) {
	l := New("test")
	b := new(bytes.Buffer)
	l.SetOutput(b)
	test(l, TRACE, t)
	assert.Contains(t, b.String(), "trace")
	assert.Contains(t, b.String(), "fatal")

	b.Reset()
	SetOutput(b)
	test(global, NOTICE, t)
	assert.NotContains(t, b.String(), "info")
	assert.Contains(t, b.String(), "notice")
	assert.Contains(t, b.String(), "fatal")
}

func test(l *Logger, v Level, t *testing.T) {
	l.SetLevel(v)
	l.Print("print")
	l.Println("println")
	l.Trace("trace")
	l.Debug("debug")
	l.Info("info")
	l.Notice("notice")
	l.Warn("warn")
	l.Error("error")
	l.Fatal("fatal")
}