Add gRPC v1 Reflection API support for client

Author: umalchanau

fixture/mock_grpc_server_interceptor.go

@@ -19,7 +19,7 @@ func NewCallStats() *CallStats {
 	}
 }
 
-func (cs *CallStats) UnaryInterceptor() grpc.UnaryServerInterceptor {
+func (callStats *CallStats) UnaryInterceptor() grpc.UnaryServerInterceptor {
 	return func(
 		ctx context.Context,
 		req any,
@@ -28,10 +28,10 @@ func (cs *CallStats) UnaryInterceptor() grpc.UnaryServerInterceptor {
 	) (any, error) {
 		resp, err := handler(ctx, req)
 
-		cs.mu.Lock()
-		defer cs.mu.Unlock()
+		callStats.mu.Lock()
+		defer callStats.mu.Unlock()
 
-		cs.StatusesByMethod[info.FullMethod] = append(cs.StatusesByMethod[info.FullMethod], status.Convert(err))
+		callStats.StatusesByMethod[info.FullMethod] = append(callStats.StatusesByMethod[info.FullMethod], status.Convert(err))
 
 		return resp, err
 	}

fixture/mock_server.go

@@ -40,12 +40,12 @@ func StartGrpcTargetTestServerReflectionV1Alpha(callStats *CallStats) (*grpc.Ser
 }
 
 // It uses the test.proto from grpc-testing: https://github.com/grpc/grpc-go/blob/40a879c23a0dc77234d17e0699d074d5fd151bd0/test/grpc_testing/test.proto
-func startGrpcTargetTestServer(callStats *CallStats, reflRegFun func(*grpc.Server)) (*grpc.Server, int) {
+func startGrpcTargetTestServer(callStats *CallStats, reflRegFunc func(*grpc.Server)) (*grpc.Server, int) {
 	server := grpc.NewServer(
 		grpc.UnaryInterceptor(callStats.UnaryInterceptor()),
 	)
 	grpc_testing.RegisterTestServiceServer(server, &grpc_testing.UnimplementedTestServiceServer{})
-	reflRegFun(server)
+	reflRegFunc(server)
 
 	listener, err := net.Listen("tcp", ":0")
 	if err != nil {

internal/pkg/grpc/client.go

@@ -18,7 +18,6 @@ import (
 	"bytes"
 	"errors"
 	"fmt"
-	"google.golang.org/grpc/credentials/insecure"
 	"log"
 	"mittens/internal/pkg/placeholders"
 	"mittens/internal/pkg/response"
@@ -31,8 +30,8 @@ import (
 	"golang.org/x/net/context"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
+	"google.golang.org/grpc/credentials/insecure"
 	"google.golang.org/grpc/metadata"
-	reflectpb "google.golang.org/grpc/reflection/grpc_reflection_v1alpha"
 )
 
 // Client represents a gRPC client.
@@ -85,7 +84,7 @@ func (c *Client) Connect(headers []string) error {
 		return fmt.Errorf("gRPC dial: %v", err)
 	}
 
-	reflectionClient := grpcreflect.NewClientV1Alpha(contextWithMetadata, reflectpb.NewServerReflectionClient(conn))
+	reflectionClient := grpcreflect.NewClientAuto(contextWithMetadata, conn)
 	descriptorSource := grpcurl.DescriptorSourceFromServer(contextWithMetadata, reflectionClient)
 
 	log.Print("gRPC client connected")

test/root_test.go

@@ -161,10 +161,10 @@ func TestHttp(t *testing.T) {
 	assert.True(t, readyFileExists)
 }
 
-func TestGrpcAndHttpWithVariousGrpcServerReflectionAPICombinations(t *testing.T) {
+func TestGrpcAndHttpWithVariousReflectionAPICombinations(t *testing.T) {
 	testConfigs := []struct {
-		name            string
-		setupGrpcServer func(*fixture.CallStats) (*grpc.Server, int)
+		name      string
+		setupFunc func(*fixture.CallStats) (*grpc.Server, int)
 	}{
 		{
 			"Run with v1 and v1alpha Reflection API support",
@@ -190,9 +190,9 @@ func TestGrpcAndHttpWithVariousGrpcServerReflectionAPICombinations(t *testing.T)
 			var server *grpc.Server
 			var port int
 
-			if testConfig.setupGrpcServer != nil {
+			if testConfig.setupFunc != nil {
 				callStats = fixture.NewCallStats()
-				server, port = testConfig.setupGrpcServer(callStats)
+				server, port = testConfig.setupFunc(callStats)
 				defer server.GracefulStop()
 			} else {
 				callStats = grpcCallStats