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

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

DDD

リリース： 2024-12-26 01:26:11

オリジナル

520 人が閲覧しました

How to Efficiently Test gRPC Services in Go Using In-Memory Connections?

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 サービスのテストにいくつかの利点があります。