From 977a979deb98d01992074d818348853e385a1de1 Mon Sep 17 00:00:00 2001 From: John Rowley Date: Sun, 10 Mar 2024 21:50:19 +0000 Subject: [PATCH] feat: added slog adapter --- fxevent/slog.go | 289 ++++++++++++++++++++++ fxevent/slog_test.go | 576 +++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 865 insertions(+) create mode 100644 fxevent/slog.go create mode 100644 fxevent/slog_test.go diff --git a/fxevent/slog.go b/fxevent/slog.go new file mode 100644 index 000000000..8c1445a73 --- /dev/null +++ b/fxevent/slog.go @@ -0,0 +1,289 @@ +// Copyright (c) 2021 Uber Technologies, Inc. +// +// Permission is hereby granted, free of charge, to any person obtaining a copy +// of this software and associated documentation files (the "Software"), to deal +// in the Software without restriction, including without limitation the rights +// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +// copies of the Software, and to permit persons to whom the Software is +// furnished to do so, subject to the following conditions: +// +// The above copyright notice and this permission notice shall be included in +// all copies or substantial portions of the Software. +// +// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN +// THE SOFTWARE. + +package fxevent + +import ( + "context" + "fmt" + "log/slog" + "strings" +) + +var _ Logger = &SlogLogger{} + +// New returns a new SlogLogger +func New(logger *slog.Logger) *SlogLogger { + return &SlogLogger{Logger: logger} +} + +// Option is an option to the WithLogger function. +type Option func(logger *slog.Logger) *slog.Logger + +// SlogLogger is a shim that allows uber/fx to use slog for logging. +type SlogLogger struct { + Logger *slog.Logger + + ctx context.Context + logLevel slog.Level + errorLevel *slog.Level +} + +// UseContext sets the context that will be used when logging to slog. +func (l *SlogLogger) UseContext(ctx context.Context) { + l.ctx = ctx +} + +// UseLogLevel sets the level of non-error logs emitted by Fx to level. +func (l *SlogLogger) UseLogLevel(level slog.Level) { + l.logLevel = level +} + +// UseErrorLevel sets the level of error logs emitted by Fx to level. +func (l *SlogLogger) UseErrorLevel(level slog.Level) { + l.errorLevel = &level +} + +func (l *SlogLogger) filter(fields []interface{}) []interface{} { + filtered := []interface{}{} + + for _, field := range fields { + if field, ok := field.(slog.Attr); ok { + if _, ok := field.Value.Any().(slogFieldSkip); ok { + continue + } + } + + filtered = append(filtered, field) + } + + return filtered +} + +func (l *SlogLogger) logEvent(msg string, fields ...interface{}) { + + l.Logger.Log(l.ctx, l.logLevel, msg, l.filter(fields)...) +} + +func (l *SlogLogger) logError(msg string, fields ...interface{}) { + lvl := slog.LevelError + if l.errorLevel != nil { + lvl = *l.errorLevel + } + + l.Logger.Log(l.ctx, lvl, msg, l.filter(fields)...) +} + +// LogEvent logs the given event to the provided Zap logger. +func (l *SlogLogger) LogEvent(event Event) { + switch e := event.(type) { + case *OnStartExecuting: + l.logEvent("OnStart hook executing", + slog.String("callee", e.FunctionName), + slog.String("caller", e.CallerName), + ) + case *OnStartExecuted: + if e.Err != nil { + l.logError("OnStart hook failed", + slog.String("callee", e.FunctionName), + slog.String("caller", e.CallerName), + slogErr(e.Err), + ) + } else { + l.logEvent("OnStart hook executed", + slog.String("callee", e.FunctionName), + slog.String("caller", e.CallerName), + slog.String("runtime", e.Runtime.String()), + ) + } + case *OnStopExecuting: + l.logEvent("OnStop hook executing", + slog.String("callee", e.FunctionName), + slog.String("caller", e.CallerName), + ) + case *OnStopExecuted: + if e.Err != nil { + l.logError("OnStop hook failed", + slog.String("callee", e.FunctionName), + slog.String("caller", e.CallerName), + slogErr(e.Err), + ) + } else { + l.logEvent("OnStop hook executed", + slog.String("callee", e.FunctionName), + slog.String("caller", e.CallerName), + slog.String("runtime", e.Runtime.String()), + ) + } + case *Supplied: + if e.Err != nil { + l.logError("error encountered while applying options", + slog.String("type", e.TypeName), + slogStrings("moduletrace", e.ModuleTrace), + slogStrings("stacktrace", e.StackTrace), + slogMaybeModuleField(e.ModuleName), + slogErr(e.Err)) + } else { + l.logEvent("supplied", + slog.String("type", e.TypeName), + slogStrings("stacktrace", e.StackTrace), + slogStrings("moduletrace", e.ModuleTrace), + slogMaybeModuleField(e.ModuleName), + ) + } + case *Provided: + for _, rtype := range e.OutputTypeNames { + l.logEvent("provided", + slog.String("constructor", e.ConstructorName), + slogStrings("stacktrace", e.StackTrace), + slogStrings("moduletrace", e.ModuleTrace), + slogMaybeModuleField(e.ModuleName), + slog.String("type", rtype), + slogMaybeBool("private", e.Private), + ) + } + if e.Err != nil { + l.logError("error encountered while applying options", + slogMaybeModuleField(e.ModuleName), + slogStrings("stacktrace", e.StackTrace), + slogStrings("moduletrace", e.ModuleTrace), + slogErr(e.Err)) + } + case *Replaced: + for _, rtype := range e.OutputTypeNames { + l.logEvent("replaced", + slogStrings("stacktrace", e.StackTrace), + slogStrings("moduletrace", e.ModuleTrace), + slogMaybeModuleField(e.ModuleName), + slog.String("type", rtype), + ) + } + if e.Err != nil { + l.logError("error encountered while replacing", + slogStrings("stacktrace", e.StackTrace), + slogStrings("moduletrace", e.ModuleTrace), + slogMaybeModuleField(e.ModuleName), + slogErr(e.Err)) + } + case *Decorated: + for _, rtype := range e.OutputTypeNames { + l.logEvent("decorated", + slog.String("decorator", e.DecoratorName), + slogStrings("stacktrace", e.StackTrace), + slogStrings("moduletrace", e.ModuleTrace), + slogMaybeModuleField(e.ModuleName), + slog.String("type", rtype), + ) + } + if e.Err != nil { + l.logError("error encountered while applying options", + slogStrings("stacktrace", e.StackTrace), + slogStrings("moduletrace", e.ModuleTrace), + slogMaybeModuleField(e.ModuleName), + slogErr(e.Err)) + } + case *Run: + if e.Err != nil { + l.logError("error returned", + slog.String("name", e.Name), + slog.String("kind", e.Kind), + slogMaybeModuleField(e.ModuleName), + slogErr(e.Err), + ) + } else { + l.logEvent("run", + slog.String("name", e.Name), + slog.String("kind", e.Kind), + slogMaybeModuleField(e.ModuleName), + ) + } + case *Invoking: + // Do not log stack as it will make logs hard to read. + l.logEvent("invoking", + slog.String("function", e.FunctionName), + slogMaybeModuleField(e.ModuleName), + ) + case *Invoked: + if e.Err != nil { + l.logError("invoke failed", + slogErr(e.Err), + slog.String("stack", e.Trace), + slog.String("function", e.FunctionName), + slogMaybeModuleField(e.ModuleName), + ) + } + case *Stopping: + l.logEvent("received signal", + slog.String("signal", strings.ToUpper(e.Signal.String()))) + case *Stopped: + if e.Err != nil { + l.logError("stop failed", slogErr(e.Err)) + } + case *RollingBack: + l.logError("start failed, rolling back", slogErr(e.StartErr)) + case *RolledBack: + if e.Err != nil { + l.logError("rollback failed", slogErr(e.Err)) + } + case *Started: + if e.Err != nil { + l.logError("start failed", slogErr(e.Err)) + } else { + l.logEvent("started") + } + case *LoggerInitialized: + if e.Err != nil { + l.logError("custom logger initialization failed", slogErr(e.Err)) + } else { + l.logEvent("initialized custom fxevent.Logger", slog.String("function", e.ConstructorName)) + } + } +} + +type slogFieldSkip struct{} + +func slogMaybeModuleField(name string) slog.Attr { + if len(name) == 0 { + return slog.Any("module", slogFieldSkip{}) + } + + return slog.String("module", name) +} + +func slogMaybeBool(name string, b bool) slog.Attr { + if !b { + return slog.Any(name, slogFieldSkip{}) + } + + return slog.Bool(name, true) +} + +func slogErr(err error) slog.Attr { + return slog.String("err", err.Error()) +} + +func slogStrings(key string, str []string) slog.Attr { + var attrs []any + for i, val := range str { + attrs = append(attrs, slog.String(fmt.Sprintf("%d", i), val)) + } + + return slog.Group(key, attrs...) +} diff --git a/fxevent/slog_test.go b/fxevent/slog_test.go new file mode 100644 index 000000000..48742101c --- /dev/null +++ b/fxevent/slog_test.go @@ -0,0 +1,576 @@ +// Copyright (c) 2021 Uber Technologies, Inc. +// +// Permission is hereby granted, free of charge, to any person obtaining a copy +// of this software and associated documentation files (the "Software"), to deal +// in the Software without restriction, including without limitation the rights +// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +// copies of the Software, and to permit persons to whom the Software is +// furnished to do so, subject to the following conditions: +// +// The above copyright notice and this permission notice shall be included in +// all copies or substantial portions of the Software. +// +// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN +// THE SOFTWARE. + +package fxevent + +import ( + "context" + "errors" + "fmt" + "log/slog" + "os" + "strings" + "testing" + "time" + + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" +) + +type slogObservableEntry struct { + record slog.Record +} + +func (s slogObservableEntry) unwrap(attr slog.Attr, out map[string]interface{}) { + anyAttr := attr.Value.Any() + + sliceAttr, ok := anyAttr.([]slog.Attr) + + if !ok { + out[attr.Key] = anyAttr + return + } + + sliceAttrValues := make([]any, len(sliceAttr)) + for i, iter := range sliceAttr { + sliceAttrValues[i] = iter.Value.Any() + } + + out[attr.Key] = sliceAttrValues +} + +func (s slogObservableEntry) ContextMap() map[string]interface{} { + contextMap := map[string]interface{}{} + + s.record.Attrs(func(a slog.Attr) bool { + s.unwrap(a, contextMap) + return true + }) + return contextMap +} + +type slogObservableLogger struct { + level slog.Level + entries []slogObservableEntry + attrs []slog.Attr +} + +func (s *slogObservableLogger) Enabled(ctx context.Context, level slog.Level) bool { + return int(s.level) <= int(level) +} + +func (s *slogObservableLogger) Handle(ctx context.Context, record slog.Record) error { + s.entries = append(s.entries, slogObservableEntry{record}) + return nil +} + +func (s *slogObservableLogger) WithAttrs(attrs []slog.Attr) slog.Handler { + return &slogObservableLogger{ + level: s.level, + entries: s.entries, + attrs: append(s.attrs, attrs...), + } +} + +func (s *slogObservableLogger) WithGroup(name string) slog.Handler { + return s +} + +func (s *slogObservableLogger) TakeAll() []slogObservableEntry { + return s.entries +} + +func newSlogObservableLogger(level slog.Level) (*slog.Logger, *slogObservableLogger) { + handler := &slogObservableLogger{level: level} + return slog.New(handler), handler +} + +func TestSlogLogger(t *testing.T) { + t.Parallel() + + someError := errors.New("some error") + + tests := []struct { + name string + give Event + wantMessage string + wantFields map[string]interface{} + }{ + { + name: "OnStartExecuting", + give: &OnStartExecuting{ + FunctionName: "hook.onStart", + CallerName: "bytes.NewBuffer", + }, + wantMessage: "OnStart hook executing", + wantFields: map[string]interface{}{ + "caller": "bytes.NewBuffer", + "callee": "hook.onStart", + }, + }, + { + name: "OnStopExecuting", + give: &OnStopExecuting{ + FunctionName: "hook.onStop1", + CallerName: "bytes.NewBuffer", + }, + wantMessage: "OnStop hook executing", + wantFields: map[string]interface{}{ + "caller": "bytes.NewBuffer", + "callee": "hook.onStop1", + }, + }, + { + name: "OnStopExecuted/Error", + give: &OnStopExecuted{ + FunctionName: "hook.onStart1", + CallerName: "bytes.NewBuffer", + Err: fmt.Errorf("some error"), + }, + wantMessage: "OnStop hook failed", + wantFields: map[string]interface{}{ + "caller": "bytes.NewBuffer", + "callee": "hook.onStart1", + "err": "some error", + }, + }, + { + name: "OnStopExecuted", + give: &OnStopExecuted{ + FunctionName: "hook.onStart1", + CallerName: "bytes.NewBuffer", + Runtime: time.Millisecond * 3, + }, + wantMessage: "OnStop hook executed", + wantFields: map[string]interface{}{ + "caller": "bytes.NewBuffer", + "callee": "hook.onStart1", + "runtime": "3ms", + }, + }, + { + name: "OnStartExecuted/Error", + give: &OnStartExecuted{ + FunctionName: "hook.onStart1", + CallerName: "bytes.NewBuffer", + Err: fmt.Errorf("some error"), + }, + wantMessage: "OnStart hook failed", + wantFields: map[string]interface{}{ + "caller": "bytes.NewBuffer", + "callee": "hook.onStart1", + "err": "some error", + }, + }, + { + name: "OnStartExecuted", + give: &OnStartExecuted{ + FunctionName: "hook.onStart1", + CallerName: "bytes.NewBuffer", + Runtime: time.Millisecond * 3, + }, + wantMessage: "OnStart hook executed", + wantFields: map[string]interface{}{ + "caller": "bytes.NewBuffer", + "callee": "hook.onStart1", + "runtime": "3ms", + }, + }, + { + name: "Supplied", + give: &Supplied{ + TypeName: "*bytes.Buffer", + StackTrace: []string{"main.main", "runtime.main"}, + ModuleTrace: []string{"main.main"}, + }, + wantMessage: "supplied", + wantFields: map[string]interface{}{ + "type": "*bytes.Buffer", + "stacktrace": []interface{}{"main.main", "runtime.main"}, + "moduletrace": []interface{}{"main.main"}, + }, + }, + { + name: "Supplied/Error", + give: &Supplied{ + TypeName: "*bytes.Buffer", + StackTrace: []string{"main.main", "runtime.main"}, + ModuleTrace: []string{"main.main"}, + Err: someError, + }, + wantMessage: "error encountered while applying options", + wantFields: map[string]interface{}{ + "type": "*bytes.Buffer", + "stacktrace": []interface{}{"main.main", "runtime.main"}, + "moduletrace": []interface{}{"main.main"}, + "err": "some error", + }, + }, + { + name: "Provide", + give: &Provided{ + ConstructorName: "bytes.NewBuffer()", + StackTrace: []string{"main.main", "runtime.main"}, + ModuleTrace: []string{"main.main"}, + ModuleName: "myModule", + OutputTypeNames: []string{"*bytes.Buffer"}, + Private: false, + }, + wantMessage: "provided", + wantFields: map[string]interface{}{ + "constructor": "bytes.NewBuffer()", + "stacktrace": []interface{}{"main.main", "runtime.main"}, + "moduletrace": []interface{}{"main.main"}, + "type": "*bytes.Buffer", + "module": "myModule", + }, + }, + { + name: "PrivateProvide", + give: &Provided{ + ConstructorName: "bytes.NewBuffer()", + StackTrace: []string{"main.main", "runtime.main"}, + ModuleTrace: []string{"main.main"}, + ModuleName: "myModule", + OutputTypeNames: []string{"*bytes.Buffer"}, + Private: true, + }, + wantMessage: "provided", + wantFields: map[string]interface{}{ + "constructor": "bytes.NewBuffer()", + "stacktrace": []interface{}{"main.main", "runtime.main"}, + "moduletrace": []interface{}{"main.main"}, + "type": "*bytes.Buffer", + "module": "myModule", + "private": true, + }, + }, + { + name: "Provide/Error", + give: &Provided{ + StackTrace: []string{"main.main", "runtime.main"}, + ModuleTrace: []string{"main.main"}, + Err: someError, + }, + wantMessage: "error encountered while applying options", + wantFields: map[string]interface{}{ + "stacktrace": []interface{}{"main.main", "runtime.main"}, + "moduletrace": []interface{}{"main.main"}, + "err": "some error", + }, + }, + { + name: "Replace", + give: &Replaced{ + ModuleName: "myModule", + StackTrace: []string{"main.main", "runtime.main"}, + ModuleTrace: []string{"main.main"}, + OutputTypeNames: []string{"*bytes.Buffer"}, + }, + wantMessage: "replaced", + wantFields: map[string]interface{}{ + "type": "*bytes.Buffer", + "stacktrace": []interface{}{"main.main", "runtime.main"}, + "moduletrace": []interface{}{"main.main"}, + "module": "myModule", + }, + }, + { + name: "Replace/Error", + give: &Replaced{ + StackTrace: []string{"main.main", "runtime.main"}, + ModuleTrace: []string{"main.main"}, + Err: someError, + }, + + wantMessage: "error encountered while replacing", + wantFields: map[string]interface{}{ + "stacktrace": []interface{}{"main.main", "runtime.main"}, + "moduletrace": []interface{}{"main.main"}, + "err": "some error", + }, + }, + { + name: "Decorate", + give: &Decorated{ + DecoratorName: "bytes.NewBuffer()", + StackTrace: []string{"main.main", "runtime.main"}, + ModuleTrace: []string{"main.main"}, + ModuleName: "myModule", + OutputTypeNames: []string{"*bytes.Buffer"}, + }, + wantMessage: "decorated", + wantFields: map[string]interface{}{ + "decorator": "bytes.NewBuffer()", + "stacktrace": []interface{}{"main.main", "runtime.main"}, + "moduletrace": []interface{}{"main.main"}, + "type": "*bytes.Buffer", + "module": "myModule", + }, + }, + { + name: "Decorate/Error", + give: &Decorated{ + StackTrace: []string{"main.main", "runtime.main"}, + ModuleTrace: []string{"main.main"}, + Err: someError, + }, + wantMessage: "error encountered while applying options", + wantFields: map[string]interface{}{ + "stacktrace": []interface{}{"main.main", "runtime.main"}, + "moduletrace": []interface{}{"main.main"}, + "err": "some error", + }, + }, + { + name: "Run", + give: &Run{Name: "bytes.NewBuffer()", Kind: "constructor"}, + wantMessage: "run", + wantFields: map[string]interface{}{ + "name": "bytes.NewBuffer()", + "kind": "constructor", + }, + }, + { + name: "Run with module", + give: &Run{ + Name: "bytes.NewBuffer()", + Kind: "constructor", + ModuleName: "myModule", + }, + wantMessage: "run", + wantFields: map[string]interface{}{ + "name": "bytes.NewBuffer()", + "kind": "constructor", + "module": "myModule", + }, + }, + { + name: "Run/Error", + give: &Run{ + Name: "bytes.NewBuffer()", + Kind: "constructor", + Err: someError, + }, + wantMessage: "error returned", + wantFields: map[string]interface{}{ + "name": "bytes.NewBuffer()", + "kind": "constructor", + "err": "some error", + }, + }, + { + name: "Invoking/Success", + give: &Invoking{ModuleName: "myModule", FunctionName: "bytes.NewBuffer()"}, + wantMessage: "invoking", + wantFields: map[string]interface{}{ + "function": "bytes.NewBuffer()", + "module": "myModule", + }, + }, + { + name: "Invoked/Error", + give: &Invoked{FunctionName: "bytes.NewBuffer()", Err: someError}, + wantMessage: "invoke failed", + wantFields: map[string]interface{}{ + "err": "some error", + "stack": "", + "function": "bytes.NewBuffer()", + }, + }, + { + name: "Start/Error", + give: &Started{Err: someError}, + wantMessage: "start failed", + wantFields: map[string]interface{}{ + "err": "some error", + }, + }, + { + name: "Stopping", + give: &Stopping{Signal: os.Interrupt}, + wantMessage: "received signal", + wantFields: map[string]interface{}{ + "signal": "INTERRUPT", + }, + }, + { + name: "Stopped/Error", + give: &Stopped{Err: someError}, + wantMessage: "stop failed", + wantFields: map[string]interface{}{ + "err": "some error", + }, + }, + { + name: "RollingBack/Error", + give: &RollingBack{StartErr: someError}, + wantMessage: "start failed, rolling back", + wantFields: map[string]interface{}{ + "err": "some error", + }, + }, + { + name: "RolledBack/Error", + give: &RolledBack{Err: someError}, + wantMessage: "rollback failed", + wantFields: map[string]interface{}{ + "err": "some error", + }, + }, + { + name: "Started", + give: &Started{}, + wantMessage: "started", + wantFields: map[string]interface{}{}, + }, + { + name: "LoggerInitialized/Error", + give: &LoggerInitialized{Err: someError}, + wantMessage: "custom logger initialization failed", + wantFields: map[string]interface{}{ + "err": "some error", + }, + }, + { + name: "LoggerInitialized", + give: &LoggerInitialized{ConstructorName: "bytes.NewBuffer()"}, + wantMessage: "initialized custom fxevent.Logger", + wantFields: map[string]interface{}{ + "function": "bytes.NewBuffer()", + }, + }, + } + + t.Run("debug observer, log at default (info)", func(t *testing.T) { + for _, tt := range tests { + tt := tt + t.Run(tt.name, func(t *testing.T) { + t.Parallel() + + core, observedLogs := newSlogObservableLogger(slog.LevelDebug) + (&SlogLogger{Logger: core}).LogEvent(tt.give) + + logs := observedLogs.TakeAll() + require.Len(t, logs, 1) + got := logs[0] + + assert.Equal(t, tt.wantMessage, got.record.Message) + assert.Equal(t, tt.wantFields, got.ContextMap()) + }) + } + }) + + t.Run("info observer, log at debug", func(t *testing.T) { + for _, tt := range tests { + tt := tt + t.Run(tt.name, func(t *testing.T) { + t.Parallel() + + core, observedLogs := newSlogObservableLogger(slog.LevelInfo) + l := &SlogLogger{Logger: core} + l.UseLogLevel(slog.LevelDebug) + l.LogEvent(tt.give) + + logs := observedLogs.TakeAll() + // logs are not visible unless they are errors + if strings.HasSuffix(tt.name, "/Error") { + require.Len(t, logs, 1) + got := logs[0] + assert.Equal(t, tt.wantMessage, got.record.Message) + assert.Equal(t, tt.wantFields, got.ContextMap()) + } else { + require.Len(t, logs, 0) + } + }) + } + }) + + t.Run("info observer, log/error at debug", func(t *testing.T) { + for _, tt := range tests { + tt := tt + t.Run(tt.name, func(t *testing.T) { + t.Parallel() + + core, observedLogs := newSlogObservableLogger(slog.LevelInfo) + l := &SlogLogger{Logger: core} + l.UseLogLevel(slog.LevelDebug) + l.UseErrorLevel(slog.LevelDebug) + l.LogEvent(tt.give) + + logs := observedLogs.TakeAll() + require.Len(t, logs, 0, "no logs should be visible") + }) + } + }) + + t.Run("test setting log levels", func(t *testing.T) { + levels := []slog.Level{ + slog.LevelError, + slog.LevelDebug, + slog.LevelWarn, + slog.LevelInfo, + } + + for _, level := range levels { + core, observedLogs := newSlogObservableLogger(level) + logger := &SlogLogger{Logger: core} + logger.UseLogLevel(level) + func() { + defer func() { + recover() + }() + logger.LogEvent(&OnStartExecuting{ + FunctionName: "hook.onStart", + CallerName: "bytes.NewBuffer", + }) + }() + logs := observedLogs.TakeAll() + require.Len(t, logs, 1) + } + }) + + t.Run("test setting error log levels", func(t *testing.T) { + levels := []slog.Level{ + slog.LevelError, + slog.LevelDebug, + slog.LevelWarn, + slog.LevelInfo, + } + + for _, level := range levels { + core, observedLogs := newSlogObservableLogger(level) + logger := &SlogLogger{Logger: core} + logger.UseErrorLevel(level) + func() { + defer func() { + recover() + }() + logger.LogEvent(&OnStopExecuted{ + FunctionName: "hook.onStart1", + CallerName: "bytes.NewBuffer", + Err: fmt.Errorf("some error"), + }) + }() + logs := observedLogs.TakeAll() + require.Len(t, logs, 1) + } + }) +}