Add pd-extreme e2e test case of defaulting to default iops when iops not

set

Author: sunnylovestiramisu

pkg/gce-cloud-provider/compute/gce-compute.go

@@ -564,12 +564,15 @@ func (cloud *CloudProvider) insertZonalDisk(
 	}
 
 	diskToCreate := &computev1.Disk{
-		Name:            volKey.Name,
-		SizeGb:          common.BytesToGbRoundUp(capBytes),
-		Description:     description,
-		Type:            cloud.GetDiskTypeURI(project, volKey, params.DiskType),
-		Labels:          params.Labels,
-		ProvisionedIops: params.ProvisionedIOPSOnCreate,
+		Name:        volKey.Name,
+		SizeGb:      common.BytesToGbRoundUp(capBytes),
+		Description: description,
+		Type:        cloud.GetDiskTypeURI(project, volKey, params.DiskType),
+		Labels:      params.Labels,
+	}
+
+	if params.ProvisionedIOPSOnCreate > 0 {
+		diskToCreate.ProvisionedIops = params.ProvisionedIOPSOnCreate
 	}
 
 	if snapshotID != "" {

test/e2e/tests/single_zone_e2e_test.go

@@ -44,16 +44,17 @@ import (
 const (
 	testNamePrefix = "gcepd-csi-e2e-"
 
-	defaultSizeGb              int64 = 5
-	defaultExtremeSizeGb       int64 = 500
-	defaultRepdSizeGb          int64 = 200
-	defaultMwSizeGb            int64 = 200
-	defaultVolumeLimit         int64 = 127
-	readyState                       = "READY"
-	standardDiskType                 = "pd-standard"
-	extremeDiskType                  = "pd-extreme"
-	provisionedIOPSOnCreate          = "12345"
-	provisionedIOPSOnCreateInt       = int64(12345)
+	defaultSizeGb                     int64 = 5
+	defaultExtremeSizeGb              int64 = 500
+	defaultRepdSizeGb                 int64 = 200
+	defaultMwSizeGb                   int64 = 200
+	defaultVolumeLimit                int64 = 127
+	readyState                              = "READY"
+	standardDiskType                        = "pd-standard"
+	extremeDiskType                         = "pd-extreme"
+	provisionedIOPSOnCreate                 = "12345"
+	provisionedIOPSOnCreateInt              = int64(12345)
+	provisionedIOPSOnCreateDefaultInt       = int64(100000)
 
 	defaultEpsilon = 500000000 // 500M
 )
@@ -419,6 +420,47 @@ var _ = Describe("GCE PD CSI Driver", func() {
 		Entry("on pd-extreme", extremeDiskType),
 	)
 
+	DescribeTable("Should create and delete pd-extreme disk with default iops",
+		func(diskType string) {
+			Expect(testContexts).ToNot(BeEmpty())
+			testContext := getRandomTestContext()
+
+			p, z, _ := testContext.Instance.GetIdentity()
+			client := testContext.Client
+
+			// Create Disk
+			disk := typeToDisk[diskType]
+			volName := testNamePrefix + string(uuid.NewUUID())
+
+			diskSize := defaultExtremeSizeGb
+
+			volume, err := client.CreateVolume(volName, disk.params, diskSize, nil, nil)
+
+			Expect(err).To(BeNil(), "CreateVolume failed with error: %v", err)
+
+			// Validate Disk Created
+			cloudDisk, err := computeService.Disks.Get(p, z, volName).Do()
+			Expect(err).To(BeNil(), "Could not get disk from cloud directly")
+			Expect(cloudDisk.Status).To(Equal(readyState))
+			Expect(cloudDisk.SizeGb).To(Equal(defaultExtremeSizeGb))
+			Expect(disk.Type).To(ContainSubstring(extremeDiskType))
+			Expect(disk.ProvisionedIops).To(Equal(provisionedIOPSOnCreateInt))
+			Expect(cloudDisk.Name).To(Equal(volName))
+			disk.validate(cloudDisk)
+
+			defer func() {
+				// Delete Disk
+				client.DeleteVolume(volume.VolumeId)
+				Expect(err).To(BeNil(), "DeleteVolume failed")
+
+				// Validate Disk Deleted
+				_, err = computeService.Disks.Get(p, z, volName).Do()
+				Expect(gce.IsGCEError(err, "notFound")).To(BeTrue(), "Expected disk to not be found")
+			}()
+		},
+		Entry("on pd-extreme", extremeDiskType),
+	)
+
 	DescribeTable("Should create and delete disk with labels",
 		func(diskType string) {
 			Expect(testContexts).ToNot(BeEmpty())