测试 gRPC service

木子哥

我想测试一个 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 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}
. l  O9 j, \, y3 W! r

我已经找到了一个例子，但是我找不到任何关于如何在 Go 中实现 gRPC 服务测试示例。
                                                               
+ C6 X" Q/ j/ C& o8 H8 q6 O0 ?    解决方案:                                                               
                                                                我想你正在寻找它google.golang.org/grpc/test/bufconn包帮助您避免使用真实的端口号启动服务，但仍允许测试流式 RPC。

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.}

这种方法的优点是，您仍然可以获得网络行为，但不使用操作系统级资源，如可能或不快速清洁的端口。它允许您以实际方式进行测试，并为您提供适当的流媒体行为。
* A$ t6 Y7 o2 ]4 W! k8 k9 p8 H" O我没有流媒体的例子，但神奇的酱汁就在上面。它为您提供所有正常网络连接的预期行为。诀窍如图所示WithDialer 选项，使用 bufconn 包创建了一个开放自己拨号程序的侦听器。我一直用这种技术来测试 gRPC 服务，而且效果很好。