github.com/argoproj/argo-cd/v3@v3.2.1/util/grpc/useragent_test.go (about)

     1  package grpc
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/Masterminds/semver/v3"
     7  	"github.com/stretchr/testify/require"
     8  	"google.golang.org/grpc/metadata"
     9  )
    10  
    11  func Test_UserAgentEnforcer(t *testing.T) {
    12  	clientName := "argo-cd"
    13  	semverConstraint, _ := semver.NewConstraint("^1")
    14  	t.Run("Test enforcing valid user-agent", func(t *testing.T) {
    15  		md := metadata.New(map[string]string{"user-agent": "argo-cd/1.0"})
    16  		ctx := metadata.NewIncomingContext(t.Context(), md)
    17  		err := userAgentEnforcer(ctx, clientName, semverConstraint)
    18  		require.NoError(t, err)
    19  	})
    20  	t.Run("Test enforcing ignored user-agent", func(t *testing.T) {
    21  		md := metadata.New(map[string]string{"user-agent": "flux/3.0"})
    22  		ctx := metadata.NewIncomingContext(t.Context(), md)
    23  		err := userAgentEnforcer(ctx, clientName, semverConstraint)
    24  		require.NoError(t, err)
    25  	})
    26  	t.Run("Test enforcing user-agent with version not matching constraint", func(t *testing.T) {
    27  		md := metadata.New(map[string]string{"user-agent": "argo-cd/3.0"})
    28  		ctx := metadata.NewIncomingContext(t.Context(), md)
    29  		err := userAgentEnforcer(ctx, clientName, semverConstraint)
    30  		require.ErrorContains(t, err, "unsatisfied client version constraint")
    31  	})
    32  	t.Run("Test legacy user-agent", func(t *testing.T) {
    33  		md := metadata.New(map[string]string{"user-agent": "grpc-go/1.15.0"})
    34  		ctx := metadata.NewIncomingContext(t.Context(), md)
    35  		err := userAgentEnforcer(ctx, clientName, semverConstraint)
    36  		require.ErrorContains(t, err, "unsatisfied client version constraint: ^1")
    37  	})
    38  	t.Run("Test invalid version", func(t *testing.T) {
    39  		md := metadata.New(map[string]string{"user-agent": "argo-cd/super"})
    40  		ctx := metadata.NewIncomingContext(t.Context(), md)
    41  		err := userAgentEnforcer(ctx, clientName, semverConstraint)
    42  		require.ErrorContains(t, err, "could not parse version")
    43  	})
    44  }