103 lines
3.0 KiB
Go
103 lines
3.0 KiB
Go
|
// Copyright © 2016 Steve Francia <spf@spf13.com>.
|
||
|
|
//
|
||
|
|
// Use of this source code is governed by an MIT-style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package jwalterweatherman
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io/ioutil"
|
||
|
|
"sync"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestThresholds(t *testing.T) {
|
||
|
|
SetStdoutThreshold(LevelError)
|
||
|
|
require.Equal(t, StdoutThreshold(), LevelError)
|
||
|
|
SetLogThreshold(LevelCritical)
|
||
|
|
require.Equal(t, LogThreshold(), LevelCritical)
|
||
|
|
require.NotEqual(t, StdoutThreshold(), LevelCritical)
|
||
|
|
SetStdoutThreshold(LevelWarn)
|
||
|
|
require.Equal(t, StdoutThreshold(), LevelWarn)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDefaultLogging(t *testing.T) {
|
||
|
|
var outputBuf, logBuf bytes.Buffer
|
||
|
|
|
||
|
|
defaultNotepad.logHandle = &logBuf
|
||
|
|
defaultNotepad.outHandle = &outputBuf
|
||
|
|
|
||
|
|
SetLogThreshold(LevelWarn)
|
||
|
|
SetStdoutThreshold(LevelError)
|
||
|
|
|
||
|
|
FATAL.Println("fatal err")
|
||
|
|
CRITICAL.Println("critical err")
|
||
|
|
ERROR.Println("an error")
|
||
|
|
WARN.Println("a warning")
|
||
|
|
INFO.Println("information")
|
||
|
|
DEBUG.Println("debugging info")
|
||
|
|
TRACE.Println("trace")
|
||
|
|
|
||
|
|
require.Contains(t, logBuf.String(), "fatal err")
|
||
|
|
require.Contains(t, logBuf.String(), "critical err")
|
||
|
|
require.Contains(t, logBuf.String(), "an error")
|
||
|
|
require.Contains(t, logBuf.String(), "a warning")
|
||
|
|
require.NotContains(t, logBuf.String(), "information")
|
||
|
|
require.NotContains(t, logBuf.String(), "debugging info")
|
||
|
|
require.NotContains(t, logBuf.String(), "trace")
|
||
|
|
|
||
|
|
require.Contains(t, outputBuf.String(), "fatal err")
|
||
|
|
require.Contains(t, outputBuf.String(), "critical err")
|
||
|
|
require.Contains(t, outputBuf.String(), "an error")
|
||
|
|
require.NotContains(t, outputBuf.String(), "a warning")
|
||
|
|
require.NotContains(t, outputBuf.String(), "information")
|
||
|
|
require.NotContains(t, outputBuf.String(), "debugging info")
|
||
|
|
require.NotContains(t, outputBuf.String(), "trace")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLogCounter(t *testing.T) {
|
||
|
|
defaultNotepad.logHandle = ioutil.Discard
|
||
|
|
defaultNotepad.outHandle = ioutil.Discard
|
||
|
|
|
||
|
|
SetLogThreshold(LevelTrace)
|
||
|
|
SetStdoutThreshold(LevelTrace)
|
||
|
|
|
||
|
|
FATAL.Println("fatal err")
|
||
|
|
CRITICAL.Println("critical err")
|
||
|
|
WARN.Println("a warning")
|
||
|
|
WARN.Println("another warning")
|
||
|
|
INFO.Println("information")
|
||
|
|
DEBUG.Println("debugging info")
|
||
|
|
TRACE.Println("trace")
|
||
|
|
|
||
|
|
wg := &sync.WaitGroup{}
|
||
|
|
|
||
|
|
for i := 0; i < 10; i++ {
|
||
|
|
wg.Add(1)
|
||
|
|
go func() {
|
||
|
|
defer wg.Done()
|
||
|
|
for j := 0; j < 10; j++ {
|
||
|
|
ERROR.Println("error", j)
|
||
|
|
// check for data races
|
||
|
|
require.True(t, LogCountForLevel(LevelError) > uint64(j))
|
||
|
|
require.True(t, LogCountForLevelsGreaterThanorEqualTo(LevelError) > uint64(j))
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
wg.Wait()
|
||
|
|
|
||
|
|
require.Equal(t, uint64(1), LogCountForLevel(LevelFatal))
|
||
|
|
require.Equal(t, uint64(1), LogCountForLevel(LevelCritical))
|
||
|
|
require.Equal(t, uint64(2), LogCountForLevel(LevelWarn))
|
||
|
|
require.Equal(t, uint64(1), LogCountForLevel(LevelInfo))
|
||
|
|
require.Equal(t, uint64(1), LogCountForLevel(LevelDebug))
|
||
|
|
require.Equal(t, uint64(1), LogCountForLevel(LevelTrace))
|
||
|
|
require.Equal(t, uint64(100), LogCountForLevel(LevelError))
|
||
|
|
require.Equal(t, uint64(102), LogCountForLevelsGreaterThanorEqualTo(LevelError))
|
||
|
|
}
|