fix backward compatibility test

Signed-off-by: alanprot <[email protected]>

Author: alanprot

integration/backward_compatibility_test.go

@@ -18,17 +18,34 @@ import (
 	"github.com/cortexproject/cortex/integration/e2ecortex"
 )
 
+type versionsImagesFlags struct {
+	flagsForOldImage func(map[string]string) map[string]string
+	flagsForNewImage func(map[string]string) map[string]string
+}
+
 var (
 	// If you change the image tag, remember to update it in the preloading done
 	// by GitHub Actions too (see .github/workflows/test-build-deploy.yml).
-	previousVersionImages = map[string]func(map[string]string) map[string]string{
-		"quay.io/cortexproject/cortex:v1.13.1": func(m map[string]string) map[string]string {
-			m["-ingester.stream-chunks-when-using-blocks"] = "true"
-			return m
+	previousVersionImages = map[string]*versionsImagesFlags{
+		"quay.io/cortexproject/cortex:v1.13.1": {
+			flagsForOldImage: func(m map[string]string) map[string]string {
+				m["-ingester.stream-chunks-when-using-blocks"] = "true"
+				return m
+			},
+			flagsForNewImage: func(m map[string]string) map[string]string {
+				m["-ingester.client.grpc-compression"] = ""
+				return m
+			},
 		},
-		"quay.io/cortexproject/cortex:v1.13.2": func(m map[string]string) map[string]string {
-			m["-ingester.stream-chunks-when-using-blocks"] = "true"
-			return m
+		"quay.io/cortexproject/cortex:v1.13.2": {
+			flagsForOldImage: func(m map[string]string) map[string]string {
+				m["-ingester.stream-chunks-when-using-blocks"] = "true"
+				return m
+			},
+			flagsForNewImage: func(m map[string]string) map[string]string {
+				m["-ingester.client.grpc-compression"] = ""
+				return m
+			},
 		},
 		"quay.io/cortexproject/cortex:v1.14.0": nil,
 		"quay.io/cortexproject/cortex:v1.14.1": nil,
@@ -44,11 +61,11 @@ var (
 )
 
 func TestBackwardCompatibilityWithBlocksStorage(t *testing.T) {
-	for previousImage, flagsFn := range previousVersionImages {
+	for previousImage, imagesFlags := range previousVersionImages {
 		t.Run(fmt.Sprintf("Backward compatibility upgrading from %s", previousImage), func(t *testing.T) {
 			flags := blocksStorageFlagsWithFlushOnShutdown()
-			if flagsFn != nil {
-				flags = flagsFn(flags)
+			if imagesFlags != nil && imagesFlags.flagsForOldImage != nil {
+				flags = imagesFlags.flagsForOldImage(flags)
 			}
 
 			runBackwardCompatibilityTestWithBlocksStorage(t, previousImage, flags)
@@ -57,14 +74,21 @@ func TestBackwardCompatibilityWithBlocksStorage(t *testing.T) {
 }
 
 func TestNewDistributorsCanPushToOldIngestersWithReplication(t *testing.T) {
-	for previousImage, flagsFn := range previousVersionImages {
+	for previousImage, imagesFlags := range previousVersionImages {
 		t.Run(fmt.Sprintf("Backward compatibility upgrading from %s", previousImage), func(t *testing.T) {
 			flags := blocksStorageFlagsWithFlushOnShutdown()
-			if flagsFn != nil {
-				flags = flagsFn(flags)
+			var flagsForNewImage func(map[string]string) map[string]string
+			if imagesFlags != nil {
+				if imagesFlags.flagsForOldImage != nil {
+					flags = imagesFlags.flagsForOldImage(flags)
+				}
+
+				if imagesFlags.flagsForNewImage != nil {
+					flagsForNewImage = imagesFlags.flagsForNewImage
+				}
 			}
 
-			runNewDistributorsCanPushToOldIngestersWithReplication(t, previousImage, flags)
+			runNewDistributorsCanPushToOldIngestersWithReplication(t, previousImage, flags, flagsForNewImage)
 		})
 	}
 }
@@ -127,7 +151,7 @@ func runBackwardCompatibilityTestWithBlocksStorage(t *testing.T, previousImage s
 }
 
 // Check for issues like https://github.com/cortexproject/cortex/issues/2356
-func runNewDistributorsCanPushToOldIngestersWithReplication(t *testing.T, previousImage string, flagsForPreviousImage map[string]string) {
+func runNewDistributorsCanPushToOldIngestersWithReplication(t *testing.T, previousImage string, flagsForPreviousImage map[string]string, flagsForNewImageFn func(map[string]string) map[string]string) {
 	s, err := e2e.NewScenario(networkName)
 	require.NoError(t, err)
 	defer s.Close()
@@ -141,6 +165,10 @@ func runNewDistributorsCanPushToOldIngestersWithReplication(t *testing.T, previo
 		"-distributor.replication-factor": "3",
 	})
 
+	if flagsForNewImageFn != nil {
+		flagsForNewImage = flagsForNewImageFn(flagsForNewImage)
+	}
+
 	// Start other Cortex components (ingester running on previous version).
 	ingester1 := e2ecortex.NewIngester("ingester-1", e2ecortex.RingStoreConsul, consul.NetworkHTTPEndpoint(), flagsForPreviousImage, previousImage)
 	ingester2 := e2ecortex.NewIngester("ingester-2", e2ecortex.RingStoreConsul, consul.NetworkHTTPEndpoint(), flagsForPreviousImage, previousImage)