mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-24 09:40:24 +08:00
ac321fc146
* backup * simplify * chore: remove unused pool * chore: fix lint errors * chore: use strings.Builder instead of bytes.Buffer * test: add more tests * chore: fix reviewdog * test: fix data race * feat: make logger customizable * chore: fix reviewdog * test: fix fails * chore: fix set writer twice * chore: use context instead of golang.org context * chore: specify uint32 for level types
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
func TestInfo(t *testing.T) {
|
|
collector := new(LogCollector)
|
|
req := httptest.NewRequest(http.MethodGet, "http://localhost", nil)
|
|
req = req.WithContext(context.WithValue(req.Context(), LogContext, collector))
|
|
Info(req, "first")
|
|
Infof(req, "second %s", "third")
|
|
val := collector.Flush()
|
|
assert.True(t, strings.Contains(val, "first"))
|
|
assert.True(t, strings.Contains(val, "second"))
|
|
assert.True(t, strings.Contains(val, "third"))
|
|
assert.True(t, strings.Contains(val, "\n"))
|
|
}
|
|
|
|
func TestError(t *testing.T) {
|
|
var buf strings.Builder
|
|
w := logx.NewWriter(&buf)
|
|
o := logx.Reset()
|
|
logx.SetWriter(w)
|
|
|
|
defer func() {
|
|
logx.Reset()
|
|
logx.SetWriter(o)
|
|
}()
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "http://localhost", nil)
|
|
Error(req, "first")
|
|
Errorf(req, "second %s", "third")
|
|
val := buf.String()
|
|
assert.True(t, strings.Contains(val, "first"))
|
|
assert.True(t, strings.Contains(val, "second"))
|
|
assert.True(t, strings.Contains(val, "third"))
|
|
}
|
|
|
|
func TestContextKey_String(t *testing.T) {
|
|
val := contextKey("foo")
|
|
assert.True(t, strings.Contains(val.String(), "foo"))
|
|
}
|