Function Test_Ingester_LabelValues

pkg/ingester/ingester_test.go:3095–3159

(t *testing.T)

}

func Test_Ingester_LabelValues(t *testing.T) {
	series := []struct {
		lbls      labels.Labels
		value     float64
		timestamp int64
	}{
		{labels.FromStrings("__name__", "test_1", "route", "get_user", "status", "200"), 1, 100000},
		{labels.FromStrings("__name__", "test_1", "route", "get_user", "status", "500"), 1, 110000},
		{labels.FromStrings("__name__", "test_2"), 2, 200000},
	}

	expected := map[string][]string{
		"__name__": {"test_1", "test_2"},
		"status":   {"200", "500"},
		"route":    {"get_user"},
		"unknown":  {},
	}

	// Create ingester
	i, err := prepareIngesterWithBlocksStorage(t, defaultIngesterTestConfig(t), prometheus.NewRegistry())
	require.NoError(t, err)
	require.NoError(t, services.StartAndAwaitRunning(context.Background(), i))
	defer services.StopAndAwaitTerminated(context.Background(), i) //nolint:errcheck

	// Wait until it's ACTIVE
	test.Poll(t, 1*time.Second, ring.ACTIVE, func() any {
		return i.lifecycler.GetState()
	})

	// Push series
	ctx := user.InjectOrgID(context.Background(), "test")

	for _, series := range series {
		req, _ := mockWriteRequest(t, series.lbls, series.value, series.timestamp)
		_, err := i.Push(ctx, req)
		require.NoError(t, err)
	}

	tests := map[string]struct {
		limit int64
		match []*labels.Matcher
	}{
		"should return all label values if no limit is set": {
			limit: 0,
		},
		"should return limited label values if a limit is set": {
			limit: 1,
		},
	}

	for testName, testData := range tests {
		t.Run(testName, func(t *testing.T) {
			for labelName, expectedValues := range expected {
				req := &client.LabelValuesRequest{LabelName: labelName, Limit: testData.limit}
				res, err := i.LabelValues(ctx, req)
				require.NoError(t, err)
				if testData.limit > 0 && len(expectedValues) > int(testData.limit) {
					expectedValues = expectedValues[:testData.limit]