114 lines
2.9 KiB
Go
114 lines
2.9 KiB
Go
package grpc_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
gclient "github.com/unistack-org/micro-client-grpc/v3"
|
|
protocodec "github.com/unistack-org/micro-codec-proto/v3"
|
|
regRouter "github.com/unistack-org/micro-router-register/v3"
|
|
gserver "github.com/unistack-org/micro-server-grpc/v3"
|
|
pb "github.com/unistack-org/micro-tests/server/grpc/proto"
|
|
"github.com/unistack-org/micro/v3/broker"
|
|
"github.com/unistack-org/micro/v3/client"
|
|
"github.com/unistack-org/micro/v3/errors"
|
|
"github.com/unistack-org/micro/v3/register"
|
|
"github.com/unistack-org/micro/v3/router"
|
|
"github.com/unistack-org/micro/v3/server"
|
|
jsonpb "google.golang.org/protobuf/encoding/protojson"
|
|
)
|
|
|
|
type testServer struct {
|
|
pb.UnimplementedTestServer
|
|
}
|
|
|
|
func NewServerHandlerWrapper() server.HandlerWrapper {
|
|
return func(fn server.HandlerFunc) server.HandlerFunc {
|
|
return func(ctx context.Context, req server.Request, rsp interface{}) error {
|
|
fmt.Printf("wrap ctx: %#+v req: %#+v\n", ctx, req)
|
|
return fn(ctx, req, rsp)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (g *testServer) Call(ctx context.Context, req *pb.Request, rsp *pb.Response) error {
|
|
if req.Name == "Error" {
|
|
return &errors.Error{Id: "id", Code: 99, Detail: "detail"}
|
|
}
|
|
rsp.Msg = "Hello " + req.Name
|
|
rsp.Broken = &pb.Broken{Field: "12345"}
|
|
|
|
return nil
|
|
}
|
|
|
|
func TestGRPCServer(t *testing.T) {
|
|
var err error
|
|
|
|
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),
|
|
server.WrapHandler(NewServerHandlerWrapper()),
|
|
)
|
|
// create router
|
|
rtr := regRouter.NewRouter(router.Register(r))
|
|
|
|
h := &testServer{}
|
|
err = pb.RegisterTestHandler(s, h)
|
|
if err != nil {
|
|
t.Fatalf("can't register handler: %v", err)
|
|
}
|
|
|
|
if err = s.Init(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err = s.Start(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
defer func() {
|
|
if err = s.Stop(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
// create client
|
|
c := gclient.NewClient(client.Codec("application/grpc+proto", protocodec.NewCodec()), client.Router(rtr), client.Register(r), client.Broker(b))
|
|
|
|
testMethods := []string{
|
|
"Test.Call",
|
|
}
|
|
|
|
for _, method := range testMethods {
|
|
req := c.NewRequest("helloworld", method, &pb.Request{
|
|
Name: "John",
|
|
})
|
|
|
|
rsp := &pb.Response{}
|
|
|
|
err = c.Call(context.TODO(), req, rsp)
|
|
if err != nil {
|
|
t.Fatalf("method: %s err: %v", method, err)
|
|
}
|
|
|
|
if rsp.Msg != "Hello John" {
|
|
t.Fatalf("Got unexpected response %v", rsp.Msg)
|
|
}
|
|
|
|
enc := &jsonpb.MarshalOptions{EmitUnpopulated: true}
|
|
_, err := enc.Marshal(rsp)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
//rsp := rpb.ServerReflectionResponse{}
|
|
//req := c.NewRequest("helloworld", "Test.ServerReflectionInfo", &rpb.ServerReflectionRequest{}, client.StreamingRequest())
|
|
//if err := c.Call(context.TODO(), req, &rsp); err != nil {
|
|
// t.Fatal(err)
|
|
//}
|
|
|
|
// select {}
|
|
}
|