fix

Signed-off-by: Yaroslav Borbat <yaroslav.borbat@flant.com>

Author: yaroslavborbat

images/virtualization-artifact/pkg/builder/vd/option.go

@@ -17,6 +17,9 @@ limitations under the License.
 package vd
 
 import (
+	"k8s.io/apimachinery/pkg/api/resource"
+	"k8s.io/utils/ptr"
+
 	"github.com/deckhouse/virtualization-controller/pkg/builder/meta"
 	"github.com/deckhouse/virtualization/api/core/v1alpha2"
 )
@@ -81,3 +84,24 @@ func WithDataSourceObjectRefFromCVI(cvi *v1alpha2.ClusterVirtualImage) Option {
 func WithDataSourceObjectRefFromVI(vi *v1alpha2.VirtualImage) Option {
 	return WithDataSourceObjectRef(v1alpha2.VirtualDiskObjectRefKindVirtualImage, vi.Name)
 }
+
+func WithPersistentVolumeClaim(storageClass *string, size *resource.Quantity) Option {
+	return func(vd *v1alpha2.VirtualDisk) {
+		vd.Spec.PersistentVolumeClaim = v1alpha2.VirtualDiskPersistentVolumeClaim{
+			StorageClass: storageClass,
+			Size:         size,
+		}
+	}
+}
+
+func WithStorageClass(storageClass string) Option {
+	return func(vd *v1alpha2.VirtualDisk) {
+		vd.Spec.PersistentVolumeClaim.StorageClass = ptr.To(storageClass)
+	}
+}
+
+func WithSize(size *resource.Quantity) Option {
+	return func(vd *v1alpha2.VirtualDisk) {
+		vd.Spec.PersistentVolumeClaim.Size = size
+	}
+}

images/virtualization-artifact/pkg/builder/vm/option.go

@@ -63,6 +63,12 @@ func WithDisks(disks ...*v1alpha2.VirtualDisk) Option {
 	}
 }
 
+func WithBlockDeviceRefs(refs ...v1alpha2.BlockDeviceSpecRef) Option {
+	return func(vm *v1alpha2.VirtualMachine) {
+		vm.Spec.BlockDeviceRefs = append(vm.Spec.BlockDeviceRefs, refs...)
+	}
+}
+
 func WithNodeSelector(nodeSelector map[string]string) Option {
 	return func(vm *v1alpha2.VirtualMachine) {
 		vm.Spec.NodeSelector = nodeSelector