add integration test for protobuf codec

Signed-off-by: Ahmed Hassan <[email protected]>
afhassan · Sep 20, 2024 · 4901a28 · 4901a28
1 parent 9b6ef8c
commit 4901a28
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/integration/query_frontend_test.go b/integration/query_frontend_test.go
@@ -203,6 +203,25 @@ func TestQueryFrontendWithVerticalShardingQueryScheduler(t *testing.T) {
 	})
 }
 
+func TestQueryFrontendProtobufCodec(t *testing.T) {
+	runQueryFrontendTest(t, queryFrontendTestConfig{
+		testMissingMetricName: false,
+		querySchedulerEnabled: true,
+		queryStatsEnabled:     true,
+		setup: func(t *testing.T, s *e2e.Scenario) (configFile string, flags map[string]string) {
+			require.NoError(t, writeFileToSharedDir(s, cortexConfigFile, []byte(BlocksStorageConfig)))
+
+			minio := e2edb.NewMinio(9000, BlocksStorageFlags()["-blocks-storage.s3.bucket-name"])
+			require.NoError(t, s.StartAndWaitReady(minio))
+
+			flags = mergeFlags(e2e.EmptyFlags(), map[string]string{
+				"-api.querier-default-codec": "protobuf",
+			})
+			return cortexConfigFile, flags
+		},
+	})
+}
+
 func TestQueryFrontendRemoteRead(t *testing.T) {
 	runQueryFrontendTest(t, queryFrontendTestConfig{
 		remoteReadEnabled: true,