Function TestSampleRingAtFloatHistogram

storage/buffer_test.go:162–229

(t *testing.T)

}

func TestSampleRingAtFloatHistogram(t *testing.T) {
	fh1 := tsdbutil.GenerateTestFloatHistogram(1)
	fh2 := tsdbutil.GenerateTestFloatHistogram(2)
	h1 := tsdbutil.GenerateTestHistogram(3)
	h2 := tsdbutil.GenerateTestHistogram(4)

	// With ValNone as the preferred type, nothing should be initialized.
	r := newSampleRing(10, 2, chunkenc.ValNone)
	require.Empty(t, r.fBuf)
	require.Empty(t, r.hBuf)
	require.Empty(t, r.fhBuf)
	require.Empty(t, r.iBuf)

	var (
		h  *histogram.Histogram
		fh *histogram.FloatHistogram
		ts int64
	)

	it := r.iterator()
	require.Equal(t, chunkenc.ValNone, it.Next())

	r.addFH(fhSample{st: 10, t: 11, fh: fh1})
	r.addFH(fhSample{st: 20, t: 21, fh: fh2})

	it = r.iterator()

	require.Equal(t, chunkenc.ValFloatHistogram, it.Next())
	ts, fh = it.AtFloatHistogram(fh)
	require.Equal(t, int64(11), ts)
	require.Equal(t, fh1, fh)
	require.Equal(t, int64(10), it.AtST())
	require.Equal(t, chunkenc.ValFloatHistogram, it.Next())
	ts, fh = it.AtFloatHistogram(fh)
	require.Equal(t, int64(21), ts)
	require.Equal(t, fh2, fh)
	require.Equal(t, int64(20), it.AtST())
	require.Equal(t, chunkenc.ValNone, it.Next())

	r.reset()
	it = r.iterator()
	require.Equal(t, chunkenc.ValNone, it.Next())

	r.addH(hSample{st: 30, t: 31, h: h1})
	r.addH(hSample{st: 40, t: 41, h: h2})

	it = r.iterator()

	require.Equal(t, chunkenc.ValHistogram, it.Next())
	ts, h = it.AtHistogram()
	require.Equal(t, int64(31), ts)
	require.Equal(t, h1, h)
	require.Equal(t, int64(30), it.AtST())
	ts, fh = it.AtFloatHistogram(fh)
	require.Equal(t, int64(31), ts)
	require.Equal(t, h1.ToFloat(nil), fh)
	require.Equal(t, int64(30), it.AtST())
	require.Equal(t, chunkenc.ValHistogram, it.Next())