测试 gRPC service

木子哥

我想测试一个 Go 编写的 gRPC 服务。我用的例子来自grpc-go repo的 Hello World 服务器示例。
/ x/ Y7 U9 H# ^. z0 i3 F4 aprotobuf 定义如下：

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 greetingsmessage HelloReply {  string message = 1;}

greeter_server主要类型为：

// server is used to implement helloworld.GreeterServer.type server struct{}// SayHello implements helloworld.GreeterServerfunc (s *server) SayHello(ctx context.Context,in *pb.HelloRequest) (*pb.HelloReply,error)    return &pb.HelloReply{Message: "Hello "   in.Name},nil}

我已经找到了一个例子，但是我找不到任何关于如何在 Go 中实现 gRPC 服务测试示例。
% t$ K1 X% V2 r$ ]% i                                                               
+ h9 i) `1 h. S/ @    解决方案:                                                               
                                                                我想你正在寻找它google.golang.org/grpc/test/bufconn包帮助您避免使用真实的端口号启动服务，但仍允许测试流式 RPC。
. p$ \7 P6 q4 W3 L" ^" q7 ~0 `

import "google.golang.org/grpc/test/bufconn"const bufSize = 1024 * 1024var lis *bufconn.Listenerfunc 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.}
3 ?5 ]! }( J/ A  B

这种方法的优点是，您仍然可以获得网络行为，但不使用操作系统级资源，如可能或不快速清洁的端口。它允许您以实际方式进行测试，并为您提供适当的流媒体行为。
- Q, ~3 q) ~; J2 I% y! O4 A4 q$ f7 V# {我没有流媒体的例子，但神奇的酱汁就在上面。它为您提供所有正常网络连接的预期行为。诀窍如图所示WithDialer 选项，使用 bufconn 包创建了一个开放自己拨号程序的侦听器。我一直用这种技术来测试 gRPC 服务，而且效果很好。