Fix: Overriding user defined labels

This PR fixes an issue where helm's postrenderer
used to override the user defined object labels instead
of merging them.

Signed-off-by: Varsha Prasad Narsing <[email protected]>

internal/controllers/bundledeployment/bundledeployment.go

@@ -519,7 +519,7 @@ func (p *postrenderer) Run(renderedManifests *bytes.Buffer) (*bytes.Buffer, erro
 		if err != nil {
 			return nil, err
 		}
-		obj.SetLabels(p.labels)
+		obj.SetLabels(util.MergeMaps(obj.GetLabels(), p.labels))
 		b, err := obj.MarshalJSON()
 		if err != nil {
 			return nil, err

internal/controllers/bundledeployment/bundledeployment_test.go

@@ -0,0 +1,96 @@
+package bundledeployment
+
+import (
+	"bytes"
+	"encoding/json"
+	"testing"
+
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+	"helm.sh/helm/v3/pkg/postrender"
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+
+	rukpakv1alpha1 "github.com/operator-framework/rukpak/api/v1alpha1"
+	"github.com/operator-framework/rukpak/internal/util"
+)
+
+func TestBundleDeploymentController(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "BundleDeployment Controller Suite")
+}
+
+var _ = Describe("BundleDeployment", func() {
+	var _ = Describe("PostRenderer", func() {
+		var (
+			postren postrender.PostRenderer
+			pod     corev1.Pod
+			inBuf   bytes.Buffer
+		)
+
+		BeforeEach(func() {
+			postren = &postrenderer{
+				labels: map[string]string{
+					util.CoreOwnerKindKey: rukpakv1alpha1.BundleDeploymentKind,
+					util.CoreOwnerNameKey: "test-owner",
+				},
+			}
+
+			pod = corev1.Pod{
+				ObjectMeta: metav1.ObjectMeta{
+					Name: "testPod",
+					Labels: map[string]string{
+						"testKey": "testValue",
+					},
+				},
+			}
+			pod.SetGroupVersionKind(schema.GroupVersionKind{Group: "", Version: "v1", Kind: "Pod"})
+
+		})
+
+		It("should preserve existing labels present in the pod", func() {
+			By("encoding pod")
+			Expect(json.NewEncoder(&inBuf).Encode(pod)).NotTo(HaveOccurred())
+
+			By("verifying if postrender ran successfully")
+			outBuf, err := postren.Run(&inBuf)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(outBuf.String()).NotTo(BeEmpty())
+
+			By("converting output string back to pod")
+			renderedPod := &corev1.Pod{}
+			Expect(json.Unmarshal(outBuf.Bytes(), renderedPod)).NotTo(HaveOccurred())
+			Expect(renderedPod).NotTo(BeNil())
+
+			labels := renderedPod.GetLabels()
+			Expect(len(labels)).To(BeEquivalentTo(3))
+			Expect(labels).Should(HaveKeyWithValue("testKey", "testValue"))
+			Expect(labels).Should(HaveKeyWithValue("core.rukpak.io/owner-kind", "BundleDeployment"))
+			Expect(labels).Should(HaveKeyWithValue("core.rukpak.io/owner-name", "test-owner"))
+		})
+
+		It("should add default labels to the pod", func() {
+			pod.SetLabels(map[string]string{})
+
+			By("encoding pod")
+			Expect(json.NewEncoder(&inBuf).Encode(pod)).NotTo(HaveOccurred())
+
+			By("verifying if postrender ran successfully")
+			outBuf, err := postren.Run(&inBuf)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(outBuf.String()).NotTo(BeEmpty())
+
+			By("converting output string back to pod")
+			renderedPod := &corev1.Pod{}
+			Expect(json.Unmarshal(outBuf.Bytes(), renderedPod)).NotTo(HaveOccurred())
+			Expect(renderedPod).NotTo(BeNil())
+
+			labels := renderedPod.GetLabels()
+			Expect(len(labels)).To(BeEquivalentTo(2))
+			Expect(labels).Should(HaveKeyWithValue("core.rukpak.io/owner-kind", "BundleDeployment"))
+			Expect(labels).Should(HaveKeyWithValue("core.rukpak.io/owner-name", "test-owner"))
+		})
+
+	})
+})