unistack-org/micro-tests
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

server/http: add metadata wrapper test

Browse Source 
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов
2021-04-30 00:44:22 +03:00
parent d00792b4a8
commit e6acf99ce0
3 changed files with 22 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
server/http/http_test.go
View File

@@ -130,10 +130,14 @@ func TestMultipart(t *testing.T) {
}
}
func NewServerHandlerWrapper() server.HandlerWrapper {
func NewServerHandlerWrapper(t *testing.T) server.HandlerWrapper {
return func(fn server.HandlerFunc) server.HandlerFunc {
return func(ctx context.Context, req server.Request, rsp interface{}) error {
//fmt.Printf("wrap ctx: %s\n", req.Service())
md, ok := metadata.FromIncomingContext(ctx)
if !ok {
t.Fatal("metadata empty")
}
t.Logf("md: %v", md)
return fn(ctx, req, rsp)
}
}
@@ -202,7 +206,7 @@ func TestNativeFormUrlencoded(t *testing.T) {
server.Register(reg),
server.Codec("application/json", jsoncodec.NewCodec()),
server.Codec("application/x-www-form-urlencoded", urlencodecodec.NewCodec()),
//server.WrapHandler(NewServerHandlerWrapper()),
//server.WrapHandler(NewServerHandlerWrapper()),
)
if err := srv.Init(); err != nil {
@@ -323,7 +327,7 @@ func TestNativeClientServer(t *testing.T) {
server.WrapHandler(mwrapper.NewHandlerWrapper(mwrapper.Meter(m))),
server.WrapHandler(lwrapper.NewServerHandlerWrapper(lwrapper.WithEnabled(false), lwrapper.WithLevel(logger.ErrorLevel))),
httpsrv.Middleware(mwf),
server.WrapHandler(NewServerHandlerWrapper()),
server.WrapHandler(NewServerHandlerWrapper(t)),
)
h := &Handler{t: t}
@@ -459,7 +463,7 @@ func TestNativeServer(t *testing.T) {
server.Register(reg),
server.Codec("application/json", jsoncodec.NewCodec()),
server.Codec("application/x-www-form-urlencoded", urlencodecodec.NewCodec()),
//server.WrapHandler(NewServerHandlerWrapper()),
server.WrapHandler(NewServerHandlerWrapper(t)),
)
h := &Handler{t: t}
@@ -485,10 +489,17 @@ func TestNativeServer(t *testing.T) {
}
// make request
rsp, err := http.Post(fmt.Sprintf("http://%s/v1/test/call/my_name?req=key&arg1=arg1&arg2=12345&nested.string_args=str1&nested.string_args=str2&nested.uint64_args=1&nested.uint64_args=2&nested.uint64_args=3", service[0].Nodes[0].Address), "application/json", nil)
req, err := http.NewRequest("POST", fmt.Sprintf("http://%s/v1/test/call/my_name?req=key&arg1=arg1&arg2=12345&nested.string_args=str1&nested.string_args=str2&nested.uint64_args=1&nested.uint64_args=2&nested.uint64_args=3", service[0].Nodes[0].Address), nil)
if err != nil {
t.Fatal(err)
}
req.Header.Set("Authorization", "test")
req.Header.Set("Content-Type", "application/json")
rsp, err := http.DefaultClient.Do(req)
if err != nil {
t.Fatal(err)
}
if rsp.StatusCode != http.StatusCreated {
t.Fatalf("invalid status received: %#+v\n", rsp)
}