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

@@ -574,12 +574,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

@@ -56,9 +56,9 @@ const (
 	hdtDiskType                            = "hyperdisk-throughput"
 	provisionedIOPSOnCreate                = "12345"
 	provisionedIOPSOnCreateInt             = int64(12345)
+	provisionedIOPSOnCreateDefaultInt       = int64(100000)
 	provisionedThroughputOnCreate          = "66Mi"
 	provisionedThroughputOnCreateInt       = int64(66)
-
 	defaultEpsilon = 500000000 // 500M
 )
 
@@ -424,6 +424,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(cloudDisk.Type).To(ContainSubstring(extremeDiskType))
+			Expect(cloudDisk.ProvisionedIops).To(Equal(provisionedIOPSOnCreateDefaultInt))
+			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())