Add data compression unit tests

Signed-off-by: alexander-demicev <alexandr.demicev@suse.com>

Author: alexander-demicev

internal/controller/manifests_downloader_test.go

@@ -17,7 +17,9 @@ limitations under the License.
 package controller
 
 import (
+	"bytes"
 	"context"
+	"strings"
 	"testing"
 
 	. "github.com/onsi/gomega"
@@ -126,3 +128,59 @@ func TestProviderDownloadWithOverrides(t *testing.T) {
 	g.Expect(p.components.Images()).To(HaveExactElements([]string{"registry.k8s.io/cluster-api/cluster-api-controller:v1.4.3"}))
 	g.Expect(p.components.Version()).To(Equal("v1.4.3"))
 }
+
+func TestCompressDecompressRoundtrip(t *testing.T) {
+	g := NewWithT(t)
+
+	original := []byte("apiVersion: v1\nkind: ConfigMap\nmetadata:\n  name: test\n")
+
+	var buf bytes.Buffer
+	err := compressData(&buf, original)
+	g.Expect(err).ToNot(HaveOccurred())
+	g.Expect(buf.Len()).To(BeNumerically(">", 0))
+
+	decompressed, err := decompressData(buf.Bytes())
+	g.Expect(err).ToNot(HaveOccurred())
+	g.Expect(decompressed).To(Equal(original))
+}
+
+func TestCompressDataEmptyInput(t *testing.T) {
+	g := NewWithT(t)
+
+	var buf bytes.Buffer
+	err := compressData(&buf, []byte{})
+	g.Expect(err).ToNot(HaveOccurred())
+
+	decompressed, err := decompressData(buf.Bytes())
+	g.Expect(err).ToNot(HaveOccurred())
+	g.Expect(decompressed).To(BeEmpty())
+}
+
+func TestDecompressDataInvalidInput(t *testing.T) {
+	g := NewWithT(t)
+
+	_, err := decompressData([]byte("not gzip data"))
+	g.Expect(err).To(HaveOccurred())
+	g.Expect(err.Error()).To(ContainSubstring("cannot open gzip reader"))
+}
+
+func TestCompressDecompressLargeData(t *testing.T) {
+	g := NewWithT(t)
+
+	// Create data larger than maxConfigMapSize to test needToCompress
+	largeData := []byte(strings.Repeat("x", maxConfigMapSize+1))
+
+	g.Expect(needToCompress(largeData)).To(BeTrue())
+	g.Expect(needToCompress([]byte("small"))).To(BeFalse())
+
+	var buf bytes.Buffer
+	err := compressData(&buf, largeData)
+	g.Expect(err).ToNot(HaveOccurred())
+
+	// Compressed size should be much smaller than original for repetitive data
+	g.Expect(buf.Len()).To(BeNumerically("<", len(largeData)))
+
+	decompressed, err := decompressData(buf.Bytes())
+	g.Expect(err).ToNot(HaveOccurred())
+	g.Expect(decompressed).To(Equal(largeData))
+}

internal/controller/suite_test.go

@@ -87,6 +87,39 @@ func TestMain(m *testing.M) {
 		panic(fmt.Sprintf("Failed to start ControlPlaneProviderReconciler: %v", err))
 	}
 
+	if err := (&GenericProviderReconciler{
+		Provider:                 &operatorv1.AddonProvider{},
+		ProviderList:             &operatorv1.AddonProviderList{},
+		Client:                   env,
+		WatchConfigSecretChanges: true,
+		WatchConfigMapChanges:    true,
+		WatchCoreProviderChanges: true,
+	}).SetupWithManager(ctx, env.Manager, controller.Options{MaxConcurrentReconciles: 1}); err != nil {
+		panic(fmt.Sprintf("Failed to start AddonProviderReconciler: %v", err))
+	}
+
+	if err := (&GenericProviderReconciler{
+		Provider:                 &operatorv1.IPAMProvider{},
+		ProviderList:             &operatorv1.IPAMProviderList{},
+		Client:                   env,
+		WatchConfigSecretChanges: true,
+		WatchConfigMapChanges:    true,
+		WatchCoreProviderChanges: true,
+	}).SetupWithManager(ctx, env.Manager, controller.Options{MaxConcurrentReconciles: 1}); err != nil {
+		panic(fmt.Sprintf("Failed to start IPAMProviderReconciler: %v", err))
+	}
+
+	if err := (&GenericProviderReconciler{
+		Provider:                 &operatorv1.RuntimeExtensionProvider{},
+		ProviderList:             &operatorv1.RuntimeExtensionProviderList{},
+		Client:                   env,
+		WatchConfigSecretChanges: true,
+		WatchConfigMapChanges:    true,
+		WatchCoreProviderChanges: true,
+	}).SetupWithManager(ctx, env.Manager, controller.Options{MaxConcurrentReconciles: 1}); err != nil {
+		panic(fmt.Sprintf("Failed to start RuntimeExtensionProviderReconciler: %v", err))
+	}
+
 	go func() {
 		if err := env.Start(ctx); err != nil {
 			panic(fmt.Sprintf("Failed to start the envtest manager: %v", err))