83 lines
1.5 KiB
Go
83 lines
1.5 KiB
Go
package logutils
|
|
|
|
import (
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type MockLog struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func NewMockLog() *MockLog {
|
|
return &MockLog{}
|
|
}
|
|
|
|
func (m *MockLog) Fatalf(format string, args ...any) {
|
|
m.Called(append([]any{format}, args...)...)
|
|
}
|
|
|
|
func (m *MockLog) Panicf(format string, args ...any) {
|
|
m.Called(append([]any{format}, args...)...)
|
|
}
|
|
|
|
func (m *MockLog) Errorf(format string, args ...any) {
|
|
m.Called(append([]any{format}, args...)...)
|
|
}
|
|
|
|
func (m *MockLog) Warnf(format string, args ...any) {
|
|
m.Called(append([]any{format}, args...)...)
|
|
}
|
|
|
|
func (m *MockLog) Infof(format string, args ...any) {
|
|
m.Called(append([]any{format}, args...)...)
|
|
}
|
|
|
|
func (m *MockLog) Child(name string) Log {
|
|
m.Called(name)
|
|
return m
|
|
}
|
|
|
|
func (m *MockLog) SetLevel(level LogLevel) {
|
|
m.Called(level)
|
|
}
|
|
|
|
func (m *MockLog) OnFatalf(format string, args ...any) *MockLog {
|
|
arguments := append([]any{format}, args...)
|
|
|
|
m.On("Fatalf", arguments...)
|
|
|
|
return m
|
|
}
|
|
|
|
func (m *MockLog) OnPanicf(format string, args ...any) *MockLog {
|
|
arguments := append([]any{format}, args...)
|
|
|
|
m.On("Panicf", arguments...)
|
|
|
|
return m
|
|
}
|
|
|
|
func (m *MockLog) OnErrorf(format string, args ...any) *MockLog {
|
|
arguments := append([]any{format}, args...)
|
|
|
|
m.On("Errorf", arguments...)
|
|
|
|
return m
|
|
}
|
|
|
|
func (m *MockLog) OnWarnf(format string, args ...any) *MockLog {
|
|
arguments := append([]any{format}, args...)
|
|
|
|
m.On("Warnf", arguments...)
|
|
|
|
return m
|
|
}
|
|
|
|
func (m *MockLog) OnInfof(format string, args ...any) *MockLog {
|
|
arguments := append([]any{format}, args...)
|
|
|
|
m.On("Infof", arguments...)
|
|
|
|
return m
|
|
}
|