| // 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)) |
| } |
