From b1da01f501f7fa1eb4772508ac3343fd8ae58f08 Mon Sep 17 00:00:00 2001
From: yanggang <gang.yang@daocloud.io>
Date: Tue, 31 Oct 2023 01:44:02 +0000
Subject: [PATCH] Add a unit test case for logrus-entry-contents

Signed-off-by: yanggang <gang.yang@daocloud.io>
---
 pkg/util/logging/dual_mode_logger_test.go | 69 +++++++++++++++++++++++
 1 file changed, 69 insertions(+)

diff --git a/pkg/util/logging/dual_mode_logger_test.go b/pkg/util/logging/dual_mode_logger_test.go
index 629ca94997..e5f3813833 100644
--- a/pkg/util/logging/dual_mode_logger_test.go
+++ b/pkg/util/logging/dual_mode_logger_test.go
@@ -18,6 +18,8 @@ package logging
 
 import (
 	"compress/gzip"
+	"errors"
+	"fmt"
 	"io"
 	"os"
 	"strings"
@@ -59,6 +61,73 @@ func TestDualModeLogger(t *testing.T) {
 	assert.Equal(t, true, os.IsNotExist(err))
 }
 
+func TestDualModeLoggerEntryHook(t *testing.T) {
+	logCounter := NewLogHook()
+	logger, err := NewTempFileLogger(
+		logrus.WarnLevel,
+		FormatText,
+		logCounter,
+		//logrus.Fields{"namespace": "ns1", "resource": "pod", "name": "podxxx"})
+		logrus.Fields{"namespace": "ns1", "resource": "podxxxx"})
+	require.NoError(t, err)
+
+	logger.Warnln("test 1")
+	logger.Warnln("test 2")
+	logger.Warnln("test 3")
+
+	logger.WithFields(logrus.Fields{"namespace": "ns2", "resource": "pod", "name": "pod1"})
+	logger.Warnln("test 1 with field ns2")
+	logger.WithFields(logrus.Fields{"namespace": "ns2", "resource": "pod", "name": "pod2"})
+	logger.Warnln("test 2 with field ns2")
+	logger.WithFields(logrus.Fields{"namespace": "ns2", "resource": "pod", "name": "pod3"})
+	logger.Warnln("test 3 with field ns2")
+
+	logger.WithFields(logrus.Fields{"namespace": "ns3", "resource": "pod", "name": "pod4"})
+	logger.Warnln("test 4 with field ns3")
+	logger.WithFields(logrus.Fields{"namespace": "ns3", "resource": "pod", "name": "pod5"})
+	logger.Warnln("test 5 with field ns3")
+	logger.WithFields(logrus.Fields{"namespace": "ns3", "resource": "pod", "name": "pod6"})
+	logger.Warnln("test 6 with field ns3")
+
+	logger.WithError(errors.New("found err 1")).WithFields(
+		logrus.Fields{"namespace": "ns4", "resource": "pod", "name": "pod7"})
+	logger.Warnln("test 7 with field ns4 and WithError")
+	logger.WithError(errors.New("found err 2")).WithFields(
+		logrus.Fields{"namespace": "ns4", "resource": "pod", "name": "pod8"})
+	logger.Warnln("test 8 with field ns4 and WithError")
+
+	logger.Errorln("test error 1 in pod 1")
+	logger.Errorln("test error 2 in pod 2")
+	logger.Errorln("test error 3 in pod 3")
+	logger.Errorln("test error 4 in pod 4")
+
+	logger.DoneForPersist(velerotest.NewLogger())
+
+	logCntForWarn := logCounter.GetCount(logrus.WarnLevel)
+	fmt.Printf("logCntForWarn=%v\n", logCntForWarn)
+
+	logCntForError := logCounter.GetCount(logrus.ErrorLevel)
+	fmt.Printf("logCntForError=%v\n", logCntForError)
+
+	logResultForWarn := logCounter.GetEntries(logrus.WarnLevel)
+	for key, value := range logResultForWarn.Namespaces {
+		fmt.Printf("logResultForWarn namespace key=%+v , value=%+v\n", key, value)
+	}
+
+	logResultForError := logCounter.GetEntries(logrus.WarnLevel)
+	for key, value := range logResultForError.Namespaces {
+		fmt.Printf("logResultForError namespace key=%+v , value=%+v\n", key, value)
+	}
+
+	logFile, err := logger.GetPersistFile()
+	require.NoError(t, err)
+
+	logger.Dispose(velerotest.NewLogger())
+
+	_, err = os.Stat(logFile.Name())
+	assert.Equal(t, true, os.IsNotExist(err))
+}
+
 func readLogString(file *os.File) (string, error) {
 	gzr, err := gzip.NewReader(file)
 	if err != nil {