Function TestBufferedSeriesIterator

storage/buffer_test.go:231–299

(t *testing.T)

}

func TestBufferedSeriesIterator(t *testing.T) {
	var it *BufferedSeriesIterator

	bufferEq := func(exp []fSample) {
		var b []fSample
		bit := it.Buffer()
		for bit.Next() == chunkenc.ValFloat {
			t, f := bit.At()
			st := bit.AtST()
			b = append(b, fSample{st: st, t: t, f: f})
		}
		require.Equal(t, exp, b, "buffer mismatch")
	}
	sampleEq := func(est, ets int64, ev float64) {
		ts, v := it.At()
		st := it.AtST()
		require.Equal(t, est, st, "start timestamp mismatch")
		require.Equal(t, ets, ts, "timestamp mismatch")
		require.Equal(t, ev, v, "value mismatch")
	}
	prevSampleEq := func(est, ets int64, ev float64, eok bool) {
		s, ok := it.PeekBack(1)
		require.Equal(t, eok, ok, "exist mismatch")
		require.Equal(t, est, s.ST(), "start timestamp mismatch")
		require.Equal(t, ets, s.T(), "timestamp mismatch")
		require.Equal(t, ev, s.F(), "value mismatch")
	}

	it = NewBufferIterator(NewListSeriesIterator(samples{
		fSample{st: -1, t: 1, f: 2},
		fSample{st: 1, t: 2, f: 3},
		fSample{st: 2, t: 3, f: 4},
		fSample{st: 3, t: 4, f: 5},
		fSample{st: 3, t: 5, f: 6},
		fSample{st: 50, t: 99, f: 8},
		fSample{st: 99, t: 100, f: 9},
		fSample{st: 100, t: 101, f: 10},
	}), 2)

	require.Equal(t, chunkenc.ValFloat, it.Seek(-123), "seek failed")
	sampleEq(-1, 1, 2)
	prevSampleEq(0, 0, 0, false)
	bufferEq(nil)

	require.Equal(t, chunkenc.ValFloat, it.Next(), "next failed")
	sampleEq(1, 2, 3)
	prevSampleEq(-1, 1, 2, true)
	bufferEq([]fSample{{st: -1, t: 1, f: 2}})

	require.Equal(t, chunkenc.ValFloat, it.Next(), "next failed")
	require.Equal(t, chunkenc.ValFloat, it.Next(), "next failed")
	require.Equal(t, chunkenc.ValFloat, it.Next(), "next failed")
	sampleEq(3, 5, 6)
	prevSampleEq(3, 4, 5, true)
	bufferEq([]fSample{{st: 1, t: 2, f: 3}, {st: 2, t: 3, f: 4}, {st: 3, t: 4, f: 5}})

	require.Equal(t, chunkenc.ValFloat, it.Seek(5), "seek failed")
	sampleEq(3, 5, 6)