Function TestPopulateWithDelSeriesIterator_WithST

tsdb/querier_test.go:2065–2146

TestPopulateWithDelSeriesIterator_WithST tests that ST (start time) values are correctly preserved when iterating through chunks with ST support.

(t *testing.T)

// TestPopulateWithDelSeriesIterator_WithST tests that ST (start time) values are
// correctly preserved when iterating through chunks with ST support.
func TestPopulateWithDelSeriesIterator_WithST(t *testing.T) {
	// Samples with non-zero ST values to test ST preservation.
	samplesWithST := [][]chunks.Sample{
		{
			sample{st: 100, t: 1000, f: 1.0},
			sample{st: 200, t: 2000, f: 2.0},
			sample{st: 300, t: 3000, f: 3.0},
		},
	}

	// Samples with varying ST patterns.
	samplesVaryingST := [][]chunks.Sample{
		{
			sample{st: 0, t: 1000, f: 1.0},    // st=0
			sample{st: 1500, t: 1500, f: 1.5}, // st=t
			sample{st: 1900, t: 2000, f: 2.0}, // st=t-100
			sample{st: 500, t: 3000, f: 3.0},  // st < t
		},
	}

	cases := []struct {
		name     string
		samples  [][]chunks.Sample
		expected []chunks.Sample
	}{
		{
			name:    "all samples have non-zero ST",
			samples: samplesWithST,
			expected: []chunks.Sample{
				sample{st: 100, t: 1000, f: 1.0},
				sample{st: 200, t: 2000, f: 2.0},
				sample{st: 300, t: 3000, f: 3.0},
			},
		},
		{
			name:    "samples with varying ST patterns",
			samples: samplesVaryingST,
			expected: []chunks.Sample{
				sample{st: 0, t: 1000, f: 1.0},
				sample{st: 1500, t: 1500, f: 1.5},
				sample{st: 1900, t: 2000, f: 2.0},
				sample{st: 500, t: 3000, f: 3.0},
			},
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			// Test with chunks (not iterables).
			t.Run("chunks", func(t *testing.T) {
				f, chkMetas := createFakeReaderAndNotPopulatedChunks(tc.samples...)
				it := &populateWithDelSeriesIterator{}
				it.reset(ulid.ULID{}, f, chkMetas, nil)

				var result []chunks.Sample
				for it.Next() != chunkenc.ValNone {
					st := it.AtST()
					ts, v := it.At()