restructured logger from budigt

console_stream_test.go

@@ -0,0 +1,52 @@
+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())
+	}
+}