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.
go-zero/rest/internal/log_test.go

45 lines
1.2 KiB
Go

4 years ago
package internal
4 years ago
import (
"context"
"net/http"
"net/http/httptest"
"strings"
"testing"
"github.com/stretchr/testify/assert"
"github.com/zeromicro/go-zero/core/logx/logtest"
4 years ago
)
func TestInfo(t *testing.T) {
collector := new(LogCollector)
req := httptest.NewRequest(http.MethodGet, "http://localhost", http.NoBody)
req = req.WithContext(WithLogCollector(req.Context(), collector))
4 years ago
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) {
c := logtest.NewCollector(t)
req := httptest.NewRequest(http.MethodGet, "http://localhost", http.NoBody)
4 years ago
Error(req, "first")
Errorf(req, "second %s", "third")
val := c.String()
4 years ago
assert.True(t, strings.Contains(val, "first"))
assert.True(t, strings.Contains(val, "second"))
assert.True(t, strings.Contains(val, "third"))
}
func TestLogCollectorContext(t *testing.T) {
ctx := context.Background()
assert.Nil(t, LogCollectorFromContext(ctx))
collector := new(LogCollector)
ctx = WithLogCollector(ctx, collector)
assert.Equal(t, collector, LogCollectorFromContext(ctx))
}