diff --git a/core/logx/tracelogger_test.go b/core/logx/tracelogger_test.go index 9985cc66..466ee87c 100644 --- a/core/logx/tracelogger_test.go +++ b/core/logx/tracelogger_test.go @@ -4,6 +4,7 @@ import ( "context" "log" "strings" + "sync/atomic" "testing" "time" @@ -31,6 +32,7 @@ func TestTraceError(t *testing.T) { ctx := context.WithValue(context.Background(), tracespec.TracingKey, mock) l := WithContext(ctx).(*traceLogger) SetLevel(InfoLevel) + atomic.StoreUint32(&initialized, 1) errorLog = newLogWriter(log.New(&buf, "", flags)) l.WithDuration(time.Second).Error(testlog) assert.True(t, strings.Contains(buf.String(), mockTraceId)) @@ -46,6 +48,7 @@ func TestTraceInfo(t *testing.T) { ctx := context.WithValue(context.Background(), tracespec.TracingKey, mock) l := WithContext(ctx).(*traceLogger) SetLevel(InfoLevel) + atomic.StoreUint32(&initialized, 1) infoLog = newLogWriter(log.New(&buf, "", flags)) l.WithDuration(time.Second).Info(testlog) assert.True(t, strings.Contains(buf.String(), mockTraceId)) @@ -61,6 +64,7 @@ func TestTraceSlow(t *testing.T) { ctx := context.WithValue(context.Background(), tracespec.TracingKey, mock) l := WithContext(ctx).(*traceLogger) SetLevel(InfoLevel) + atomic.StoreUint32(&initialized, 1) slowLog = newLogWriter(log.New(&buf, "", flags)) l.WithDuration(time.Second).Slow(testlog) assert.True(t, strings.Contains(buf.String(), mockTraceId)) @@ -75,6 +79,7 @@ func TestTraceWithoutContext(t *testing.T) { var buf mockWriter l := WithContext(context.Background()).(*traceLogger) SetLevel(InfoLevel) + atomic.StoreUint32(&initialized, 1) infoLog = newLogWriter(log.New(&buf, "", flags)) l.WithDuration(time.Second).Info(testlog) assert.False(t, strings.Contains(buf.String(), mockTraceId)) diff --git a/readme.md b/readme.md index 428c2698..c65c2acd 100644 --- a/readme.md +++ b/readme.md @@ -169,4 +169,4 @@ go get -u github.com/tal-tech/go-zero ## 9. 微信交流群 -wechat +wechat