fix

Signed-off-by: yaroslavborbat <[email protected]>

Author: yaroslavborbat

images/virtualization-artifact/pkg/controller/vmbda/internal/deletion.go

@@ -26,6 +26,7 @@ import (
 	"sigs.k8s.io/controller-runtime/pkg/reconcile"
 
 	"github.com/deckhouse/virtualization-controller/pkg/common/object"
+	"github.com/deckhouse/virtualization-controller/pkg/controller/kvbuilder"
 	"github.com/deckhouse/virtualization-controller/pkg/logger"
 	virtv2 "github.com/deckhouse/virtualization/api/core/v1alpha2"
 )
@@ -68,7 +69,16 @@ func (h DeletionHandler) cleanUp(ctx context.Context, vmbda *virtv2.VirtualMachi
 		return nil
 	}
 
-	diskName := vmbda.Spec.BlockDeviceRef.Name
+	var diskName string
+	switch vmbda.Spec.BlockDeviceRef.Kind {
+	case virtv2.VMBDAObjectRefKindVirtualDisk:
+		diskName = kvbuilder.GenerateVMDDiskName(vmbda.Spec.BlockDeviceRef.Name)
+	case virtv2.VMBDAObjectRefKindVirtualImage:
+		diskName = kvbuilder.GenerateVMIDiskName(vmbda.Spec.BlockDeviceRef.Name)
+	case virtv2.VMBDAObjectRefKindClusterVirtualImage:
+		diskName = kvbuilder.GenerateCVMIDiskName(vmbda.Spec.BlockDeviceRef.Name)
+	}
+
 	kvvm, err := object.FetchObject(ctx, types.NamespacedName{Namespace: vmbda.GetNamespace(), Name: vmbda.Spec.VirtualMachineName}, h.client, &virtv1.VirtualMachine{})
 	if err != nil {
 		return err