micro-tests/codec/segmentio/grpc_test.go
Vasiliy Tolstov 5a63953d3f update for latest micro
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
2021-10-01 01:37:54 +03:00

96 lines
2.5 KiB
Go

package grpc_test
import (
"context"
"testing"
gclient "github.com/unistack-org/micro-client-grpc/v3"
// protocodec "github.com/unistack-org/micro-codec-proto/v3"
protocodec "github.com/unistack-org/micro-codec-segmentio/v3/proto"
regRouter "github.com/unistack-org/micro-router-register/v3"
gserver "github.com/unistack-org/micro-server-grpc/v3"
gpb "github.com/unistack-org/micro-tests/server/grpc/gproto"
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"
)
type testServer struct {
pb.UnimplementedTestServer
}
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
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("127.0.0.1:0"), server.Register(r), server.Name("helloworld"), gserver.Reflection(true))
// create router
rtr := regRouter.NewRouter(router.Register(r))
h := &testServer{}
err = gpb.RegisterTestServer(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)
}
}
//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 {}
}