kmm: add support for tolerations

Author: cdvultur

pkg/kmm/module.go

@@ -179,6 +179,50 @@ func (builder *ModuleBuilder) WithDevicePluginVolume(name string, configMapName
 	return builder
 }
 
+// WithToleration adds the specified toleration to the Module.
+func (builder *ModuleBuilder) WithToleration(key string, operator string,
+	value string, effect string, seconds int64) *ModuleBuilder {
+	if valid, _ := builder.validate(); !valid {
+		return builder
+	}
+
+	if key == "" {
+		builder.errorMsg = "cannot redefine with empty 'key' value"
+
+		return builder
+	}
+
+	if operator == "" {
+		builder.errorMsg = "cannot redefine with empty 'operator' value"
+
+		return builder
+	}
+
+	if effect == "" {
+		builder.errorMsg = "cannot redefine with empty 'effect' value"
+
+		return builder
+	}
+
+	if builder.Definition.Spec.Tolerations == nil {
+		builder.Definition.Spec.Tolerations = []corev1.Toleration{}
+	}
+
+	glog.V(100).Infof(
+		"Appending toleration with key: %s, operator: %s, value: %s, effect: %s and seconds: %s",
+		key, operator, value, effect, seconds)
+
+	builder.Definition.Spec.Tolerations = append(builder.Definition.Spec.Tolerations, []corev1.Toleration{{
+		Key:               key,
+		Effect:            corev1.TaintEffect(effect),
+		Operator:          corev1.TolerationOperator(operator),
+		Value:             value,
+		TolerationSeconds: &seconds,
+	}}...)
+
+	return builder
+}
+
 // WithModuleLoaderContainer adds the specified ModuleLoader container to the Module.
 func (builder *ModuleBuilder) WithModuleLoaderContainer(
 	container *moduleV1Beta1.ModuleLoaderContainerSpec) *ModuleBuilder {

pkg/kmm/module_test.go

@@ -7,6 +7,7 @@ import (
 	"github.com/openshift-kni/eco-goinfra/pkg/clients"
 	"github.com/openshift-kni/eco-goinfra/pkg/schemes/kmm/v1beta1"
 	"github.com/stretchr/testify/assert"
+	v1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/errors"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/runtime"
@@ -514,6 +515,62 @@ func TestModuleWithOptions(t *testing.T) {
 	assert.Equal(t, "error", testBuilder.errorMsg)
 }
 
+func TestModuleWithToleration(t *testing.T) {
+	testCases := []struct {
+		key         string
+		operator    string
+		value       string
+		effect      string
+		seconds     int64
+		expectedErr string
+	}{
+		{
+			key:         "",
+			operator:    "test",
+			value:       "test",
+			effect:      "test",
+			seconds:     10,
+			expectedErr: "cannot redefine with empty 'key' value",
+		},
+		{
+			key:         "test",
+			operator:    "",
+			value:       "test",
+			effect:      "test",
+			seconds:     10,
+			expectedErr: "cannot redefine with empty 'operator' value",
+		},
+		{
+			key:         "test",
+			operator:    "test",
+			value:       "test",
+			effect:      "",
+			seconds:     10,
+			expectedErr: "cannot redefine with empty 'effect' value",
+		},
+		{
+			key:         "testkey",
+			operator:    "Equals",
+			value:       "testvalue",
+			effect:      "NoSchedule",
+			seconds:     0,
+			expectedErr: "",
+		},
+	}
+	for _, testCase := range testCases {
+		testSettings := buildModuleTestClientWithDummyObject()
+		testBuilder := buildValidTestModule(testSettings).WithToleration(
+			testCase.key, testCase.operator, testCase.value, testCase.effect, testCase.seconds)
+
+		if testCase.expectedErr == "" {
+			assert.Equal(t, testCase.key, testBuilder.Definition.Spec.Tolerations[0].Key)
+			assert.Equal(t, v1.TaintEffect(testCase.effect), testBuilder.Definition.Spec.Tolerations[0].Effect)
+		} else {
+			assert.Equal(t, testCase.expectedErr, testBuilder.errorMsg)
+		}
+	}
+}
+
 func TestModuleBuildModuleSpec(t *testing.T) {
 	testCases := []struct {
 		testModule    *ModuleBuilder