Function Test_Ingester_LabelNames

pkg/ingester/ingester_test.go:3039–3093

(t *testing.T)

}

func Test_Ingester_LabelNames(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 := []string{"__name__", "route", "status"}

	// 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    int
		expected []string
	}{
		"should return all label names if no limit is set": {
			expected: expected,
		},
		"should return limited label names if a limit is set": {
			limit:    2,
			expected: expected[:2],
		},
	}

	for testName, testData := range tests {
		t.Run(testName, func(t *testing.T) {
			// Get label names
			res, err := i.LabelNames(ctx, &client.LabelNamesRequest{Limit: int64(testData.limit)})
			require.NoError(t, err)
			assert.ElementsMatch(t, testData.expected, res.LabelNames)
		})
	}
}

func Test_Ingester_LabelValues(t *testing.T) {
	series := []struct {