forked from kyma-project/kyma
-
Notifications
You must be signed in to change notification settings - Fork 0
/
logger_mock.go
34 lines (28 loc) · 814 Bytes
/
logger_mock.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
package logger
import (
"time"
"github.com/sirupsen/logrus"
)
// THTimeForcedFormatter is a Logrus compatible formatter which wraps original formatter and forces specific time.
// Designed to be used in testing.
type THTimeForcedFormatter struct {
// OrigFormatter is an original formatter
OrigFormatter logrus.Formatter
// Time is a time to be forces in entries.
Time time.Time
}
// Format entry but forces time for testing purposes
func (f *THTimeForcedFormatter) Format(entry *logrus.Entry) ([]byte, error) {
data := make(logrus.Fields, len(entry.Data))
for k, v := range entry.Data {
data[k] = v
}
entryModified := &logrus.Entry{
Logger: entry.Logger,
Data: data,
Time: f.Time,
Level: entry.Level,
Message: entry.Message,
}
return f.OrigFormatter.Format(entryModified)
}