Hulk did something

[bytom/vapor.git] / vendor / github.com / spf13 / jwalterweatherman / notepad_test.go

diff --git a/vendor/github.com/spf13/jwalterweatherman/notepad_test.go b/vendor/github.com/spf13/jwalterweatherman/notepad_test.go
new file mode 100644 (file)
index 0000000..69ad6f8
--- /dev/null
+++ b/vendor/github.com/spf13/jwalterweatherman/notepad_test.go
@@ -0,0 +1,50 @@
+// 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"
+       "testing"
+
+       "github.com/stretchr/testify/require"
+)
+
+func TestNotepad(t *testing.T) {
+       var logHandle, outHandle bytes.Buffer
+
+       n := NewNotepad(LevelCritical, LevelError, &outHandle, &logHandle, "TestNotePad", 0)
+
+       require.Equal(t, LevelCritical, n.GetStdoutThreshold())
+       require.Equal(t, LevelError, n.GetLogThreshold())
+
+       n.DEBUG.Println("Some debug")
+       n.ERROR.Println("Some error")
+       n.CRITICAL.Println("Some critical error")
+
+       require.Contains(t, logHandle.String(), "[TestNotePad] ERROR Some error")
+       require.NotContains(t, logHandle.String(), "Some debug")
+       require.NotContains(t, outHandle.String(), "Some error")
+       require.Contains(t, outHandle.String(), "Some critical error")
+
+       require.Equal(t, n.LogCountForLevel(LevelError), uint64(1))
+       require.Equal(t, n.LogCountForLevel(LevelDebug), uint64(1))
+       require.Equal(t, n.LogCountForLevel(LevelTrace), uint64(0))
+}
+
+func TestThresholdString(t *testing.T) {
+       require.Equal(t, LevelError.String(), "ERROR")
+       require.Equal(t, LevelTrace.String(), "TRACE")
+}
+
+func BenchmarkLogPrintOnlyToCounter(b *testing.B) {
+       var logHandle, outHandle bytes.Buffer
+       n := NewNotepad(LevelCritical, LevelCritical, &outHandle, &logHandle, "TestNotePad", 0)
+
+       b.ResetTimer()
+       for i := 0; i < b.N; i++ {
+               n.INFO.Print("Test")
+       }
+}