File size: 683 Bytes
fea99b3 | 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 35 | package log
import (
"context"
"log/slog"
"github.com/stretchr/testify/mock"
)
var _ slog.Handler = &MockHandler{}
func NewMockHandler() *MockHandler {
return &MockHandler{}
}
type MockHandler struct {
mock.Mock
}
func (m *MockHandler) Enabled(ctx context.Context, level slog.Level) bool {
return m.Called(ctx, level).Bool(0)
}
func (m *MockHandler) Handle(ctx context.Context, record slog.Record) error {
return m.Called(ctx, record).Error(0)
}
func (m *MockHandler) WithAttrs(attrs []slog.Attr) slog.Handler {
return m.Called(attrs).Get(0).(slog.Handler)
}
func (m *MockHandler) WithGroup(name string) slog.Handler {
return m.Called(name).Get(0).(slog.Handler)
}
|