@@ -31,6 +31,7 @@ import (
3131 csiv1alpha1 "k8s.io/csi-api/pkg/apis/csi/v1alpha1"
3232 csiclient "k8s.io/csi-api/pkg/client/clientset/versioned"
3333 "k8s.io/kubernetes/test/e2e/framework"
34+ "k8s.io/kubernetes/test/e2e/storage/testsuites"
3435 "k8s.io/kubernetes/test/e2e/storage/utils"
3536 imageutils "k8s.io/kubernetes/test/utils/image"
3637
@@ -49,7 +50,7 @@ const (
4950type csiTestDriver interface {
5051 createCSIDriver ()
5152 cleanupCSIDriver ()
52- createStorageClassTest (node v1.Node ) storageClassTest
53+ createStorageClassTest (node v1.Node ) testsuites. StorageClassTest
5354}
5455
5556var csiTestDrivers = map [string ]func (f * framework.Framework , config framework.VolumeTestConfig ) csiTestDriver {
@@ -110,7 +111,7 @@ var _ = utils.SIGDescribe("CSI Volumes", func() {
110111 claim := newClaim (t , ns .GetName (), "" )
111112 class := newStorageClass (t , ns .GetName (), "" )
112113 claim .Spec .StorageClassName = & class .ObjectMeta .Name
113- testDynamicProvisioning (t , cs , claim , class )
114+ testsuites . TestDynamicProvisioning (t , cs , claim , class )
114115 })
115116 })
116117 }
@@ -187,7 +188,7 @@ var _ = utils.SIGDescribe("CSI Volumes", func() {
187188 By ("Checking if VolumeAttachment was created for the pod" )
188189 // Check that VolumeAttachment does not exist
189190 handle := getVolumeHandle (cs , claim )
190- attachmentHash := sha256 .Sum256 ([]byte (fmt .Sprintf ("%s%s%s" , handle , t .provisioner , node .Name )))
191+ attachmentHash := sha256 .Sum256 ([]byte (fmt .Sprintf ("%s%s%s" , handle , t .Provisioner , node .Name )))
191192 attachmentName := fmt .Sprintf ("csi-%x" , attachmentHash )
192193 _ , err = cs .StorageV1beta1 ().VolumeAttachments ().Get (attachmentName , metav1.GetOptions {})
193194 if err != nil {
@@ -242,7 +243,7 @@ func getVolumeHandle(cs clientset.Interface, claim *v1.PersistentVolumeClaim) st
242243 return pv .Spec .CSI .VolumeHandle
243244}
244245
245- func startPausePod (cs clientset.Interface , t storageClassTest , ns string ) (* storagev1.StorageClass , * v1.PersistentVolumeClaim , * v1.Pod ) {
246+ func startPausePod (cs clientset.Interface , t testsuites. StorageClassTest , ns string ) (* storagev1.StorageClass , * v1.PersistentVolumeClaim , * v1.Pod ) {
246247 class := newStorageClass (t , ns , "" )
247248 class , err := cs .StorageV1 ().StorageClasses ().Create (class )
248249 framework .ExpectNoError (err , "Failed to create class : %v" , err )
@@ -283,8 +284,8 @@ func startPausePod(cs clientset.Interface, t storageClassTest, ns string) (*stor
283284 },
284285 }
285286
286- if len (t .nodeName ) != 0 {
287- pod .Spec .NodeName = t .nodeName
287+ if len (t .NodeName ) != 0 {
288+ pod .Spec .NodeName = t .NodeName
288289 }
289290 pod , err = cs .CoreV1 ().Pods (ns ).Create (pod )
290291 framework .ExpectNoError (err , "Failed to create pod: %v" , err )
@@ -311,14 +312,14 @@ func initCSIHostpath(f *framework.Framework, config framework.VolumeTestConfig)
311312 }
312313}
313314
314- func (h * hostpathCSIDriver ) createStorageClassTest (node v1.Node ) storageClassTest {
315- return storageClassTest {
316- name : "csi-hostpath" ,
317- provisioner : "csi-hostpath" ,
318- parameters : map [string ]string {},
319- claimSize : "1Gi" ,
320- expectedSize : "1Gi" ,
321- nodeName : node .Name ,
315+ func (h * hostpathCSIDriver ) createStorageClassTest (node v1.Node ) testsuites. StorageClassTest {
316+ return testsuites. StorageClassTest {
317+ Name : "csi-hostpath" ,
318+ Provisioner : "csi-hostpath" ,
319+ Parameters : map [string ]string {},
320+ ClaimSize : "1Gi" ,
321+ ExpectedSize : "1Gi" ,
322+ NodeName : node .Name ,
322323 }
323324}
324325
@@ -379,14 +380,14 @@ func initCSIgcePD(f *framework.Framework, config framework.VolumeTestConfig) csi
379380 }
380381}
381382
382- func (g * gcePDCSIDriver ) createStorageClassTest (node v1.Node ) storageClassTest {
383- return storageClassTest {
384- name : "com.google.csi.gcepd" ,
385- provisioner : "com.google.csi.gcepd" ,
386- parameters : map [string ]string {"type" : "pd-standard" },
387- claimSize : "5Gi" ,
388- expectedSize : "5Gi" ,
389- nodeName : node .Name ,
383+ func (g * gcePDCSIDriver ) createStorageClassTest (node v1.Node ) testsuites. StorageClassTest {
384+ return testsuites. StorageClassTest {
385+ Name : "com.google.csi.gcepd" ,
386+ Provisioner : "com.google.csi.gcepd" ,
387+ Parameters : map [string ]string {"type" : "pd-standard" },
388+ ClaimSize : "5Gi" ,
389+ ExpectedSize : "5Gi" ,
390+ NodeName : node .Name ,
390391 }
391392}
392393
0 commit comments