Function mockHistogramWriteRequest

pkg/ingester/ingester_test.go:4343–4396

(t *testing.T, lbls labels.Labels, value int64, timestampMs int64, float bool)

}

func mockHistogramWriteRequest(t *testing.T, lbls labels.Labels, value int64, timestampMs int64, float bool) (*cortexpb.WriteRequest, *client.QueryStreamResponse) {
	var (
		histograms []cortexpb.WrappedHistogram
		h          *histogram.Histogram
		fh         *histogram.FloatHistogram
		c          chunkenc.Chunk
	)
	if float {
		fh = tsdbutil.GenerateTestFloatHistogram(value)
		histograms = []cortexpb.WrappedHistogram{
			cortexpb.WrapHistogram(cortexpb.FloatHistogramToHistogramProto(timestampMs, fh)),
		}
		c = chunkenc.NewFloatHistogramChunk()
	} else {
		h = tsdbutil.GenerateTestHistogram(value)
		histograms = []cortexpb.WrappedHistogram{
			cortexpb.WrapHistogram(cortexpb.HistogramToHistogramProto(timestampMs, h)),
		}
		c = chunkenc.NewHistogramChunk()
	}

	app, err := c.Appender()
	require.NoError(t, err)
	if float {
		_, _, _, err = app.AppendFloatHistogram(nil, timestampMs, fh, true)
	} else {
		_, _, _, err = app.AppendHistogram(nil, timestampMs, h, true)
	}
	require.NoError(t, err)
	c.Compact()

	req := cortexpb.ToWriteRequest([]labels.Labels{lbls}, nil, nil, histograms, cortexpb.API)
	enc := int32(encoding.PrometheusHistogramChunk)
	if float {
		enc = int32(encoding.PrometheusFloatHistogramChunk)
	}
	expectedQueryStreamResChunks := &client.QueryStreamResponse{
		Chunkseries: []client.TimeSeriesChunk{
			{
				Labels: cortexpb.FromLabelsToLabelAdapters(lbls),
				Chunks: []client.Chunk{
					{
						StartTimestampMs: timestampMs,
						EndTimestampMs:   timestampMs,
						Encoding:         enc,
						Data:             c.Bytes(),
					},
				},
			},
		},
	}

	return req, expectedQueryStreamResChunks
}

func prepareIngesterWithBlocksStorage(t testing.TB, ingesterCfg Config, registerer prometheus.Registerer) (*Ingester, error) {
	limits := defaultLimitsTestConfig()