update tests

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
2021-03-06 22:24:49 +03:00
parent ba7d0703a3
commit 0cde3df915
5 changed files with 24 additions and 26 deletions

View File

@@ -48,7 +48,7 @@ func TestGRPCServer(t *testing.T) {
r := register.NewRegister()
b := broker.NewBroker(broker.Register(r))
s := gserver.NewServer(server.Codec("application/grpc+proto", protocodec.NewCodec()), server.Address(":12345"), server.Register(r), server.Name("helloworld"), gserver.Reflection(true),
s := gserver.NewServer(server.Codec("application/grpc+proto", protocodec.NewCodec()), server.Address(":0"), server.Register(r), server.Name("helloworld"), gserver.Reflection(true),
server.WrapHandler(NewServerHandlerWrapper()),
)
// create router

View File

@@ -51,6 +51,9 @@ func (h *Handler) Call(ctx context.Context, req *pb.CallReq, rsp *pb.CallRsp) er
}
rsp.Rsp = "name_my_name"
httpsrv.SetRspCode(ctx, http.StatusCreated)
md = metadata.New(1)
md.Set("my-key", "my-val")
metadata.SetOutgoingContext(ctx, md)
return nil
}
@@ -185,6 +188,10 @@ func TestNativeServer(t *testing.T) {
t.Fatalf("Expected response %s, got %s", `{"rsp":"name_my_name"}`, s)
}
if v := rsp.Header.Get("My-Key"); v != "my-val" {
t.Fatalf("empty response header: %#+v", rsp.Header)
}
// make request with error
rsp, err = http.Post(fmt.Sprintf("http://%s/v1/test/callerror/my_name", service[0].Nodes[0].Address), "application/json", nil)
if err != nil {