Add test for output interceptor.

edoger · Feb 7, 2021 · 5857cac · 5857cac
1 parent af7dc11
commit 5857cac
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/logger_test.go b/logger_test.go
@@ -20,6 +20,7 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"io"
 	"os"
 	"strings"
 	"testing"
@@ -638,3 +639,26 @@ func TestLogger_BigLog(t *testing.T) {
 		t.Fatalf("Big log: %d", n)
 	}
 }
+
+func TestLogger_Interceptor(t *testing.T) {
+	w1 := new(bytes.Buffer)
+	o := New("test")
+	o.SetOutput(w1)
+	o.SetLevel(TraceLevel)
+
+	w2 := new(bytes.Buffer)
+	interceptor := func(summary Summary, writer io.Writer) (int, error) {
+		return w2.Write([]byte(summary.Message())) // message only
+	}
+	if o.SetOutputInterceptor(interceptor) == nil {
+		t.Fatal("Logger.SetOutputInterceptor(): nil")
+	}
+
+	o.Echo("foo")
+	if got := w1.String(); got != "" {
+		t.Fatalf("Logger.SetOutputInterceptor(): got %s", got)
+	}
+	if got := w2.String(); got != "foo" {
+		t.Fatalf("Logger.SetOutputInterceptor(): got %s", got)
+	}
+}