Function TestStream

pkg/querier/batch/stream_test.go:14–58

(t *testing.T)

)

func TestStream(t *testing.T) {
	t.Parallel()
	forEncodings(t, func(t *testing.T, enc encoding.Encoding) {
		for i, tc := range []struct {
			input1, input2 []promchunk.Batch
			output         batchStream
		}{
			{
				input1: []promchunk.Batch{mkBatch(0, enc)},
				output: []promchunk.Batch{mkBatch(0, enc)},
			},

			{
				input1: []promchunk.Batch{mkBatch(0, enc)},
				input2: []promchunk.Batch{mkBatch(0, enc)},
				output: []promchunk.Batch{mkBatch(0, enc)},
			},

			{
				input1: []promchunk.Batch{mkBatch(0, enc)},
				input2: []promchunk.Batch{mkBatch(promchunk.BatchSize, enc)},
				output: []promchunk.Batch{mkBatch(0, enc), mkBatch(promchunk.BatchSize, enc)},
			},

			{
				input1: []promchunk.Batch{mkBatch(0, enc), mkBatch(promchunk.BatchSize, enc)},
				input2: []promchunk.Batch{mkBatch(promchunk.BatchSize/2, enc), mkBatch(2*promchunk.BatchSize, enc)},
				output: []promchunk.Batch{mkBatch(0, enc), mkBatch(promchunk.BatchSize, enc), mkBatch(2*promchunk.BatchSize, enc)},
			},

			{
				input1: []promchunk.Batch{mkBatch(promchunk.BatchSize/2, enc), mkBatch(3*promchunk.BatchSize/2, enc), mkBatch(5*promchunk.BatchSize/2, enc)},
				input2: []promchunk.Batch{mkBatch(0, enc), mkBatch(promchunk.BatchSize, enc), mkBatch(3*promchunk.BatchSize, enc)},
				output: []promchunk.Batch{mkBatch(0, enc), mkBatch(promchunk.BatchSize, enc), mkBatch(2*promchunk.BatchSize, enc), mkBatch(3*promchunk.BatchSize, enc)},
			},
		} {
			t.Run(strconv.Itoa(i), func(t *testing.T) {
				t.Parallel()
				result := make(batchStream, len(tc.input1)+len(tc.input2))
				result = mergeStreams(tc.input1, tc.input2, result, promchunk.BatchSize)
				require.Equal(t, tc.output, result)
			})
		}
	})
}

func mkBatch(from int64, enc encoding.Encoding) promchunk.Batch {
	var result promchunk.Batch