9cd6f6eef2
Current implementation stores a copy of the logger as a pointer and update its content, which is now unecessary since the introduction of WithContext. The new WithContext now takes the pointer and store it in the context if none is stored already or if the last one stored is a different pointer. This way it is still possible to update the context of a logger stored in the context, but it is also possible to store a copy of the logger in a sub-context. Fix #80
64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package zerolog
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestCtx(t *testing.T) {
|
|
log := New(ioutil.Discard)
|
|
ctx := log.WithContext(context.Background())
|
|
log2 := Ctx(ctx)
|
|
if !reflect.DeepEqual(log, *log2) {
|
|
t.Error("Ctx did not return the expected logger")
|
|
}
|
|
|
|
// update
|
|
log = log.Level(InfoLevel)
|
|
ctx = log.WithContext(ctx)
|
|
log2 = Ctx(ctx)
|
|
if !reflect.DeepEqual(log, *log2) {
|
|
t.Error("Ctx did not return the expected logger")
|
|
}
|
|
|
|
log2 = Ctx(context.Background())
|
|
if log2 != disabledLogger {
|
|
t.Error("Ctx did not return the expected logger")
|
|
}
|
|
}
|
|
|
|
func TestCtxDisabled(t *testing.T) {
|
|
dl := New(ioutil.Discard).Level(Disabled)
|
|
ctx := dl.WithContext(context.Background())
|
|
if ctx != context.Background() {
|
|
t.Error("WithContext stored a disabled logger")
|
|
}
|
|
|
|
l := New(ioutil.Discard).With().Str("foo", "bar").Logger()
|
|
ctx = l.WithContext(ctx)
|
|
if Ctx(ctx) != &l {
|
|
t.Error("WithContext did not store logger")
|
|
}
|
|
|
|
l.UpdateContext(func(c Context) Context {
|
|
return c.Str("bar", "baz")
|
|
})
|
|
ctx = l.WithContext(ctx)
|
|
if Ctx(ctx) != &l {
|
|
t.Error("WithContext did not store updated logger")
|
|
}
|
|
|
|
l = l.Level(DebugLevel)
|
|
ctx = l.WithContext(ctx)
|
|
if Ctx(ctx) != &l {
|
|
t.Error("WithContext did not store copied logger")
|
|
}
|
|
|
|
ctx = dl.WithContext(ctx)
|
|
if Ctx(ctx) != &dl {
|
|
t.Error("WithContext did not overide logger with a disabled logger")
|
|
}
|
|
}
|