feat: add level get an set methods (#29)

danteay · Apr 17, 2024 · 181c67a · 181c67a
1 parent 0296332
commit 181c67a
Show file tree

Hide file tree

Showing 2 changed files with 108 additions and 5 deletions.
diff --git a/adapters/slog/slog.go b/adapters/slog/slog.go
@@ -47,6 +47,16 @@ func (a *Adapter) Writer() io.Writer {
 
 func (a *Adapter) SetWriter(w io.Writer) {
 	a.writer = w
+	a.logger = getSlogInstance(a.level, a.writer)
+}
+
+func (a *Adapter) Level() levels.Level {
+	return a.level
+}
+
+func (a *Adapter) SetLevel(level levels.Level) {
+	a.level = level
+	a.logger = getSlogInstance(level, a.writer)
 }
 
 // Logger returns the slog logger instance

diff --git a/adapters/slog/slog_test.go b/adapters/slog/slog_test.go
@@ -15,11 +15,12 @@ import (
 )
 
 type testMsg struct {
-	Level   string `json:"level"`
-	Message string `json:"msg"`
-	Error   string `json:"error"`
-	Key1    string `json:"key1"`
-	Key2    int    `json:"key2"`
+	Level   string   `json:"level"`
+	Message string   `json:"msg"`
+	Error   string   `json:"error"`
+	Key1    string   `json:"key1"`
+	Key2    int      `json:"key2"`
+	Stack   []string `json:"stack"`
 }
 
 func TestAdapter_Log(t *testing.T) {
@@ -136,6 +137,98 @@ func TestAdapter_Log(t *testing.T) {
 
 		assert.Equal(t, "", logOutput.String())
 	})
+
+	t.Run("should log message with trace", func(t *testing.T) {
+		var logOutput bytes.Buffer
+
+		logger := New(WithLevel(levels.TraceLevel), WithWriter(&logOutput), WithTrace())
+
+		logFields := fields.New().SetMap(map[string]any{
+			"key1": "value1",
+			"key2": 42,
+		})
+
+		msg := "Test message"
+		err := errors.New("test error")
+
+		logger.Log(levels.Debug, err, logFields, msg)
+
+		res := testMsg{}
+
+		if errUnmarshall := json.Unmarshal(logOutput.Bytes(), &res); errUnmarshall != nil {
+			t.Fatal(errUnmarshall)
+		}
+
+		assert.Equal(t, levels.Debug.String(), strings.ToLower(res.Level))
+		assert.Equal(t, msg, res.Message)
+		assert.Equal(t, logFields.Get("key1"), res.Key1)
+		assert.Equal(t, logFields.Get("key2"), res.Key2)
+		assert.Equal(t, err.Error(), res.Error)
+		assert.NotEmpty(t, res.Stack)
+	})
+
+	t.Run("should change level", func(t *testing.T) {
+		var logOutput bytes.Buffer
+
+		logger := New(WithLevel(levels.Info), WithWriter(&logOutput))
+
+		logFields := fields.New().SetMap(map[string]any{
+			"key1": "value1",
+			"key2": 42,
+		})
+
+		msg := "Test message"
+		err := errors.New("test error")
+
+		logger.Log(levels.Debug, err, logFields, msg)
+
+		assert.Equal(t, "", logOutput.String())
+
+		logger.SetLevel(levels.Debug)
+
+		logger.Log(levels.Debug, err, logFields, msg)
+
+		res := testMsg{}
+
+		if errUnmarshall := json.Unmarshal(logOutput.Bytes(), &res); errUnmarshall != nil {
+			t.Fatal(errUnmarshall)
+		}
+
+		assert.Equal(t, levels.Debug.String(), strings.ToLower(res.Level))
+		assert.Equal(t, msg, res.Message)
+		assert.Equal(t, logFields.Get("key1"), res.Key1)
+		assert.Equal(t, logFields.Get("key2"), res.Key2)
+		assert.Equal(t, err.Error(), res.Error)
+	})
+
+	t.Run("should change writer", func(t *testing.T) {
+		var logOutput bytes.Buffer
+
+		logger := New(WithLevel(levels.Debug), WithWriter(&logOutput))
+
+		logFields := fields.New().SetMap(map[string]any{
+			"key1": "value1",
+			"key2": 42,
+		})
+
+		msg := "Test message"
+		err := errors.New("test error")
+
+		logger.Log(levels.Debug, err, logFields, msg)
+
+		assert.NotEqual(t, "", logOutput.String())
+
+		logOutput.Reset()
+
+		var newLogOutput bytes.Buffer
+
+		logger.SetWriter(&newLogOutput)
+
+		logger.Log(levels.Debug, err, logFields, msg)
+
+		assert.NotEqual(t, "", newLogOutput.String())
+		assert.Equal(t, "", logOutput.String())
+	})
 }
 
 func TestGetLevels(t *testing.T) {