From aad05e3889e245bb25575b9ae08545277d8c40f6 Mon Sep 17 00:00:00 2001 From: bwplotka Date: Mon, 9 Mar 2026 16:16:00 +0000 Subject: [PATCH] tsdb/record: add test for clear() on histograms --- tsdb/record/buffers_test.go | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 tsdb/record/buffers_test.go diff --git a/tsdb/record/buffers_test.go b/tsdb/record/buffers_test.go new file mode 100644 index 0000000000..688c264775 --- /dev/null +++ b/tsdb/record/buffers_test.go @@ -0,0 +1,35 @@ +package record + +import ( + "testing" + "github.com/prometheus/prometheus/model/histogram" + "github.com/stretchr/testify/require" +) + +func TestBuffersPool_PtrClear(t *testing.T) { + pool := NewBuffersPool() + + h := pool.GetHistograms(1) + h = append(h, RefHistogramSample{ + H: &histogram.Histogram{}, + }) + pool.PutHistograms(h) + + h2 := pool.GetHistograms(1) + require.Len(t, h2, 0) + require.Equal(t, 1, cap(h2)) + h2 = h2[:1] // extend to capacity to check previously stored item + require.Nil(t, h2[0].H) + + fh := pool.GetFloatHistograms(1) + fh = append(fh, RefFloatHistogramSample{ + FH: &histogram.FloatHistogram{}, + }) + pool.PutFloatHistograms(fh) + + fh2 := pool.GetFloatHistograms(1) + require.Len(t, fh2, 0) + require.Equal(t, 1, cap(fh2)) + fh2 = fh2[:1] // extend to capacity + require.Nil(t, fh2[0].FH) +}