fix

apecloud · Nov 29, 2024 · 4b605f0 · 4b605f0
1 parent 0760b09
commit 4b605f0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/pkg/dataprotection/backup/request.go b/pkg/dataprotection/backup/request.go
@@ -170,7 +170,7 @@ func (r *Request) buildBackupDataAction(targetPod *corev1.Pod, name string) (act
 
 	backupDataAct := r.ActionSet.Spec.Backup.BackupData
 	switch r.ActionSet.Spec.BackupType {
-	case dpv1alpha1.BackupTypeFull:
+	case dpv1alpha1.BackupTypeFull, dpv1alpha1.BackupTypeSelective:
 		podSpec, err := r.BuildJobActionPodSpec(targetPod, BackupDataContainerName, &backupDataAct.JobActionSpec)
 		if err != nil {
 			return nil, fmt.Errorf("failed to build job action pod spec: %w", err)

diff --git a/pkg/dataprotection/restore/utils.go b/pkg/dataprotection/restore/utils.go
@@ -275,7 +275,7 @@ func ValidateAndInitRestoreMGR(reqCtx intctrlutil.RequestCtx,
 
 	// build backupActionSets of prepareData and postReady stage based on the specified backup's type.
 	switch backupType {
-	case dpv1alpha1.BackupTypeFull:
+	case dpv1alpha1.BackupTypeFull, dpv1alpha1.BackupTypeSelective:
 		restoreMgr.SetBackupSets(*backupSet)
 	case dpv1alpha1.BackupTypeIncremental:
 		err = restoreMgr.BuildIncrementalBackupActionSets(reqCtx, cli, *backupSet)

diff --git a/pkg/testutil/dataprotection/utils.go b/pkg/testutil/dataprotection/utils.go
@@ -171,6 +171,7 @@ func MockActionSetWithSchema(testCtx *testutil.TestContext, actionSet *dpv1alpha
 		}
 		as.Spec.Backup.WithParameters = []string{ParameterString, ParameterArray}
 		as.Spec.Restore.WithParameters = []string{ParameterString, ParameterArray}
+		as.Spec.BackupType = dpv1alpha1.BackupTypeSelective
 	})).Should(Succeed())
 	By("the actionSet should be available")
 	Eventually(testapps.CheckObj(testCtx, client.ObjectKeyFromObject(actionSet),