add UT for pkg/util

Signed-off-by: Lyndon-Li <lyonghui@vmware.com>
This commit is contained in:
Lyndon-Li
2023-06-07 18:57:05 +08:00
parent 4227a824cd
commit 217b1dd066
5 changed files with 504 additions and 9 deletions

View File

@@ -34,3 +34,19 @@ func NewLoggerWithLevel(level logrus.Level) logrus.FieldLogger {
logger.Level = level
return logrus.NewEntry(logger)
}
type singleLogRecorder struct {
buffer *string
}
func (s *singleLogRecorder) Write(p []byte) (n int, err error) {
*s.buffer = string(p[:])
return len(p), nil
}
func NewSingleLogger(buffer *string) logrus.FieldLogger {
logger := logrus.New()
logger.Out = &singleLogRecorder{buffer: buffer}
logger.Level = logrus.TraceLevel
return logrus.NewEntry(logger)
}