From 553620cc11b2e287e9d8204fc1007d6b2a7522f1 Mon Sep 17 00:00:00 2001
From: Fan Shang Xiang <shafan@microsoft.com>
Date: Fri, 25 Oct 2024 01:40:58 +0000
Subject: [PATCH] add UA for e2e client

---
 test/utils/azure/azure_helper.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/test/utils/azure/azure_helper.go b/test/utils/azure/azure_helper.go
index e23b5fd18..42e5ad287 100644
--- a/test/utils/azure/azure_helper.go
+++ b/test/utils/azure/azure_helper.go
@@ -24,6 +24,7 @@ import (
 
 	"github.com/Azure/azure-sdk-for-go/sdk/azcore/to"
 	resources "github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/resources/armresources"
+	"sigs.k8s.io/blob-csi-driver/pkg/blob"
 	"sigs.k8s.io/cloud-provider-azure/pkg/azclient"
 	"sigs.k8s.io/cloud-provider-azure/pkg/azclient/accountclient"
 	"sigs.k8s.io/cloud-provider-azure/pkg/azclient/identityclient"
@@ -45,8 +46,9 @@ type Client struct {
 
 func GetClient(cloud, subscriptionID, clientID, tenantID, clientSecret string, aadFederatedTokenFile string) (*Client, error) {
 	armConfig := &azclient.ARMClientConfig{
-		Cloud:    cloud,
-		TenantID: tenantID,
+		Cloud:     cloud,
+		TenantID:  tenantID,
+		UserAgent: blob.GetUserAgent(blob.DefaultDriverName, "", "e2e-test"),
 	}
 	useFederatedWorkloadIdentityExtension := false
 	if aadFederatedTokenFile != "" {