You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
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", http.NoBody)
|
|
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", http.NoBody)
|
|
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"))
|
|
}
|