52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
package splinter
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"testing"
|
|
)
|
|
|
|
func TestConsoleStream_JSON(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
logger := New(WithStream(NewConsoleStream(ConsoleJSON, LevelDebug, ConsoleWriter(&buf))))
|
|
logger.Info("hello", "key", "value")
|
|
|
|
var entry map[string]any
|
|
if err := json.Unmarshal(buf.Bytes(), &entry); err != nil {
|
|
t.Fatalf("parse JSON: %v\nbody: %s", err, buf.String())
|
|
}
|
|
if entry["msg"] != "hello" {
|
|
t.Errorf("expected msg=hello, got %v", entry["msg"])
|
|
}
|
|
if entry["key"] != "value" {
|
|
t.Errorf("expected key=value, got %v", entry["key"])
|
|
}
|
|
}
|
|
|
|
func TestConsoleStream_Pretty(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
logger := New(WithStream(NewConsoleStream(ConsolePretty, LevelDebug, ConsoleWriter(&buf))))
|
|
logger.Info("pretty test", "foo", "bar")
|
|
|
|
if buf.Len() == 0 {
|
|
t.Fatal("expected text output, got nothing")
|
|
}
|
|
if bytes.Contains(buf.Bytes(), []byte("{")) {
|
|
t.Errorf("expected text format, got JSON-like output: %s", buf.String())
|
|
}
|
|
if !bytes.Contains(buf.Bytes(), []byte("foo=bar")) {
|
|
t.Errorf("expected foo=bar in output: %s", buf.String())
|
|
}
|
|
}
|
|
|
|
func TestConsoleStream_LevelFilter(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
logger := New(WithStream(NewConsoleStream(ConsoleJSON, LevelError, ConsoleWriter(&buf))))
|
|
logger.Info("skipped")
|
|
logger.Error("kept")
|
|
|
|
lines := bytes.Split(bytes.TrimSpace(buf.Bytes()), []byte("\n"))
|
|
if len(lines) != 1 {
|
|
t.Fatalf("expected 1 line, got %d: %s", len(lines), buf.String())
|
|
}
|
|
}
|