consul/test/integration/consul-container/libs/assert/grpc.go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: BUSL-1.1

package assert

import (
	"context"
	"testing"
	"time"

	"fortio.org/fortio/fgrpc"
	"github.com/hashicorp/consul/sdk/testutil/retry"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
)

// GRPCPing sends a fgrpc.PingMessage to a fortio server at addr, analogous to
// the CLI command `fortio grpcping`. It retries for up to 1m, with a 25ms gap.
func GRPCPing(t *testing.T, addr string) {
	t.Helper()
	pingConn, err := grpc.Dial(addr, grpc.WithTransportCredentials(insecure.NewCredentials()))
	require.NoError(t, err)
	pingCl := fgrpc.NewPingServerClient(pingConn)
	var msg *fgrpc.PingMessage
	retries := 0
	retry.RunWith(&retry.Timer{Timeout: time.Minute, Wait: 25 * time.Millisecond}, t, func(r *retry.R) {
		t.Logf("making grpc call to %s", addr)
		retries += 1
		msg, err = pingCl.Ping(context.Background(), &fgrpc.PingMessage{
			// use addr as payload so we have something variable to check against
			Payload: addr,
		})
		if err != nil {
			r.Error(err)
		}
	})
	assert.Equal(t, addr, msg.Payload)
}