Function TestHeadAppender_STStorage_ChunkEncoding

tsdb/head_test.go:8289–8356

TestHeadAppender_STStorage_ChunkEncoding verifies that the correct chunk encoding is used based on EnableSTStorage setting.

(t *testing.T)

// TestHeadAppender_STStorage_ChunkEncoding verifies that the correct chunk encoding
// is used based on EnableSTStorage setting.
func TestHeadAppender_STStorage_ChunkEncoding(t *testing.T) {
	samples := []struct {
		st      int64
		ts      int64
		fSample float64
	}{
		{st: 10, ts: 100, fSample: 1.0},
		{st: 20, ts: 200, fSample: 2.0},
	}

	for _, enableST := range []bool{false, true} {
		t.Run(fmt.Sprintf("EnableSTStorage=%t", enableST), func(t *testing.T) {
			opts := newTestHeadDefaultOptions(DefaultBlockDuration, false)
			opts.EnableSTStorage.Store(enableST)
			if enableST {
				opts.FloatChunkEncoding.Store(uint32(chunkenc.EncXOR2))
			} else {
				opts.FloatChunkEncoding.Store(uint32(chunkenc.EncXOR))
			}
			h, _ := newTestHeadWithOptions(t, compression.None, opts)

			lbls := labels.FromStrings("foo", "bar")
			a := h.Appender(context.Background())
			for _, s := range samples {
				_, err := a.AppendSTZeroSample(0, lbls, s.ts, s.st)
				require.NoError(t, err)
				_, err = a.Append(0, lbls, s.ts, s.fSample)
				require.NoError(t, err)
			}
			require.NoError(t, a.Commit())

			ctx := context.Background()
			idxReader, err := h.Index()
			require.NoError(t, err)
			defer idxReader.Close()

			chkReader, err := h.Chunks()
			require.NoError(t, err)
			defer chkReader.Close()

			p, err := idxReader.Postings(ctx, "foo", "bar")
			require.NoError(t, err)

			var lblBuilder labels.ScratchBuilder
			require.True(t, p.Next())
			sRef := p.At()

			var chkMetas []chunks.Meta
			require.NoError(t, idxReader.Series(sRef, &lblBuilder, &chkMetas))
			require.NotEmpty(t, chkMetas)

			for _, meta := range chkMetas {
				chk, iterable, err := chkReader.ChunkOrIterable(meta)
				require.NoError(t, err)
				require.Nil(t, iterable)

				encoding := chk.Encoding()
				if enableST {