package grpc_test import ( "context" "fmt" "io" "net" "testing" "time" grpc "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" pb "github.com/unistack-org/micro-tests/client/grpc/proto" "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" pgrpc "google.golang.org/grpc" "google.golang.org/grpc/codes" "google.golang.org/grpc/status" ) type testServer struct { pb.UnimplementedTestServer } func (g *testServer) Call(ctx context.Context, in *pb.Request) (*pb.Response, error) { if in.Name == "Error" { return nil, &errors.Error{Id: "id", Code: 99, Detail: "detail"} } return &pb.Response{Msg: "Hello " + in.Name}, nil } func (g *testServer) Stream(stream pb.Test_StreamServer) error { rsp := &pb.Response{} for { req, err := stream.Recv() if err != nil && err == io.EOF { break } else if err != nil { return err } rsp.Msg = req.Name if err = stream.Send(rsp); err != nil { return err } time.Sleep(200 * time.Millisecond) } return nil } func TestGRPCClient(t *testing.T) { ctx := context.Background() l, err := net.Listen("tcp", "127.0.0.1:0") if err != nil { t.Fatalf("failed to listen: %v", err) } defer l.Close() s := pgrpc.NewServer() pb.RegisterTestServer(s, &testServer{}) go func() { if err := s.Serve(l); err != nil { t.Log(err) } }() defer s.Stop() // create mock register r := register.NewRegister() // register service if err := r.Register(ctx, ®ister.Service{ Name: "helloworld", Version: "test", Nodes: []*register.Node{ { Id: "test-1", Address: l.Addr().String(), Metadata: map[string]string{ "protocol": "grpc", }, }, }, }); err != nil { t.Fatal(err) } // create router rtr := regRouter.NewRouter(router.Register(r)) // create client c := grpc.NewClient(client.Codec("application/grpc+proto", protocodec.NewCodec()), client.Router(rtr)) testMethods := []string{ "/helloworld.Test/Call", "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.Fatal(err) } if rsp.Msg != "Hello John" { t.Fatalf("Got unexpected response %v", rsp.Msg) } } for _, method := range testMethods { req := c.NewRequest("helloworld", method, &pb.Request{ Name: "Error", }) rsp := pb.Response{} err = c.Call(context.TODO(), req, &rsp) if err == nil { t.Fatal("nil error received") } verr, ok := err.(*errors.Error) if !ok { t.Fatalf("invalid error received %#+v\n", err) } if verr.Code != 99 && verr.Id != "id" && verr.Detail != "detail" { t.Fatalf("invalid error received %#+v\n", verr) } } ctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond) defer cancel() req := c.NewRequest("helloworld", "Test.Stream", &pb.Request{}, client.StreamingRequest()) stream, err := c.Stream(ctx, req) if err != nil { t.Fatal(err) } defer stream.Close() go func() { for i := 0; i < 5; i++ { fmt.Printf("send to stream\n") if err = stream.Send(&pb.Request{Name: "test name"}); err != nil { t.Fatal(err) } } }() rsp := &pb.Response{} for i := 0; i < 5; i++ { fmt.Printf("recv from stream\n") if err = stream.Recv(rsp); err != nil { st, ok := status.FromError(err) if !ok { t.Fatalf("%v", err) } if st.Code() != codes.DeadlineExceeded { t.Fatalf("%v", err) } } if rsp.Msg != "test name" { t.Fatalf("invalid msg: %v", rsp) } } }