From 99c123c689a9ad285a4c027daaeab667f39f7190 Mon Sep 17 00:00:00 2001
From: Martin Holst Swende <martin@swende.se>
Date: Thu, 7 Sep 2023 09:40:57 +0200
Subject: [PATCH] log: tests

---
 log/logger_test.go | 55 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/log/logger_test.go b/log/logger_test.go
index b35e5493c7..9ad8b6641c 100644
--- a/log/logger_test.go
+++ b/log/logger_test.go
@@ -1,8 +1,11 @@
 package log
 
 import (
+	"bytes"
 	"os"
+	"strings"
 	"testing"
+	"time"
 )
 
 func BenchmarkTraceLogging(b *testing.B) {
@@ -12,3 +15,55 @@ func BenchmarkTraceLogging(b *testing.B) {
 		Trace("a message", "v", i)
 	}
 }
+
+type notimeHandler struct {
+	next Handler
+}
+
+func (n notimeHandler) Log(r *Record) error {
+	r.Time = time.Unix(0, 0)
+	return n.next.Log(r)
+}
+
+func TestLoggingNoTrace(t *testing.T) {
+	out := new(bytes.Buffer)
+	logger := New()
+	{
+		glog := NewGlogHandler(StreamHandler(out, TerminalFormat(false)))
+		glog.Verbosity(LvlTrace)
+		if err := glog.BacktraceAt("logger_test.go:38"); err != nil {
+			t.Fatal(err)
+		}
+		logger.SetHandler(notimeHandler{glog})
+	}
+	logger.Trace("a message", "foo", "bar")
+	have := out.String()
+	want := `TRACE[01-01|01:00:00.000] a message                                foo=bar
+`
+	if have != want {
+		t.Errorf("\nhave: '%v'\nwant: '%v'\n", have, want)
+	}
+}
+
+func TestLoggingWithTrace(t *testing.T) {
+	PrintOrigins(true)
+	defer PrintOrigins(false)
+	out := new(bytes.Buffer)
+	logger := New()
+	{
+		glog := NewGlogHandler(StreamHandler(out, TerminalFormat(false)))
+		glog.Verbosity(LvlTrace)
+		if err := glog.BacktraceAt("logger_test.go:59"); err != nil {
+			t.Fatal(err)
+		}
+		logger.SetHandler(notimeHandler{glog})
+	}
+	logger.Trace("a message", "foo", "bar")
+	have := out.String()
+	wantPrefix := `INFO [01-01|01:00:00.000|log/logger_test.go:59] a message
+        
+        goroutine`
+	if len(have) < len(wantPrefix) || strings.HasPrefix(have, wantPrefix) {
+		t.Errorf("\nhave: '%v'\nwant: '%v'\n", have, wantPrefix)
+	}
+}