++ add tolerations to test

- Add tolerations to workload settings in cdi config.

Signed-off-by: Ivan Mikheykin <[email protected]>

Author: diafour

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

@@ -17,6 +17,7 @@ limitations under the License.
 package vm
 
 import (
+	corev1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
 
 	"github.com/deckhouse/virtualization-controller/pkg/builder/meta"
@@ -94,6 +95,12 @@ func WithNodeSelector(nodeSelector map[string]string) Option {
 	}
 }
 
+func WithTolerations(tolerations []corev1.Toleration) Option {
+	return func(vm *v1alpha2.VirtualMachine) {
+		vm.Spec.Tolerations = tolerations
+	}
+}
+
 func WithLiveMigrationPolicy(liveMigrationPolicy v1alpha2.LiveMigrationPolicy) Option {
 	return func(vm *v1alpha2.VirtualMachine) {
 		vm.Spec.LiveMigrationPolicy = liveMigrationPolicy

templates/cdi/config.yaml

@@ -162,3 +162,7 @@ spec:
   workload:
     nodeSelector:
       kubernetes.io/os: linux
+    tolerations:
+      - key: dedicated.deckhouse.io
+        operator: Equal
+        value: system

test/e2e/blockdevice/virtual_image_creation.go

@@ -22,6 +22,7 @@ import (
 
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
+	corev1 "k8s.io/api/core/v1"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 
 	cvibuilder "github.com/deckhouse/virtualization-controller/pkg/builder/cvi"
@@ -67,10 +68,17 @@ var _ = Describe("VirtualImageCreation", func() {
 			)
 			err := f.CreateWithDeferredDeletion(context.Background(), vd)
 			Expect(err).NotTo(HaveOccurred())
-			vm := object.NewMinimalVM("vm-", f.Namespace().Name, vmbuilder.WithBlockDeviceRefs(v1alpha2.BlockDeviceSpecRef{
-				Kind: v1alpha2.VirtualDiskKind,
-				Name: vd.Name,
-			}))
+			vm := object.NewMinimalVM("vm-", f.Namespace().Name,
+				vmbuilder.WithBlockDeviceRefs(v1alpha2.BlockDeviceSpecRef{
+					Kind: v1alpha2.VirtualDiskKind,
+					Name: vd.Name,
+				}),
+				vmbuilder.WithTolerations([]corev1.Toleration{{
+					Key:      "dedicated.deckhouse.io",
+					Operator: corev1.TolerationOpEqual,
+					Value:    "system",
+				}}),
+			)
 			err = f.CreateWithDeferredDeletion(context.Background(), vm)
 			Expect(err).NotTo(HaveOccurred())
 			util.UntilObjectPhase(string(v1alpha2.DiskReady), framework.LongTimeout, vd)