plugeth/ethlog/loggers_test.go

109 lines
2.7 KiB
Go
Raw Normal View History

package ethlog
import (
2014-06-26 17:45:57 +00:00
"fmt"
"io/ioutil"
"os"
"testing"
)
type TestLogSystem struct {
2014-06-26 17:45:57 +00:00
Output string
level LogLevel
}
func (t *TestLogSystem) Println(v ...interface{}) {
2014-06-26 17:45:57 +00:00
t.Output += fmt.Sprintln(v...)
}
func (t *TestLogSystem) Printf(format string, v ...interface{}) {
2014-06-26 17:45:57 +00:00
t.Output += fmt.Sprintf(format, v...)
}
func (t *TestLogSystem) SetLogLevel(i LogLevel) {
2014-06-26 17:45:57 +00:00
t.level = i
}
func (t *TestLogSystem) GetLogLevel() LogLevel {
2014-06-26 17:45:57 +00:00
return t.level
}
func TestLoggerPrintln(t *testing.T) {
2014-06-26 17:45:57 +00:00
logger := NewLogger("TEST")
testLogSystem := &TestLogSystem{level: WarnLevel}
AddLogSystem(testLogSystem)
logger.Errorln("error")
logger.Warnln("warn")
logger.Infoln("info")
logger.Debugln("debug")
Flush()
Reset()
2014-06-26 17:45:57 +00:00
output := testLogSystem.Output
if output != "[TEST] error\n[TEST] warn\n" {
t.Error("Expected logger output '[TEST] error\\n[TEST] warn\\n', got ", testLogSystem.Output)
2014-06-26 17:45:57 +00:00
}
}
func TestLoggerPrintf(t *testing.T) {
2014-06-26 17:45:57 +00:00
logger := NewLogger("TEST")
testLogSystem := &TestLogSystem{level: WarnLevel}
AddLogSystem(testLogSystem)
logger.Errorf("error to %v\n", *testLogSystem)
logger.Warnf("warn")
logger.Infof("info")
logger.Debugf("debug")
Flush()
Reset()
2014-06-26 17:45:57 +00:00
output := testLogSystem.Output
if output != "[TEST] error to { 2}\n[TEST] warn" {
t.Error("Expected logger output '[TEST] error to { 2}\\n[TEST] warn', got ", testLogSystem.Output)
2014-06-26 17:45:57 +00:00
}
}
func TestMultipleLogSystems(t *testing.T) {
2014-06-26 17:45:57 +00:00
logger := NewLogger("TEST")
testLogSystem0 := &TestLogSystem{level: ErrorLevel}
testLogSystem1 := &TestLogSystem{level: WarnLevel}
AddLogSystem(testLogSystem0)
AddLogSystem(testLogSystem1)
logger.Errorln("error")
logger.Warnln("warn")
Flush()
Reset()
2014-06-26 17:45:57 +00:00
output0 := testLogSystem0.Output
output1 := testLogSystem1.Output
if output0 != "[TEST] error\n" {
t.Error("Expected logger 0 output '[TEST] error\\n', got ", testLogSystem0.Output)
2014-06-26 17:45:57 +00:00
}
if output1 != "[TEST] error\n[TEST] warn\n" {
t.Error("Expected logger 1 output '[TEST] error\\n[TEST] warn\\n', got ", testLogSystem1.Output)
2014-06-26 17:45:57 +00:00
}
}
func TestFileLogSystem(t *testing.T) {
2014-06-26 17:45:57 +00:00
logger := NewLogger("TEST")
filename := "test.log"
file, _ := os.OpenFile(filename, os.O_RDWR|os.O_CREATE, os.ModePerm)
testLogSystem := NewStdLogSystem(file, 0, WarnLevel)
AddLogSystem(testLogSystem)
logger.Errorf("error to %s\n", filename)
logger.Warnln("warn")
Flush()
Reset()
2014-06-26 17:45:57 +00:00
contents, _ := ioutil.ReadFile(filename)
output := string(contents)
if output != "[TEST] error to test.log\n[TEST] warn\n" {
t.Error("Expected contents of file 'test.log': '[TEST] error to test.log\\n[TEST] warn\\n', got ", output)
2014-06-26 17:45:57 +00:00
} else {
os.Remove(filename)
}
}
func TestNoLogSystem(t *testing.T) {
2014-06-26 17:45:57 +00:00
logger := NewLogger("TEST")
logger.Warnln("warn")
fmt.Println("1")
2014-06-26 17:45:57 +00:00
Flush()
Reset()
}