Function checkRPCSendOK

balancer/ringhash/ringhash_e2e_test.go:243–255

checkRPCSendOK sends num RPCs to the client. It returns a map of backend addresses as keys and number of RPCs sent to this address as value. Abort the test if any RPC fails.

(ctx context.Context, t *testing.T, client testgrpc.TestServiceClient, num int)

// addresses as keys and number of RPCs sent to this address as value. Abort the
// test if any RPC fails.
func checkRPCSendOK(ctx context.Context, t *testing.T, client testgrpc.TestServiceClient, num int) map[string]int {
	t.Helper()

	backendCount := make(map[string]int)
	for i := 0; i < num; i++ {
		var remote peer.Peer
		if _, err := client.EmptyCall(ctx, &testpb.Empty{}, grpc.Peer(&remote)); err != nil {
			t.Fatalf("rpc EmptyCall() failed: %v", err)
		}
		backendCount[remote.Addr.String()]++
	}
	return backendCount
}

// makeUnreachableBackends returns a slice of addresses of backends that close
// connections as soon as they are established. Useful to simulate servers that