Add data download controller UT

Signed-off-by: Ming Qiu <mqiu@vmware.com>
vmware-tanzu · Jul 4, 2023 · 2f667f5 · 2f667f5
1 parent 5171ab0
commit 2f667f5
Show file tree

Hide file tree

Showing 2 changed files with 392 additions and 24 deletions.
diff --git a/pkg/controller/data_download_controller.go b/pkg/controller/data_download_controller.go
@@ -431,7 +431,9 @@ func prepareDataDownload(ssb *velerov2alpha1api.DataDownload) {
 }
 
 func (r *DataDownloadReconciler) errorOut(ctx context.Context, dd *velerov2alpha1api.DataDownload, err error, msg string, log logrus.FieldLogger) (ctrl.Result, error) {
-	r.restoreExposer.CleanUp(ctx, getDataDownloadOwnerObject(dd))
+	if r.restoreExposer != nil {
+		r.restoreExposer.CleanUp(ctx, getDataDownloadOwnerObject(dd))
+	}
 	return ctrl.Result{}, r.updateStatusToFailed(ctx, dd, err, msg, log)
 }