Function TestGetPostingForLabels

pkg/ingester/user_state_test.go:226–282

(t *testing.T)

}

func TestGetPostingForLabels(t *testing.T) {
	ctx := context.Background()
	testErr := errors.New("err")
	for _, tc := range []struct {
		name     string
		ir       tsdb.IndexReader
		lbls     labels.Labels
		postings []storage.SeriesRef
		err      error
	}{
		{
			name: "single label",
			ir: &mockIndexReader{postings: map[string]map[string]index.Postings{
				"foo": {"bar": index.NewListPostings([]storage.SeriesRef{1, 2, 3, 4, 5})},
			}},
			lbls:     labels.FromStrings("foo", "bar"),
			postings: []storage.SeriesRef{1, 2, 3, 4, 5},
		},
		{
			name: "intersect",
			ir: &mockIndexReader{postings: map[string]map[string]index.Postings{
				"foo": {"bar": index.NewListPostings([]storage.SeriesRef{1, 2, 3, 4, 5})},
				"job": {"prometheus": index.NewListPostings([]storage.SeriesRef{1, 3, 5})},
			}},
			lbls:     labels.FromStrings("foo", "bar", "job", "prometheus"),
			postings: []storage.SeriesRef{1, 3, 5},
		},
		{
			name: "error",
			ir: &mockIndexReader{postings: map[string]map[string]index.Postings{
				"foo": {"bar": index.ErrPostings(testErr)},
				"job": {"prometheus": index.NewListPostings([]storage.SeriesRef{1, 3, 5})},
			}},
			lbls:     labels.FromStrings("foo", "bar", "job", "prometheus"),
			postings: []storage.SeriesRef{1, 3, 5},
			err:      testErr,
		},
	} {
		t.Run(tc.name, func(t *testing.T) {
			p, err := getPostingForLabels(ctx, tc.ir, tc.lbls)
			if err != nil {
				require.EqualError(t, err, tc.err.Error())
			} else {
				require.NoError(t, err)
				series := make([]storage.SeriesRef, 0)
				for p.Next() {
					series = append(series, p.At())
				}
				if tc.err != nil {
					require.EqualError(t, p.Err(), tc.err.Error())
				} else {
					require.Equal(t, tc.postings, series)
				}
			}
		})
	}
}