Fix pvc_annotator_test.go

avoltz · May 14, 2024 · 8f46ea3 · 8f46ea3
1 parent fee881b
commit 8f46ea3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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",