Refactor log_test.go and add unit tests for Setup function

beatlabs · Oct 20, 2024 · 1578b27 · 1578b27
1 parent 2c82f61
commit 1578b27
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/observability/log/log_test.go b/observability/log/log_test.go
@@ -9,6 +9,28 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestSetup(t *testing.T) {
+	t.Run("JSON", func(t *testing.T) {
+		cfg := &Config{
+			Attributes: []slog.Attr{},
+			IsJSON:     true,
+			Level:      "debug",
+		}
+		Setup(cfg)
+		assert.NotNil(t, slog.Default())
+	})
+
+	t.Run("Text", func(t *testing.T) {
+		cfg := &Config{
+			Attributes: []slog.Attr{},
+			IsJSON:     false,
+			Level:      "debug",
+		}
+		Setup(cfg)
+		assert.NotNil(t, slog.Default())
+	})
+}
+
 func TestContext(t *testing.T) {
 	l := slog.Default()