From 8f46ea3b3afe21f6449cb211bf64e0ee92a6757e Mon Sep 17 00:00:00 2001 From: Stephen Mkandawire Date: Tue, 14 May 2024 15:16:08 -0400 Subject: [PATCH] Fix pvc_annotator_test.go --- pkg/blob/nodeserver.go | 2 +- pkg/edgecache/cachevolume/pvc_annotator_test.go | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/pkg/blob/nodeserver.go b/pkg/blob/nodeserver.go index cd5e5a4ca..4fbdc9b4b 100644 --- a/pkg/blob/nodeserver.go +++ b/pkg/blob/nodeserver.go @@ -29,13 +29,13 @@ import ( "sigs.k8s.io/blob-csi-driver/pkg/edgecache" cv "sigs.k8s.io/blob-csi-driver/pkg/edgecache/cachevolume" blobcsiutil "sigs.k8s.io/blob-csi-driver/pkg/util" + volumehelper "sigs.k8s.io/blob-csi-driver/pkg/util" azcache "sigs.k8s.io/cloud-provider-azure/pkg/cache" "sigs.k8s.io/cloud-provider-azure/pkg/metrics" "github.com/container-storage-interface/spec/lib/go/csi" v1 "k8s.io/api/core/v1" - "k8s.io/apimachinery/pkg/util/wait" "k8s.io/klog/v2" "k8s.io/kubernetes/pkg/volume" "k8s.io/kubernetes/pkg/volume/util" diff --git a/pkg/edgecache/cachevolume/pvc_annotator_test.go b/pkg/edgecache/cachevolume/pvc_annotator_test.go index 2b293cfc4..c57a0214f 100644 --- a/pkg/edgecache/cachevolume/pvc_annotator_test.go +++ b/pkg/edgecache/cachevolume/pvc_annotator_test.go @@ -25,6 +25,7 @@ import ( "k8s.io/client-go/kubernetes/fake" "github.com/stretchr/testify/assert" + "sigs.k8s.io/cloud-provider-azure/pkg/azclient" "sigs.k8s.io/cloud-provider-azure/pkg/provider/config" ) @@ -152,7 +153,7 @@ func TestSendProvisionVolumeSuccess(t *testing.T) { testcases := []SuccessTestCase{ { name: "ValidWIAuth", - config: config.AzureAuthConfig{UseFederatedWorkloadIdentityExtension: true}, + config: config.AzureAuthConfig{AzureAuthConfig: azclient.AzureAuthConfig{UseFederatedWorkloadIdentityExtension: true}}, blobAuth: NewBlobAuth(suffix, acct, container, "", "", "WorkloadIdentity"), expectedAnnotations: map[string]string{ volumeStateAnnotation: "not created",