✅ test: add some new tests case for logger

gookit · Dec 1, 2023 · 61216e2 · 61216e2
1 parent 8348571
commit 61216e2
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 0 deletions.
diff --git a/common_test.go b/common_test.go
@@ -128,6 +128,8 @@ type testHandler struct {
 	errOnHandle bool
 	errOnFlush  bool
 	errOnClose  bool
+	// hooks
+	callOnFlush func()
 }
 
 func newTestHandler() *testHandler {
@@ -151,6 +153,9 @@ func (h *testHandler) Flush() error {
 	if h.errOnFlush {
 		return errorx.Raw("flush error")
 	}
+	if h.callOnFlush != nil {
+		h.callOnFlush()
+	}
 
 	h.Reset()
 	return nil

diff --git a/logger_test.go b/logger_test.go
@@ -225,6 +225,26 @@ func TestLogger_option_BackupArgs(t *testing.T) {
 	assert.StrContains(t, s, "field message3")
 }
 
+func TestLogger_FlushTimeout(t *testing.T) {
+	h := newTestHandler()
+	l := slog.NewWithHandlers(h)
+
+	// test flush error
+	h.errOnFlush = true
+	l.FlushTimeout(time.Millisecond * 2)
+
+	// test flush timeout
+	h.errOnFlush = false
+	h.callOnFlush = func() {
+		time.Sleep(time.Millisecond * 25)
+	}
+	l.FlushTimeout(time.Millisecond * 20)
+
+	assert.Panics(t, func() {
+		l.StopDaemon()
+	})
+}
+
 func TestLogger_rewrite_record(t *testing.T) {
 	h := newTestHandler()
 	l := slog.NewWithHandlers(h)

diff --git a/util_test.go b/util_test.go
@@ -44,6 +44,10 @@ func TestInner_parseTemplateToFields(t *testing.T) {
 	// dump.P(ss, str)
 }
 
+func TestUtil_EncodeToString(t *testing.T) {
+	assert.Eq(t, "{a:1}", EncodeToString(map[string]any{"a": 1}))
+}
+
 func TestUtil_formatArgsWithSpaces(t *testing.T) {
 	// tests for formatArgsWithSpaces
 	tests := []struct {