Function createOverlappingSeriesHandler

storage/remote/client_test.go:1060–1096

createOverlappingSeriesHandler creates responses with same series from multiple queries.

(t *testing.T, queries []*prompb.Query)

// createOverlappingSeriesHandler creates responses with same series from multiple queries.
func createOverlappingSeriesHandler(t *testing.T, queries []*prompb.Query) http.HandlerFunc {
	return http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
		w.Header().Set("Content-Type", "application/x-streamed-protobuf; proto=prometheus.ChunkedReadResponse")

		flusher, ok := w.(http.Flusher)
		require.True(t, ok)

		cw := NewChunkedWriter(w, flusher)

		// Same series labels for both queries (will be merged)
		commonLabels := []prompb.Label{
			{Name: "__name__", Value: "up"},
			{Name: "job", Value: "prometheus"},
		}

		// Send response for each query with the same series
		for queryIndex := range queries {
			chunk := buildTestChunks(t)[0] // Use first chunk with 5 samples

			cSeries := prompb.ChunkedSeries{
				Labels: commonLabels,
				Chunks: []prompb.Chunk{chunk},
			}

			readResp := prompb.ChunkedReadResponse{
				ChunkedSeries: []*prompb.ChunkedSeries{&cSeries},
				QueryIndex:    int64(queryIndex),
			}

			b, err := proto.Marshal(&readResp)
			require.NoError(t, err)

			_, err = cw.Write(b)
			require.NoError(t, err)
		}
	})
}