You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.2 KiB
77 lines
2.2 KiB
3 years ago
|
package logging
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/go-kratos/kratos/v2/log"
|
||
|
|
"github.com/go-kratos/kratos/v2/middleware"
|
||
|
|
"github.com/go-kratos/kratos/v2/transport"
|
||
|
|
"github.com/go-kratos/kratos/v2/transport/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestHTTP(t *testing.T) {
|
||
|
|
var req = httptest.NewRequest("GET", "http://example.com/foo", nil)
|
||
|
|
var err = errors.New("reply.error")
|
||
|
|
var bf = bytes.NewBuffer(nil)
|
||
|
|
var logger = log.NewStdLogger(bf)
|
||
|
|
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
kind func(logger log.Logger) middleware.Middleware
|
||
|
|
err error
|
||
|
|
ctx context.Context
|
||
|
|
}{
|
||
|
|
{"http-server@fail",
|
||
|
|
Server,
|
||
|
|
err,
|
||
|
|
func() context.Context {
|
||
|
|
res := httptest.NewRecorder()
|
||
|
|
ctx := transport.NewContext(context.Background(), transport.Transport{Kind: transport.KindHTTP, Endpoint: "endpoint"})
|
||
|
|
return http.NewServerContext(ctx, http.ServerInfo{Request: req, Response: res})
|
||
|
|
}(),
|
||
|
|
},
|
||
|
|
{"http-server@succ",
|
||
|
|
Server,
|
||
|
|
nil,
|
||
|
|
func() context.Context {
|
||
|
|
res := httptest.NewRecorder()
|
||
|
|
ctx := transport.NewContext(context.Background(), transport.Transport{Kind: transport.KindHTTP, Endpoint: "endpoint"})
|
||
|
|
return http.NewServerContext(ctx, http.ServerInfo{Request: req, Response: res})
|
||
|
|
}(),
|
||
|
|
},
|
||
|
|
{"http-client@succ",
|
||
|
|
Client,
|
||
|
|
nil,
|
||
|
|
func() context.Context {
|
||
|
|
ctx := transport.NewContext(context.Background(), transport.Transport{Kind: transport.KindHTTP, Endpoint: "endpoint"})
|
||
|
|
return http.NewClientContext(ctx, http.ClientInfo{Request: req, PathPattern: "{name}"})
|
||
|
|
}(),
|
||
|
|
},
|
||
|
|
{"http-client@fail",
|
||
|
|
Client,
|
||
|
|
err,
|
||
|
|
func() context.Context {
|
||
|
|
ctx := transport.NewContext(context.Background(), transport.Transport{Kind: transport.KindHTTP, Endpoint: "endpoint"})
|
||
|
|
return http.NewClientContext(ctx, http.ClientInfo{Request: req, PathPattern: "{name}"})
|
||
|
|
}(),
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, test := range tests {
|
||
|
|
t.Run(test.name, func(t *testing.T) {
|
||
|
|
bf.Reset()
|
||
|
|
next := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||
|
|
return "reply", test.err
|
||
|
|
}
|
||
|
|
next = test.kind(logger)(next)
|
||
|
|
v, e := next(test.ctx, "req.args")
|
||
|
|
t.Logf("[%s]reply: %v, error: %v", test.name, v, e)
|
||
|
|
t.Logf("[%s]buffer:%s", test.name, bf.String())
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|