chore: restructured test files

Author: Ash-exp

kubelink/pkg/service/commonHelmService/k8sService_integration_test.go

@@ -0,0 +1,179 @@
+/*
+ * Copyright (c) 2024. Devtron Inc.
+ *
+ * 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.
+ */
+
+package commonHelmService
+
+import (
+	"context"
+	"fmt"
+	"github.com/devtron-labs/common-lib/utils/k8s"
+	"github.com/devtron-labs/kubelink/config"
+	"github.com/devtron-labs/kubelink/internals/logger"
+	"github.com/devtron-labs/kubelink/pkg/service/commonHelmService/testdata"
+	"k8s.io/client-go/rest"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+)
+
+func getTestCases() []testdata.TestCase {
+	return []testdata.TestCase{
+		testdata.GetFluxCDDeploymentTestCase(),
+		testdata.FluentBitDaemonsetTestCase(),
+		testdata.GetRolloutDeploymentTestCase(),
+		testdata.GetDeploymentTestCase(),
+		testdata.GetCronJobTestCase(),
+		testdata.GetMongodbCommunityTestCase(),
+		testdata.GetKubePrometheusDaemonsetTestcase(),
+		testdata.GetKubePrometheusDeploymentTestcase(),
+		testdata.GetKubePrometheusAlertManagerTestcase(),
+		testdata.GetKubePrometheusTestcase(),
+	}
+}
+
+func TestGetChildObjectsIntegration(t *testing.T) {
+	// Mock K8sService implementation
+	sugaredLogger := logger.NewSugaredLogger()
+	helmReleaseConfig, err := config.GetHelmReleaseConfig()
+	if err != nil {
+		t.Fatalf("Failed to get helm release config: %v", err)
+		return
+	}
+	helmReleaseConfig.ParentChildGvkMapping = testdata.GetParentChildGvkMapForIntegrationTest()
+	k8sService, err := NewK8sServiceImpl(sugaredLogger, helmReleaseConfig)
+	if err != nil {
+		t.Fatalf("Failed to create K8sService: %v", err)
+		return
+	}
+	runtimeCfg, err := k8s.GetRuntimeConfig()
+	if err != nil {
+		t.Fatalf("Failed to get runtime config: %v", err)
+		return
+	}
+	runtimeCfg.LocalDevMode = true
+	k8sUtil, err := k8s.NewK8sUtil(sugaredLogger, runtimeCfg)
+	if err != nil {
+		t.Fatalf("Failed to create K8sUtil: %v", err)
+		return
+	}
+	restConfig, err := k8sUtil.GetK8sInClusterRestConfig()
+	if err != nil {
+		t.Fatalf("Failed to get rest config: %v", err)
+		return
+	}
+	for _, tt := range getTestCases() {
+		t.Run(tt.Name, func(t *testing.T) {
+			// Create context
+			ctx, cancel := context.WithCancel(context.Background())
+			t.Cleanup(func() {
+				// cleanup logic
+				cancel()
+				_ = tt.Cleanup()
+			})
+			// setup logic
+			if err = tt.Setup(ctx); err != nil {
+				t.Fatalf("Failed to setup test: %v", err)
+				return
+			}
+			testGetChildObjects(t, tt, k8sService, restConfig)
+			return
+		})
+	}
+	return
+}
+
+func testGetChildObjects(t *testing.T, tt testdata.TestCase, k8sService K8sService, restConfig *rest.Config) {
+	t.Run(fmt.Sprintf("GVK-%q", tt.ParentGvk.String()), func(t *testing.T) {
+		parentApiVersion, _ := tt.ParentGvk.ToAPIVersionAndKind()
+		resultV1, errV1 := k8sService.GetChildObjectsV1(restConfig, tt.Namespace, tt.ParentGvk, tt.ParentName, parentApiVersion)
+		resultV2, errV2 := k8sService.GetChildObjectsV2(restConfig, tt.Namespace, tt.ParentGvk, tt.ParentName)
+		assert.Truef(t, compareValidResourceTree(resultV1, resultV2), "Resource trees do not match for %s", tt.Name)
+		if errV1 != nil {
+			assert.EqualError(t, errV1, errV2.Error(), "Errors do not match for %s", tt.Name)
+		} else {
+			assert.NoError(t, errV2, "Errors do not match for %s", tt.Name)
+		}
+		fmt.Println("=== Child Object Count:", len(resultV1))
+		for _, manifest := range resultV1 {
+			gvk := manifest.GroupVersionKind()
+			if k8sService.CanHaveChild(gvk) {
+				tt.ParentGvk = gvk
+				tt.ParentName = manifest.GetName()
+				tt.Namespace = manifest.GetNamespace()
+				// Recursively call the test function for child objects
+				testGetChildObjects(t, tt, k8sService, restConfig)
+			}
+		}
+	})
+}
+
+// compareValidResourceTree compares two slices of unstructured.Unstructured objects
+func compareValidResourceTree(v1Result []*unstructured.Unstructured, v2Result []*unstructured.Unstructured) bool {
+	if len(v1Result) != len(v2Result) {
+		return false
+	}
+	// group by gvk and compare for each gvk
+	v1Map := make(map[string][]*unstructured.Unstructured)
+	v2Map := make(map[string][]*unstructured.Unstructured)
+	for _, v1Obj := range v1Result {
+		gvk := v1Obj.GroupVersionKind()
+		key := gvk.String()
+		v1Map[key] = append(v1Map[key], v1Obj)
+	}
+	// group by gvk and compare for each gvk
+	for _, v2Obj := range v2Result {
+		gvk := v2Obj.GroupVersionKind()
+		key := gvk.String()
+		v2Map[key] = append(v2Map[key], v2Obj)
+	}
+	// compare length of each gvk
+	if len(v1Map) != len(v2Map) {
+		return false
+	}
+	// compare each gvk
+	for key, v1Objs := range v1Map {
+		v2Objs, ok := v2Map[key]
+		if !ok {
+			return false
+		}
+		if len(v1Objs) != len(v2Objs) {
+			return false
+		}
+		for i, v1Obj := range v1Objs {
+			v2Obj := v2Objs[i]
+			if v1Obj.GetKind() != v2Obj.GetKind() {
+				return false
+			}
+			if v1Obj.GetAPIVersion() != v2Obj.GetAPIVersion() {
+				return false
+			}
+			if v1Obj.GetNamespace() != v2Obj.GetNamespace() {
+				return false
+			}
+			if v1Obj.GetName() != v2Obj.GetName() {
+				return false
+			}
+			if v1Obj.GetResourceVersion() != v2Obj.GetResourceVersion() {
+				return false
+			}
+			if v1Obj.GetCreationTimestamp() != v2Obj.GetCreationTimestamp() {
+				return false
+			}
+		}
+	}
+	return true
+}