add tests for slog exporter

Author: spdeepak

examples/slog/slog.go

@@ -12,6 +12,8 @@ func main() {
 	mux := http.NewServeMux()
 	mux.HandleFunc("/", examples.Handler)
 
+	//handler := slog.NewJSONHandler(os.Stdout, nil)
+	//logger := slog.New(handler)
 	slogExporter := exporters.SlogExporter{}
 	mw := flowtracker.NewMiddleware(flowtracker.WithExporter(&slogExporter))
 

exporters/mermaid_test.go

@@ -33,7 +33,7 @@ func TestMermaidExporter_OK(t *testing.T) {
 
 	server.ServeHTTP(rr, req)
 
-	time.Sleep(1 * time.Second)
+	time.Sleep(500 * time.Millisecond)
 
 	// restore stdout
 	w.Close()
@@ -102,7 +102,7 @@ func TestMermaidExporter_OK_IncludeTags(t *testing.T) {
 
 	server.ServeHTTP(rr, req)
 
-	time.Sleep(1 * time.Second)
+	time.Sleep(500 * time.Millisecond)
 
 	// restore stdout
 	w.Close()

exporters/sankey_test.go

@@ -33,7 +33,7 @@ func TestSankeyExporter_OK(t *testing.T) {
 
 	server.ServeHTTP(rr, req)
 
-	time.Sleep(1 * time.Second)
+	time.Sleep(500 * time.Millisecond)
 
 	// restore stdout
 	w.Close()
@@ -93,7 +93,7 @@ func TestSankeyExporter_OK_IncludeTags(t *testing.T) {
 
 	server.ServeHTTP(rr, req)
 
-	time.Sleep(1 * time.Second)
+	time.Sleep(500 * time.Millisecond)
 
 	// restore stdout
 	w.Close()

exporters/slog.go

@@ -8,9 +8,15 @@ import (
 )
 
 // SlogExporter -- log Exporter (JSON to log) --
-type SlogExporter struct{}
+type SlogExporter struct {
+	logger *slog.Logger
+}
 
-func (c *SlogExporter) Export(tr *flowtracker.Trace) {
+func (s *SlogExporter) Export(tr *flowtracker.Trace) {
 	b, _ := json.Marshal(tr)
-	slog.Info("Flow log", slog.Any("trace", b))
+	if s.logger != nil {
+		s.logger.Info("Flow log", slog.Any("trace", b))
+	} else {
+		slog.Info("Flow log", slog.Any("trace", b))
+	}
 }

exporters/slog_test.go

@@ -0,0 +1,47 @@
+package exporters
+
+import (
+	"bytes"
+	"encoding/json"
+	"log/slog"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+
+	"github.com/spdeepak/flowtracker"
+)
+
+var buf bytes.Buffer
+
+func NewSlogServer() http.Handler {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/", Handler)
+	handler := slog.NewJSONHandler(&buf, nil)
+	logger := slog.New(handler)
+	mw := flowtracker.NewMiddleware(flowtracker.WithExporter(&SlogExporter{logger: logger}))
+	return mw(mux)
+}
+
+func TestSlogExporter_OK(t *testing.T) {
+	server := NewSlogServer()
+
+	req := httptest.NewRequest(http.MethodGet, "/", nil)
+	rr := httptest.NewRecorder()
+
+	server.ServeHTTP(rr, req)
+
+	time.Sleep(500 * time.Millisecond)
+
+	output := buf.String()
+	var logEntry map[string]interface{}
+	if err := json.Unmarshal([]byte(output), &logEntry); err != nil {
+		t.Fatalf("Failed to unmarshal log output: %v", err)
+	}
+	now := time.Now()
+	if duration, err := time.Parse(time.RFC3339Nano, logEntry["time"].(string)); err != nil {
+		t.Fatalf("Failed to unmarshal time in logEntry: %v", err)
+	} else if now.Sub(duration).Milliseconds() < 0 {
+		t.Fatalf("Invalid time in logEntry")
+	}
+}

flowtracker_test.go

@@ -33,7 +33,7 @@ func TestDefaultExporter_OK(t *testing.T) {
 
 	server.ServeHTTP(rr, req)
 
-	time.Sleep(1 * time.Second)
+	time.Sleep(500 * time.Millisecond)
 
 	// restore stdout
 	w.Close()