Merge pull request #136 from ZeroMagic/dynamically_provisioned_reclaim_policy

test: add e2e test "dynamically_provisioned_reclaim_policy_test"

Author: k8s-ci-robot

test/e2e/dynamic_provisioning.go

@@ -39,11 +39,11 @@ var _ = Describe("Dynamic Provisioning", func() {
 	)
 
 	nodeid := os.Getenv("nodeid")
-	blobfuseDriver := azuredisk.NewDriver(nodeid)
+	azurediskDriver := azuredisk.NewDriver(nodeid)
 	endpoint := "unix:///tmp/csi.sock"
 
 	go func() {
-		blobfuseDriver.Run(endpoint)
+		azurediskDriver.Run(endpoint)
 	}()
 
 	BeforeEach(func() {
@@ -160,4 +160,37 @@ var _ = Describe("Dynamic Provisioning", func() {
 		}
 		test.Run(cs, ns)
 	})
+
+	It(fmt.Sprintf("should delete PV with reclaimPolicy %q", v1.PersistentVolumeReclaimDelete), func() {
+		reclaimPolicy := v1.PersistentVolumeReclaimDelete
+		volumes := []testsuites.VolumeDetails{
+			{
+				FSType:        "ext4",
+				ClaimSize:     "10Gi",
+				ReclaimPolicy: &reclaimPolicy,
+			},
+		}
+		test := testsuites.DynamicallyProvisionedReclaimPolicyTest{
+			CSIDriver: testDriver,
+			Volumes:   volumes,
+		}
+		test.Run(cs, ns)
+	})
+
+	It(fmt.Sprintf("[env] should retain PV with reclaimPolicy %q", v1.PersistentVolumeReclaimRetain), func() {
+		reclaimPolicy := v1.PersistentVolumeReclaimRetain
+		volumes := []testsuites.VolumeDetails{
+			{
+				FSType:        "ext4",
+				ClaimSize:     "10Gi",
+				ReclaimPolicy: &reclaimPolicy,
+			},
+		}
+		test := testsuites.DynamicallyProvisionedReclaimPolicyTest{
+			CSIDriver: testDriver,
+			Volumes:   volumes,
+			Azuredisk: azurediskDriver,
+		}
+		test.Run(cs, ns)
+	})
 })

test/e2e/testsuites/dynamically_provisioned_reclaim_policy_tester.go

@@ -0,0 +1,47 @@
+/*
+Copyright 2019 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+   http://www.apache.org/licenses/LICENSE-2.0
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package testsuites
+
+import (
+	"github.com/kubernetes-sigs/azuredisk-csi-driver/pkg/azuredisk"
+	"github.com/kubernetes-sigs/azuredisk-csi-driver/test/e2e/driver"
+
+	"k8s.io/api/core/v1"
+	clientset "k8s.io/client-go/kubernetes"
+)
+
+// DynamicallyProvisionedReclaimPolicyTest will provision required PV(s) and PVC(s)
+// Testing the correct behavior for different reclaimPolicies
+type DynamicallyProvisionedReclaimPolicyTest struct {
+	CSIDriver driver.DynamicPVTestDriver
+	Volumes   []VolumeDetails
+	Azuredisk *azuredisk.Driver
+}
+
+func (t *DynamicallyProvisionedReclaimPolicyTest) Run(client clientset.Interface, namespace *v1.Namespace) {
+	for _, volume := range t.Volumes {
+		tpvc, _ := volume.SetupDynamicPersistentVolumeClaim(client, namespace, t.CSIDriver)
+
+		// will delete the PVC
+		// will also wait for PV to be deleted when reclaimPolicy=Delete
+		tpvc.Cleanup()
+		// first check PV stills exists, then manually delete it
+		if tpvc.ReclaimPolicy() == v1.PersistentVolumeReclaimRetain {
+			tpvc.WaitForPersistentVolumePhase(v1.VolumeReleased)
+			tpvc.DeleteBoundPersistentVolume()
+			tpvc.DeleteBackingVolume(t.Azuredisk)
+		}
+	}
+}