67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
|
package grpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/micro/go-micro/registry/memory"
|
||
|
|
"github.com/micro/go-micro/server"
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
|
||
|
|
pb "github.com/micro/examples/greeter/srv/proto/hello"
|
||
|
|
)
|
||
|
|
|
||
|
|
// server is used to implement helloworld.GreeterServer.
|
||
|
|
type sayServer struct{}
|
||
|
|
|
||
|
|
// SayHello implements helloworld.GreeterServer
|
||
|
|
func (s *sayServer) Hello(ctx context.Context, req *pb.Request, rsp *pb.Response) error {
|
||
|
|
rsp.Msg = "Hello " + req.Name
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGRPCServer(t *testing.T) {
|
||
|
|
r := memory.NewRegistry()
|
||
|
|
s := NewServer(
|
||
|
|
server.Name("foo"),
|
||
|
|
server.Registry(r),
|
||
|
|
)
|
||
|
|
|
||
|
|
pb.RegisterSayHandler(s, &sayServer{})
|
||
|
|
|
||
|
|
if err := s.Start(); err != nil {
|
||
|
|
t.Fatalf("failed to start: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// check registration
|
||
|
|
services, err := r.GetService("foo")
|
||
|
|
if err != nil || len(services) == 0 {
|
||
|
|
t.Fatalf("failed to get service: %v # %d", err, len(services))
|
||
|
|
}
|
||
|
|
|
||
|
|
defer func() {
|
||
|
|
if err := s.Stop(); err != nil {
|
||
|
|
t.Fatalf("failed to stop: %v", err)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
cc, err := grpc.Dial(s.Options().Address, grpc.WithInsecure())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to dial server: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
testMethods := []string{"/helloworld.Say/Hello", "/greeter.helloworld.Say/Hello"}
|
||
|
|
|
||
|
|
for _, method := range testMethods {
|
||
|
|
rsp := pb.Response{}
|
||
|
|
|
||
|
|
if err := cc.Invoke(context.Background(), method, &pb.Request{Name: "John"}, &rsp); err != nil {
|
||
|
|
t.Fatalf("error calling server: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if rsp.Msg != "Hello John" {
|
||
|
|
t.Fatalf("Got unexpected response %v", rsp.Msg)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|