feat(subscriber): Add unit tests for k8s operations

chaoscenter/subscriber/go.mod

@@ -23,6 +23,7 @@ require (
 	github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578 // indirect
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/emicklei/go-restful v2.16.0+incompatible // indirect
+	github.com/evanphx/json-patch v5.6.0+incompatible // indirect
 	github.com/go-logr/logr v1.2.3 // indirect
 	github.com/go-openapi/jsonpointer v0.19.5 // indirect
 	github.com/go-openapi/jsonreference v0.19.6 // indirect

chaoscenter/subscriber/pkg/k8s/operations_test.go

@@ -0,0 +1,130 @@
+package k8s
+
+import (
+	"context"
+	"reflect"
+	"testing"
+
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/client-go/dynamic/fake"
+)
+
+func TestAddCustomLabels(t *testing.T) {
+	// 1. ARRANGE
+	obj := &unstructured.Unstructured{}
+	obj.SetLabels(map[string]string{"existing": "label"})
+	labelsToAdd := map[string]string{
+		"new_label": "value1",
+		"owner":     "test",
+	}
+	expected := map[string]string{
+		"existing":  "label",
+		"new_label": "value1",
+		"owner":     "test",
+	}
+
+	// 2. ACT
+	addCustomLabels(obj, labelsToAdd)
+
+	// 3. ASSERT
+	result := obj.GetLabels()
+	if !reflect.DeepEqual(result, expected) {
+		t.Errorf("Test failed: expected labels %v, but got %v", expected, result)
+	}
+}
+
+func TestApplyRequest(t *testing.T) {
+	// 1. ARRANGE: Create a fake pod object to "get"
+	fakePod := &corev1.Pod{
+		TypeMeta: metav1.TypeMeta{
+			Kind:       "Pod",
+			APIVersion: "v1",
+		},
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "my-fake-pod",
+			Namespace: "litmus",
+		},
+		Spec: corev1.PodSpec{
+			Containers: []corev1.Container{
+				{Name: "test-container", Image: "test-image"},
+			},
+		},
+	}
+
+	// Convert the typed Pod into an Unstructured object for the test
+	unstructuredPod, err := runtime.DefaultUnstructuredConverter.ToUnstructured(fakePod)
+	if err != nil {
+		t.Fatalf("Failed to convert fake pod: %v", err)
+	}
+	objToGet := &unstructured.Unstructured{Object: unstructuredPod}
+
+	// Create the fake dynamic client and pre-load it with our fake pod
+	// We also need to register the "Pod" type with the client's scheme
+	s := runtime.NewScheme()
+	s.AddKnownTypes(corev1.SchemeGroupVersion, &corev1.Pod{})
+
+	fakeDynamicClient := fake.NewSimpleDynamicClient(s, objToGet)
+
+	// The 'dr' (dynamic resource) variable needs to be set, just like in the real code
+	// We tell it to look for "pods" in the "litmus" namespace
+	dr = fakeDynamicClient.Resource(corev1.SchemeGroupVersion.WithResource("pods")).Namespace("litmus")
+
+	// 2. ACT: Call the 'get' operation
+	response, err := applyRequest("get", objToGet)
+
+	// 3. ASSERT: Check that we got our fake pod back
+	if err != nil {
+		t.Fatalf("applyRequest 'get' failed: %v", err)
+	}
+	if response == nil {
+		t.Fatal("applyRequest 'get' returned nil response")
+	}
+	if response.GetName() != "my-fake-pod" {
+		t.Errorf("Expected pod 'my-fake-pod', but got '%s'", response.GetName())
+	}
+}
+
+func TestApplyRequest_Create(t *testing.T) {
+	// 1. ARRANGE: Create an object to be created
+	objToCreate := &unstructured.Unstructured{
+		Object: map[string]interface{}{
+			"apiVersion": "v1",
+			"kind":       "ConfigMap",
+			"metadata": map[string]interface{}{
+				"name":      "new-cm",
+				"namespace": "litmus",
+			},
+			"data": map[string]interface{}{
+				"key": "value",
+			},
+		},
+	}
+
+	// Create an EMPTY fake dynamic client (it has no objects yet)
+	s := runtime.NewScheme()
+	s.AddKnownTypes(corev1.SchemeGroupVersion, &corev1.ConfigMap{})
+	fakeDynamicClient := fake.NewSimpleDynamicClient(s)
+
+	// Set the global 'dr' variable for the test
+	dr = fakeDynamicClient.Resource(corev1.SchemeGroupVersion.WithResource("configmaps")).Namespace("litmus")
+
+	// 2. ACT: Call the 'create' operation
+	response, err := applyRequest("create", objToCreate)
+	if err != nil {
+		t.Fatalf("applyRequest 'create' failed: %v", err)
+	}
+
+	// 3. ASSERT: Check that the object was created
+	if response.GetName() != "new-cm" {
+		t.Errorf("Expected created object 'new-cm', but got '%s'", response.GetName())
+	}
+
+	// Also, try to "get" the object from the fake client to prove it's there
+	_, err = dr.Get(context.TODO(), "new-cm", metav1.GetOptions{})
+	if err != nil {
+		t.Errorf("Failed to 'get' the newly created object from the fake client: %v", err)
+	}
+}