fix

Signed-off-by: danish siddiqui <[email protected]>

cmd/agent/app/configmanager/grpc/manager.go

@@ -17,6 +17,7 @@ package grpc
 import (
 	"context"
 	"errors"
+	"fmt"
 
 	"google.golang.org/grpc"
 
@@ -38,7 +39,11 @@ func NewConfigManager(conn *grpc.ClientConn) *ConfigManagerProxy {
 
 // GetSamplingStrategy returns sampling strategies from collector.
 func (s *ConfigManagerProxy) GetSamplingStrategy(ctx context.Context, serviceName string) (*api_v2.SamplingStrategyResponse, error) {
-	return s.client.GetSamplingStrategy(ctx, &api_v2.SamplingStrategyParameters{ServiceName: serviceName})
+	resp, err := s.client.GetSamplingStrategy(ctx, &api_v2.SamplingStrategyParameters{ServiceName: serviceName})
+	if err != nil {
+		return nil, fmt.Errorf("our internal message/explanation: %w", err)
+	}
+	return resp, nil
 }
 
 // GetBaggageRestrictions returns baggage restrictions from collector.

cmd/agent/app/configmanager/grpc/manager_test.go

@@ -55,7 +55,7 @@ func TestSamplingManager_GetSamplingStrategy_error(t *testing.T) {
 	resp, err := manager.GetSamplingStrategy(context.Background(), "any")
 	require.Nil(t, resp)
 	require.Error(t, err)
-	assert.Contains(t, err.Error(), "Error while dialing")
+	assert.Contains(t, err.Error(), "our internal message/explanation")
 }
 
 func TestSamplingManager_GetBaggageRestrictions(t *testing.T) {

cmd/agent/app/processors/thrift_processor_test.go

@@ -82,7 +82,6 @@ func createProcessor(t *testing.T, mFactory metrics.Factory, tFactory thrift.TPr
 func initCollectorAndReporter(t *testing.T) (*metricstest.Factory, *testutils.GrpcCollector, reporter.Reporter, *grpc.ClientConn) {
 	grpcCollector := testutils.StartGRPCCollector(t)
 	conn, err := grpc.NewClient(grpcCollector.Listener().Addr().String(), grpc.WithTransportCredentials(insecure.NewCredentials()))
-	grpc.NewClient(grpcCollector.Listener().Addr().String(), grpc.WithTransportCredentials(insecure.NewCredentials()))
 	require.NoError(t, err)
 	rep := grpcrep.NewReporter(conn, map[string]string{}, zaptest.NewLogger(t))
 	metricsFactory := metricstest.NewFactory(0)