测试 gRPC service

木子哥

我想测试一个 Go 编写的 gRPC 服务。我用的例子来自grpc-go repo的 Hello World 服务器示例。
# q0 B+ S! j1 m! w* }. [# u! {+ {protobuf 定义如下：
; y+ r5 {, e- q) U8 g

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;}
4 L+ T/ G) n  X& u0 A% x4 `

greeter_server主要类型为：
3 @/ R6 Q  _8 P+ r' T( K4 `9 @

// 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 服务测试示例。
9 r+ f4 r$ X( ^; p$ G                                                               
4 ?$ }3 O5 o, X! i( L0 R    解决方案:                                                               
. b5 t/ }' T  n                                                                我想你正在寻找它google.golang.org/grpc/test/bufconn包帮助您避免使用真实的端口号启动服务，但仍允许测试流式 RPC。
, P( A! X- L! S; i) {9 }" w! T

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.}
  j, a9 N" A8 h! p0 O& q

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