chore: let logx.SetWriter can be called anytime (#2186)

2022-07-24 14:15:57 +08:00
parent 685d14e662
commit 9b334b5428
3 changed files with 5 additions and 8 deletions
--- a/core/logx/logs_test.go
+++ b/core/logx/logs_test.go
@@ -603,8 +603,9 @@ func TestSetWriter(t *testing.T) {
 	SetWriter(nopWriter{})
 	assert.NotNil(t, writer.Load())
 	assert.True(t, writer.Load() == nopWriter{})
-	SetWriter(new(mockWriter))
-	assert.True(t, writer.Load() == nopWriter{})
+	mocked := new(mockWriter)
+	SetWriter(mocked)
+	assert.Equal(t, mocked, writer.Load())
 }

 func TestWithGzip(t *testing.T) {