Add slog attributes at pool, worker, and connection levels.

2026-04-23 17:58:40 -04:00
parent 91717457dd
commit 727dc18b57
17 changed files with 488 additions and 248 deletions
@@ -0,0 +1,84 @@
+package logging
+
+import (
+	"git.wisehodl.dev/jay/go-honeybee/honeybeetest"
+	// "github.com/stretchr/testify/assert"
+	"log/slog"
+	"testing"
+)
+
+// Helpers
+
+func log(level slog.Level, msg string, attrs map[string]any) honeybeetest.ExpectedLog {
+	return honeybeetest.ExpectedLog{Level: level, Msg: msg, Attrs: attrs}
+}
+
+// Tests
+
+func TestOutboundLogger(t *testing.T) {
+	const POOL_ID = "pool-1"
+	const PEER_ID = "wss://test"
+
+	handler := honeybeetest.NewMockSlogHandler()
+	poolLogger := NewOutboundPoolLogger(handler, POOL_ID)
+	workerLogger := NewOutboundWorkerLogger(handler, POOL_ID, PEER_ID)
+	connLogger := NewConnectionLogger(handler, POOL_ID, PEER_ID)
+
+	poolLogger.Info("test")
+	workerLogger.Info("test")
+	connLogger.Info("test")
+
+	honeybeetest.Eventually(t, func() bool {
+		return len(handler.GetRecords()) == 3
+	}, "expected a log record")
+
+	records := handler.GetRecords()
+
+	honeybeetest.AssertAttributePresent(t, records[0], KEY_MODULE, MODULE_NAME)
+	honeybeetest.AssertAttributePresent(t, records[0], KEY_COMPONENT, COMPONENT_OUTBOUND_POOL)
+	honeybeetest.AssertAttributePresent(t, records[0], KEY_POOL_ID, POOL_ID)
+
+	honeybeetest.AssertAttributePresent(t, records[1], KEY_MODULE, MODULE_NAME)
+	honeybeetest.AssertAttributePresent(t, records[1], KEY_COMPONENT, COMPONENT_OUTBOUND_WORKER)
+	honeybeetest.AssertAttributePresent(t, records[1], KEY_POOL_ID, POOL_ID)
+	honeybeetest.AssertAttributePresent(t, records[1], KEY_PEER_ID, PEER_ID)
+
+	honeybeetest.AssertAttributePresent(t, records[2], KEY_MODULE, MODULE_NAME)
+	honeybeetest.AssertAttributePresent(t, records[2], KEY_COMPONENT, COMPONENT_CONNECTION)
+	honeybeetest.AssertAttributePresent(t, records[2], KEY_POOL_ID, POOL_ID)
+	honeybeetest.AssertAttributePresent(t, records[2], KEY_PEER_ID, PEER_ID)
+}
+
+func TestInboundLogger(t *testing.T) {
+	const POOL_ID = "pool-1"
+	const PEER_ID = "peer-1"
+
+	handler := honeybeetest.NewMockSlogHandler()
+	poolLogger := NewInboundPoolLogger(handler, POOL_ID)
+	workerLogger := NewInboundWorkerLogger(handler, POOL_ID, PEER_ID)
+	connLogger := NewConnectionLogger(handler, POOL_ID, PEER_ID)
+
+	poolLogger.Info("test")
+	workerLogger.Info("test")
+	connLogger.Info("test")
+
+	honeybeetest.Eventually(t, func() bool {
+		return len(handler.GetRecords()) == 3
+	}, "expected a log record")
+
+	records := handler.GetRecords()
+
+	honeybeetest.AssertAttributePresent(t, records[0], KEY_MODULE, MODULE_NAME)
+	honeybeetest.AssertAttributePresent(t, records[0], KEY_COMPONENT, COMPONENT_INBOUND_POOL)
+	honeybeetest.AssertAttributePresent(t, records[0], KEY_POOL_ID, POOL_ID)
+
+	honeybeetest.AssertAttributePresent(t, records[1], KEY_MODULE, MODULE_NAME)
+	honeybeetest.AssertAttributePresent(t, records[1], KEY_COMPONENT, COMPONENT_INBOUND_WORKER)
+	honeybeetest.AssertAttributePresent(t, records[1], KEY_POOL_ID, POOL_ID)
+	honeybeetest.AssertAttributePresent(t, records[1], KEY_PEER_ID, PEER_ID)
+
+	honeybeetest.AssertAttributePresent(t, records[2], KEY_MODULE, MODULE_NAME)
+	honeybeetest.AssertAttributePresent(t, records[2], KEY_COMPONENT, COMPONENT_CONNECTION)
+	honeybeetest.AssertAttributePresent(t, records[2], KEY_POOL_ID, POOL_ID)
+	honeybeetest.AssertAttributePresent(t, records[2], KEY_PEER_ID, PEER_ID)
+}