rook: Add lokocfg conversion tests

Add unit tests that verify if the conversion worked correctly.

Signed-off-by: Suraj Deshmukh <suraj@kinvolk.io>

pkg/components/rook/component_test.go

@@ -0,0 +1,216 @@
+// Copyright 2020 The Lokomotive Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// nolint:testpackage
+package rook
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/hashicorp/hcl/v2"
+	appsv1 "k8s.io/api/apps/v1"
+	corev1 "k8s.io/api/core/v1"
+	"sigs.k8s.io/yaml"
+
+	"github.com/kinvolk/lokomotive/pkg/components/util"
+)
+
+func renderManifest(t *testing.T, configHCL string) map[string]string {
+	component := newComponent()
+
+	body, diagnostics := util.GetComponentBody(configHCL, name)
+	if diagnostics != nil {
+		t.Fatalf("Error getting component body: %v", diagnostics)
+	}
+
+	diagnostics = component.LoadConfig(body, &hcl.EvalContext{})
+	if diagnostics.HasErrors() {
+		t.Fatalf("Valid config should not return error, got: %s", diagnostics)
+	}
+
+	ret, err := component.RenderManifests()
+	if err != nil {
+		t.Fatalf("Rendering manifests with valid config should succeed, got: %s", err)
+	}
+
+	return ret
+}
+
+func getOperatorDeployment(t *testing.T, m map[string]string) *appsv1.Deployment {
+	deployStr, ok := m["rook-ceph/templates/deployment.yaml"]
+	if !ok {
+		t.Fatalf("operator deployment config not found")
+	}
+
+	deploy := &appsv1.Deployment{}
+	if err := yaml.Unmarshal([]byte(deployStr), deploy); err != nil {
+		t.Fatalf("failed unmarshaling manifest: %v", err)
+	}
+
+	return deploy
+}
+
+// nolint:funlen
+func TestConversion(t *testing.T) {
+	configHCL := `
+component "rook" {
+  enable_monitoring = true
+
+  node_selector = {
+    "storage_node_key1" = "storage_node_value1"
+    "storage_node_key2" = "storage_node_value2"
+  }
+
+  toleration {
+    key      = "toleration_key1"
+    operator = "Equal"
+    value    = "toleration_value1"
+    effect   = "NoSchedule"
+  }
+
+  toleration {
+    key      = "toleration_key2"
+    operator = "Equal"
+    value    = "toleration_value2"
+    effect   = "NoSchedule"
+  }
+
+  agent_toleration_key    = "agent_toleration_key"
+  agent_toleration_effect = "NoSchedule"
+
+  discover_toleration_key    = "discover_toleration_key"
+  discover_toleration_effect = "NoSchedule"
+
+  csi_plugin_node_selector = {
+    "node_key1" = "node_value1"
+    "node_key2" = "node_value2"
+  }
+
+  csi_plugin_toleration {
+    key = "other_node_key1"
+    operator = "Equal"
+    value = "other_node_value1"
+    effect = "NoSchedule"
+  }
+}
+`
+
+	m := renderManifest(t, configHCL)
+	if len(m) == 0 {
+		t.Fatalf("Rendered manifests shouldn't be empty")
+	}
+
+	deploy := getOperatorDeployment(t, m)
+
+	tcs := []struct {
+		Name string
+		Test func(*testing.T, map[string]string)
+	}{
+		{"MemoryResources", func(t *testing.T, m map[string]string) {
+			expected := "512Mi"
+			got := deploy.Spec.Template.Spec.Containers[0].Resources.Limits.Memory().String()
+			if expected != got {
+				t.Fatalf("expected: %s, got: %s", expected, got)
+			}
+		}},
+		{"NodeSelector", func(t *testing.T, m map[string]string) {
+			expected := map[string]string{
+				"storage_node_key1": "storage_node_value1",
+				"storage_node_key2": "storage_node_value2",
+			}
+			got := deploy.Spec.Template.Spec.NodeSelector
+
+			if !reflect.DeepEqual(expected, got) {
+				t.Fatalf("expcted: %v, got: %v", expected, got)
+			}
+		}},
+		{"Monitoring", func(t *testing.T, m map[string]string) {
+			expectedConfig := []string{
+				"rook-ceph/templates/ceph-cluster.yaml",
+				"rook-ceph/templates/ceph-osd.yaml",
+				"rook-ceph/templates/ceph-pools.yaml",
+				"rook-ceph/templates/csi-metrics-service-monitor.yaml",
+				"rook-ceph/templates/prometheus-ceph-v14-rules.yaml",
+				"rook-ceph/templates/service-monitor.yaml",
+			}
+
+			for _, ec := range expectedConfig {
+				if _, ok := m[ec]; !ok {
+					t.Fatalf("expected %q to be generated but it is not available", ec)
+				}
+			}
+		}},
+		{"OperatorTolerations", func(t *testing.T, m map[string]string) {
+			expected := []corev1.Toleration{
+				{Key: "toleration_key1", Operator: "Equal", Value: "toleration_value1", Effect: "NoSchedule"},
+				{Key: "toleration_key2", Operator: "Equal", Value: "toleration_value2", Effect: "NoSchedule"},
+			}
+			got := deploy.Spec.Template.Spec.Tolerations
+
+			if !reflect.DeepEqual(expected, got) {
+				t.Fatalf("expected: %v, got: %v", expected, got)
+			}
+		}},
+		{"EnvVars", func(t *testing.T, m map[string]string) {
+			// nolint:lll
+			subTCs := map[string]string{
+				"AGENT_TOLERATION_KEY":        "agent_toleration_key",
+				"AGENT_TOLERATION":            "NoSchedule",
+				"AGENT_TOLERATIONS":           "- effect: NoSchedule\n  key: toleration_key1\n  operator: Equal\n  value: toleration_value1\n- effect: NoSchedule\n  key: toleration_key2\n  operator: Equal\n  value: toleration_value2",
+				"DISCOVER_TOLERATION_KEY":     "discover_toleration_key",
+				"DISCOVER_TOLERATION":         "NoSchedule",
+				"DISCOVER_TOLERATIONS":        "- effect: NoSchedule\n  key: toleration_key1\n  operator: Equal\n  value: toleration_value1\n- effect: NoSchedule\n  key: toleration_key2\n  operator: Equal\n  value: toleration_value2",
+				"CSI_PROVISIONER_TOLERATIONS": "- effect: NoSchedule\n  key: toleration_key1\n  operator: Equal\n  value: toleration_value1\n- effect: NoSchedule\n  key: toleration_key2\n  operator: Equal\n  value: toleration_value2",
+				"CSI_PLUGIN_TOLERATIONS":      "- effect: NoSchedule\n  key: other_node_key1\n  operator: Equal\n  value: other_node_value1",
+
+				// TODO: Fix the ordering in implementation of convertNodeSelector. Once that is not
+				// consistent, we can start testing following cases.
+				// https://github.com/kinvolk/lokomotive/issues/942.
+				// "AGENT_NODE_AFFINITY":           "storage_node_key1=storage_node_value1; storage_node_key2=storage_node_value2;",
+				// "DISCOVER_AGENT_NODE_AFFINITY":  "storage_node_key1=storage_node_value1; storage_node_key2=storage_node_value2;",
+				// "CSI_PROVISIONER_NODE_AFFINITY": "storage_node_key1=storage_node_value1; storage_node_key2=storage_node_value2;",
+				// "CSI_PLUGIN_NODE_AFFINITY":      "node_key1=node_value1; node_key2=node_value2;",
+			}
+
+			envs := deploy.Spec.Template.Spec.Containers[0].Env
+			envVars := make(map[string]string, len(envs))
+			for _, en := range envs {
+				envVars[en.Name] = en.Value
+			}
+
+			for k, v := range subTCs {
+				k, v := k, v
+				t.Run(k, func(t *testing.T) {
+					gotVal, ok := envVars[k]
+					if !ok {
+						t.Fatalf("expected: %q env", k)
+					}
+
+					if gotVal != v {
+						t.Fatalf("expected value of env %q:%q, got %q:%q", k, v, k, gotVal)
+					}
+				})
+			}
+		}},
+	}
+
+	for _, tc := range tcs {
+		tc := tc
+		t.Run(tc.Name, func(t *testing.T) {
+			t.Parallel()
+			tc.Test(t, m)
+		})
+	}
+}