package e2e import ( "context" "fmt" "os" "path/filepath" "strings" . "github.com/onsi/ginkgo" // nolint v1 "k8s.io/api/core/v1" "k8s.io/apimachinery/pkg/api/resource" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" clientset "k8s.io/client-go/kubernetes" "k8s.io/kubernetes/test/e2e/framework" e2elog "k8s.io/kubernetes/test/e2e/framework/log" ) var _ = Describe("CephFS Upgrade Testing", func() { f := framework.NewDefaultFramework("upgrade-test-cephfs") var ( c clientset.Interface pvc *v1.PersistentVolumeClaim pvcClone *v1.PersistentVolumeClaim app *v1.Pod appClone *v1.Pod // cwd stores the initial working directory. cwd string err error // checkSum stores the md5sum of a file to verify uniqueness. checkSum string // newCheckSum stores the md5sum of a file in the cloned pvc. newCheckSum string ) const ( pvcSize = "2Gi" appKey = "app" appLabel = "cephfs-upgrade-testing" ) // deploy cephfs CSI BeforeEach(func() { if !upgradeTesting || !testCephFS { Skip("Skipping CephFS Upgrade Test") } c = f.ClientSet if cephCSINamespace != defaultNs { err = createNamespace(c, cephCSINamespace) if err != nil { e2elog.Failf("failed to create namespace with error %v", err) } } // fetch current working directory to switch back // when we are done upgrading. cwd, err = os.Getwd() if err != nil { e2elog.Failf("failed to getwd with error %v", err) } err = upgradeAndDeployCSI(upgradeVersion, "cephfs") if err != nil { e2elog.Failf("failed to upgrade csi with error %v", err) } err = createConfigMap(cephfsDirPath, f.ClientSet, f) if err != nil { e2elog.Failf("failed to create configmap with error %v", err) } err = createCephfsSecret(f.ClientSet, f) if err != nil { e2elog.Failf("failed to create secret with error %v", err) } err = createCephFSSnapshotClass(f) if err != nil { e2elog.Failf("failed to create snapshotclass with error %v", err) } err = createCephfsStorageClass(f.ClientSet, f, true, nil) if err != nil { e2elog.Failf("failed to create storageclass with error %v", err) } }) AfterEach(func() { if !testCephFS || !upgradeTesting { Skip("Skipping CephFS Upgrade Test") } if CurrentGinkgoTestDescription().Failed { // log pods created by helm chart logsCSIPods("app=ceph-csi-cephfs", c) // log provisoner logsCSIPods("app=csi-cephfsplugin-provisioner", c) // log node plugin logsCSIPods("app=csi-cephfsplugin", c) } err = deleteConfigMap(cephfsDirPath) if err != nil { e2elog.Failf("failed to delete configmap with error %v", err) } err = deleteResource(cephfsExamplePath + "secret.yaml") if err != nil { e2elog.Failf("failed to delete secret with error %v", err) } err = deleteResource(cephfsExamplePath + "storageclass.yaml") if err != nil { e2elog.Failf("failed to delete storageclass with error %v", err) } err = deleteResource(cephfsExamplePath + "snapshotclass.yaml") if err != nil { e2elog.Failf("failed to delete storageclass with error %v", err) } if deployCephFS { deleteCephfsPlugin() if cephCSINamespace != defaultNs { err = deleteNamespace(c, cephCSINamespace) if err != nil { if err != nil { e2elog.Failf("failed to delete namespace with error %v", err) } } } } }) Context("Cephfs Upgrade Test", func() { It("Cephfs Upgrade Test", func() { By("checking provisioner deployment is running", func() { err = waitForDeploymentComplete(cephfsDeploymentName, cephCSINamespace, f.ClientSet, deployTimeout) if err != nil { e2elog.Failf("timeout waiting for deployment %s with error %v", cephfsDeploymentName, err) } }) By("checking nodeplugin deamonset pods are running", func() { err = waitForDaemonSets(cephfsDeamonSetName, cephCSINamespace, f.ClientSet, deployTimeout) if err != nil { e2elog.Failf("timeout waiting for daemonset %s with error%v", cephfsDeamonSetName, err) } }) By("upgrade to latest changes and verify app re-mount", func() { // TODO: fetch pvc size from spec. pvcPath := cephfsExamplePath + "pvc.yaml" appPath := cephfsExamplePath + "pod.yaml" data := "check data persists" label := make(map[string]string) pvc, err = loadPVC(pvcPath) if err != nil { e2elog.Failf("failed to load pvc with error %v", err) } pvc.Namespace = f.UniqueName app, err = loadApp(appPath) if err != nil { e2elog.Failf("failed to load application with error %v", err) } label[appKey] = appLabel app.Namespace = f.UniqueName app.Labels = label pvc.Namespace = f.UniqueName pvc.Spec.Resources.Requests[v1.ResourceStorage] = resource.MustParse(pvcSize) err = createPVCAndApp("", f, pvc, app, deployTimeout) if err != nil { e2elog.Failf("failed to create pvc and application with error %v", err) } opt := metav1.ListOptions{ LabelSelector: fmt.Sprintf("%s=%s", appKey, label[appKey]), } // fetch the path where volume is mounted. mountPath := app.Spec.Containers[0].VolumeMounts[0].MountPath filePath := filepath.Join(mountPath, "testClone") // create a test file at the mountPath. _, stdErr := execCommandInPodAndAllowFail(f, fmt.Sprintf("echo %s > %s", data, filePath), app.Namespace, &opt) if stdErr != "" { e2elog.Failf("failed to write data to a file %s", stdErr) } // force an immediate write of all cached data to disk. _, stdErr = execCommandInPodAndAllowFail(f, fmt.Sprintf("sync %s", filePath), app.Namespace, &opt) if stdErr != "" { e2elog.Failf("failed to sync data to a disk %s", stdErr) } e2elog.Logf("Calculating checksum of %s", filePath) checkSum, err = calculateSHA512sum(f, app, filePath, &opt) if err != nil { e2elog.Failf("failed to calculate checksum of %s", filePath) } // pvc clone is only supported from v1.16+ if k8sVersionGreaterEquals(f.ClientSet, 1, 17) { // Create snapshot of the pvc snapshotPath := cephfsExamplePath + "snapshot.yaml" snap := getSnapshot(snapshotPath) snap.Name = "cephfs-pvc-snapshot" snap.Namespace = f.UniqueName snap.Spec.Source.PersistentVolumeClaimName = &pvc.Name err = createSnapshot(&snap, deployTimeout) if err != nil { e2elog.Failf("failed to create snapshot %v", err) } } err = deletePod(app.Name, app.Namespace, f.ClientSet, deployTimeout) if err != nil { e2elog.Failf("failed to delete application with error %v", err) } deleteCephfsPlugin() // switch back to current changes. err = os.Chdir(cwd) if err != nil { e2elog.Failf("failed to d chdir with error %v", err) } deployCephfsPlugin() app.Labels = label // validate if the app gets bound to a pvc created by // an earlier release. err = createApp(f.ClientSet, app, deployTimeout) if err != nil { e2elog.Failf("failed to create application with error %v", err) } }) By("Create clone from a snapshot", func() { pvcClonePath := cephfsExamplePath + "pvc-restore.yaml" appClonePath := cephfsExamplePath + "pod-restore.yaml" label := make(map[string]string) // pvc clone is only supported from v1.16+ if k8sVersionGreaterEquals(f.ClientSet, 1, 17) { pvcClone, err = loadPVC(pvcClonePath) if err != nil { e2elog.Failf("failed to load pvc with error %v", err) } pvcClone.Namespace = f.UniqueName pvcClone.Spec.Resources.Requests[v1.ResourceStorage] = resource.MustParse(pvcSize) appClone, err = loadApp(appClonePath) if err != nil { e2elog.Failf("failed to load application with error %v", err) } label[appKey] = "validate-snap-cephfs" appClone.Namespace = f.UniqueName appClone.Name = "snap-clone-cephfs" appClone.Labels = label err = createPVCAndApp("", f, pvcClone, appClone, deployTimeout) if err != nil { e2elog.Failf("failed to create pvc and application with error %v", err) } opt := metav1.ListOptions{ LabelSelector: fmt.Sprintf("%s=%s", appKey, label[appKey]), } mountPath := appClone.Spec.Containers[0].VolumeMounts[0].MountPath testFilePath := filepath.Join(mountPath, "testClone") newCheckSum, err = calculateSHA512sum(f, appClone, testFilePath, &opt) if err != nil { e2elog.Failf("failed to calculate checksum of %s", testFilePath) } if strings.Compare(newCheckSum, checkSum) != 0 { e2elog.Failf("The checksum of files did not match, expected %s received %s ", checkSum, newCheckSum) } e2elog.Logf("The checksum of files matched") // delete cloned pvc and pod err = deletePVCAndApp("", f, pvcClone, appClone) if err != nil { Fail(err.Error()) } // Delete the snapshot of the parent pvc. snapshotPath := cephfsExamplePath + "snapshot.yaml" snap := getSnapshot(snapshotPath) snap.Name = "cephfs-pvc-snapshot" snap.Namespace = f.UniqueName snap.Spec.Source.PersistentVolumeClaimName = &pvc.Name err = deleteSnapshot(&snap, deployTimeout) if err != nil { e2elog.Failf("failed to delete snapshot %v", err) } } }) By("Create clone from existing PVC", func() { pvcSmartClonePath := cephfsExamplePath + "pvc-clone.yaml" appSmartClonePath := cephfsExamplePath + "pod-clone.yaml" label := make(map[string]string) // pvc clone is only supported from v1.16+ if k8sVersionGreaterEquals(f.ClientSet, 1, 16) { pvcClone, err = loadPVC(pvcSmartClonePath) if err != nil { e2elog.Failf("failed to load pvc with error %v", err) } pvcClone.Spec.DataSource.Name = pvc.Name pvcClone.Namespace = f.UniqueName pvcClone.Spec.Resources.Requests[v1.ResourceStorage] = resource.MustParse(pvcSize) appClone, err = loadApp(appSmartClonePath) if err != nil { e2elog.Failf("failed to load application with error %v", err) } label[appKey] = "validate-snap-cephfs" appClone.Namespace = f.UniqueName appClone.Name = "appclone" appClone.Labels = label err = createPVCAndApp("", f, pvcClone, appClone, deployTimeout) if err != nil { e2elog.Failf("failed to create pvc and application with error %v", err) } opt := metav1.ListOptions{ LabelSelector: fmt.Sprintf("%s=%s", appKey, label[appKey]), } mountPath := appClone.Spec.Containers[0].VolumeMounts[0].MountPath testFilePath := filepath.Join(mountPath, "testClone") newCheckSum, err = calculateSHA512sum(f, appClone, testFilePath, &opt) if err != nil { e2elog.Failf("failed to calculate checksum of %s", testFilePath) } if strings.Compare(newCheckSum, checkSum) != 0 { e2elog.Failf("The checksum of files did not match, expected %s received %s", checkSum, newCheckSum) } e2elog.Logf("The checksum of files matched") // delete cloned pvc and pod err = deletePVCAndApp("", f, pvcClone, appClone) if err != nil { e2elog.Failf("failed to delete pvc and application with error %v", err) } } }) By("Resize pvc and verify expansion", func() { pvcExpandSize := "5Gi" label := make(map[string]string) // Resize 0.3.0 is only supported from v1.15+ if k8sVersionGreaterEquals(f.ClientSet, 1, 15) { label[appKey] = appLabel opt := metav1.ListOptions{ LabelSelector: fmt.Sprintf("%s=%s", appKey, label[appKey]), } pvc, err = f.ClientSet.CoreV1().PersistentVolumeClaims(pvc.Namespace).Get(context.TODO(), pvc.Name, metav1.GetOptions{}) if err != nil { e2elog.Failf("failed to get pvc with error %v", err) } // resize PVC err = expandPVCSize(f.ClientSet, pvc, pvcExpandSize, deployTimeout) if err != nil { e2elog.Failf("failed to expand pvc with error %v", err) } // wait for application pod to come up after resize err = waitForPodInRunningState(app.Name, app.Namespace, f.ClientSet, deployTimeout) if err != nil { e2elog.Failf("timeout waiting for pod to be in running state with error %v", err) } // validate if resize is successful. err = checkDirSize(app, f, &opt, pvcExpandSize) if err != nil { e2elog.Failf("failed to check directory size with error %v", err) } } }) By("delete pvc and app") err = deletePVCAndApp("", f, pvc, app) if err != nil { e2elog.Failf("failed to delete pvc and application with error %v", err) } }) }) })