@@ -421,21 +421,32 @@ func (t *Tester) addCSIDriverFlags() error {
421421 return err
422422 }
423423
424- if cluster .Spec .CloudConfig != nil &&
425- cluster .Spec .CloudConfig .AWSEBSCSIDriver != nil &&
426- cluster .Spec .CloudConfig .AWSEBSCSIDriver .Enabled != nil &&
427- * cluster .Spec .CloudConfig .AWSEBSCSIDriver .Enabled {
424+ var driverFlags string
425+ if cluster .Spec .CloudConfig != nil {
428426 cwd , err := os .Getwd ()
429427 if err != nil {
430428 return err
431429 }
432- klog .Infof ("Setting --storage.testdriver=%s/tests/e2e/csi-manifests/ebs.yaml --storage.migratedPlugins=kubernetes.io/aws-ebs" , cwd )
433- t .TestArgs += fmt .Sprintf (" --storage.testdriver=%s/tests/e2e/csi-manifests/aws-ebs/driver.yaml --storage.migratedPlugins=kubernetes.io/aws-ebs" , cwd )
430+
431+ switch {
432+ case cluster .Spec .CloudConfig .AWSEBSCSIDriver != nil &&
433+ cluster .Spec .CloudConfig .AWSEBSCSIDriver .Enabled != nil &&
434+ * cluster .Spec .CloudConfig .AWSEBSCSIDriver .Enabled :
435+ driverFlags = fmt .Sprintf (" --storage.testdriver=%s/tests/e2e/csi-manifests/aws-ebs/driver.yaml --storage.migratedPlugins=kubernetes.io/aws-ebs" , cwd )
436+ case cluster .Spec .CloudConfig .GCPPDCSIDriver != nil &&
437+ cluster .Spec .CloudConfig .GCPPDCSIDriver .Enabled != nil &&
438+ * cluster .Spec .CloudConfig .GCPPDCSIDriver .Enabled :
439+ driverFlags = fmt .Sprintf (" --storage.testdriver=%s/tests/e2e/csi-manifests/gcp-pd/driver.yaml --storage.migratedPlugins=kubernetes.io/gce-pd" , cwd )
440+ }
441+ }
442+
443+ if driverFlags != "" {
444+ klog .Infof ("Setting %v" , driverFlags )
445+ t .TestArgs += driverFlags
434446 } else {
435- klog .Info ("EBS CSI driver not enabled. Skipping tests" )
447+ klog .Info ("CSI driver not enabled. Skipping tests" )
436448 }
437449 return nil
438-
439450}
440451
441452func (t * Tester ) execute () error {
0 commit comments