我想测试用Go编写的gRPC服务。我使用的示例是来自grpc-go repo的Hello World服务器示例。
protobuf的定义如下:
syntax = "proto3"; package helloworld; // The greeting service definition. service Greeter { // Sends a greeting rpc SayHello (HelloRequest) returns (HelloReply) {} } // The request message containing the user's name. message HelloRequest { string name = 1; } // The response message containing the greetings message HelloReply { string message = 1; }
greeter_server主要类型是:
greeter_server
// server is used to implement helloworld.GreeterServer. type server struct{} // SayHello implements helloworld.GreeterServer func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) { return &pb.HelloReply{Message: "Hello " + in.Name}, nil }
我在寻找示例,但找不到有关如何在Go中实现gRPC服务测试的任何示例。
我认为您正在寻找该google.golang.org/grpc/test/bufconn软件包,以帮助您避免使用真实端口号启动服务,但仍允许测试流式RPC。
google.golang.org/grpc/test/bufconn
import "google.golang.org/grpc/test/bufconn" const bufSize = 1024 * 1024 var lis *bufconn.Listener func init() { lis = bufconn.Listen(bufSize) s := grpc.NewServer() pb.RegisterGreeterServer(s, &server{}) go func() { if err := s.Serve(lis); err != nil { log.Fatalf("Server exited with error: %v", err) } }() } func bufDialer(context.Context, string) (net.Conn, error) { return lis.Dial() } func TestSayHello(t *testing.T) { ctx := context.Background() conn, err := grpc.DialContext(ctx, "bufnet", grpc.WithContextDialer(bufDialer), grpc.WithInsecure()) if err != nil { t.Fatalf("Failed to dial bufnet: %v", err) } defer conn.Close() client := pb.NewGreeterClient(conn) resp, err := client.SayHello(ctx, &pb.HelloRequest{"Dr. Seuss"}) if err != nil { t.Fatalf("SayHello failed: %v", err) } log.Printf("Response: %+v", resp) // Test for output here. }
这种方法的好处是,您仍然可以通过网络连接来获得网络行为,但无需使用操作系统级别的资源(例如端口,这些端口可能会很快清除也可能不会很快清除),而是通过内存中的连接。它允许您以实际使用的方式对其进行测试,并为您提供适当的流式传输行为。
我的头上没有流式传输示例,但是魔术酱却在上面。它为您提供了正常网络连接的所有预期行为。诀窍是使用bufconn包来设置所示的WithDialer选项,以使用公开自己的拨号程序的侦听器。我一直在使用这种技术来测试gRPC服务,并且效果很好。