Function createSampledResponseHandler

storage/remote/client_test.go:1011–1057

createSampledResponseHandler creates a mock handler for sampled responses.

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

// createSampledResponseHandler creates a mock handler for sampled responses.
func createSampledResponseHandler(t *testing.T, queries []*prompb.Query) http.HandlerFunc {
	return http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
		w.Header().Set("Content-Type", "application/x-protobuf")

		var results []*prompb.QueryResult
		for queryIndex, query := range queries {
			var timeseries []*prompb.TimeSeries

			// Create 2 series per query
			for seriesIndex := range 2 {
				var labels []prompb.Label
				if queryIndex == 0 {
					labels = []prompb.Label{
						{Name: "job", Value: "prometheus"},
						{Name: "instance", Value: fmt.Sprintf("localhost:%d", 9090+seriesIndex)},
					}
				} else {
					labels = []prompb.Label{
						{Name: "job", Value: "node_exporter"},
						{Name: "instance", Value: fmt.Sprintf("localhost:%d", 9100+seriesIndex)},
					}
				}

				// Create 2 samples per series within query time range
				samples := []prompb.Sample{
					{Timestamp: query.StartTimestampMs, Value: float64(queryIndex*10 + seriesIndex)},
					{Timestamp: query.EndTimestampMs, Value: float64(queryIndex*10 + seriesIndex + 1)},
				}

				timeseries = append(timeseries, &prompb.TimeSeries{
					Labels:  labels,
					Samples: samples,
				})
			}

			results = append(results, &prompb.QueryResult{Timeseries: timeseries})
		}

		resp := &prompb.ReadResponse{Results: results}
		data, err := proto.Marshal(resp)
		require.NoError(t, err)

		compressed := snappy.Encode(nil, data)
		_, err = w.Write(compressed)
		require.NoError(t, err)
	})
}

// createOverlappingSeriesHandler creates responses with same series from multiple queries.
func createOverlappingSeriesHandler(t *testing.T, queries []*prompb.Query) http.HandlerFunc {