インメモリ接続を使用して Go で gRPC サービスを効率的にテストする方法

Go での gRPC サービスのテスト

概要

gRPC、リモート プロシージャ コール (RPC) の作成に広く使用されているフレームワーク、サービス間通信のための堅牢なメカニズムを提供します。 gRPC サービスの信頼性と正確性を確保するには、テストが重要な役割を果たします。この記事では、インメモリ接続を使用して Go で記述された gRPC サービスのテストについて説明します。

Bufconn パッケージの使用

テスト中に実際のネットワーク ポートを使用する複雑さを回避するにはgRPC サービスには、google.golang.org/grpc/test/bufconn パッケージが推奨されます。このパッケージを使用すると、オペレーティング システムを介さずにネットワーク動作をエミュレートするメモリ内リスナーを作成できます。

コード例

次のプロト定義を持つ gRPC サービスを考えてみましょう。およびサーバー実装:

syntax = "proto3";

package helloworld;

service Greeter {
  rpc SayHello (HelloRequest) returns (HelloReply) {}
}

message HelloRequest {
  string name = 1;
}

message HelloReply {
  string message = 1;
}

// SayHello implements GreeterServer
func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) {
    return &pb.HelloReply{Message: "Hello " + in.Name}, nil
}

テスト実装

SayHello 関数をテストするには、bufconn を使用してテスト関数を作成できます。

func TestSayHello(t *testing.T) {
    ctx := context.Background()
    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)
        }
    }()
    defer s.GracefulStop()
    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)
    // Perform additional tests on the response.
}

このテストは、メモリ内リスナーを初期化し、gRPC サーバーを起動し、 bufconn のダイヤラーを使用したサーバーへの接続。次に、SayHello 関数を呼び出し、応答の正しさをアサートします。

Bufconn を使用する利点

bufconn を使用すると、gRPC サービスのテストにいくつかの利点があります。

• ネットワークの動作をエミュレートし、現実的なテストを提供します
• 特定のポートや外部依存関係を必要とせずにテストできます。
• ストリーミング RPC をサポートし、ストリーミング機能の徹底的なテストを可能にします。
• テストのセットアップとティアダウンを簡素化し、コストを削減します。定型コード。

以上がインメモリ接続を使用して Go で gRPC サービスを効率的にテストする方法の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。