Function TestIngester_UpdateLabelSetMetrics

pkg/ingester/ingester_test.go:7266–7345

(t *testing.T)

}

func TestIngester_UpdateLabelSetMetrics(t *testing.T) {
	cfg := defaultIngesterTestConfig(t)
	cfg.BlocksStorageConfig.TSDB.BlockRanges = []time.Duration{2 * time.Hour}
	reg := prometheus.NewRegistry()
	limits := defaultLimitsTestConfig()
	limits.EnableNativeHistograms = true
	userID := "1"
	ctx := user.InjectOrgID(context.Background(), userID)

	limits.LimitsPerLabelSet = []validation.LimitsPerLabelSet{
		{
			LabelSet: labels.FromMap(map[string]string{
				"foo": "bar",
			}),
			Limits: validation.LimitsPerLabelSetEntry{
				MaxSeries: 1,
			},
		},
		{
			LabelSet: labels.EmptyLabels(),
		},
	}
	tenantLimits := newMockTenantLimits(map[string]*validation.Limits{userID: &limits})

	dir := t.TempDir()
	chunksDir := filepath.Join(dir, "chunks")
	blocksDir := filepath.Join(dir, "blocks")
	require.NoError(t, os.Mkdir(chunksDir, os.ModePerm))
	require.NoError(t, os.Mkdir(blocksDir, os.ModePerm))

	i, err := prepareIngesterWithBlocksStorageAndLimits(t, cfg, limits, tenantLimits, blocksDir, reg)
	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, time.Second, ring.ACTIVE, func() any {
		return i.lifecycler.GetState()
	})
	// Add user ID.
	wreq := &cortexpb.WriteRequest{
		Timeseries: []cortexpb.PreallocTimeseries{
			{TimeSeries: &cortexpb.TimeSeries{
				Labels:  cortexpb.FromLabelsToLabelAdapters(labels.FromMap(map[string]string{model.MetricNameLabel: "test", "foo": "bar"})),
				Samples: []cortexpb.Sample{{Value: 0, TimestampMs: 1}},
			}},
		},
	}
	_, err = i.Push(ctx, wreq)
	require.NoError(t, err)

	// Push one more series will trigger throttle by label set.
	wreq = &cortexpb.WriteRequest{
		Timeseries: []cortexpb.PreallocTimeseries{
			{TimeSeries: &cortexpb.TimeSeries{
				Labels:  cortexpb.FromLabelsToLabelAdapters(labels.FromMap(map[string]string{model.MetricNameLabel: "test2", "foo": "bar"})),
				Samples: []cortexpb.Sample{{Value: 0, TimestampMs: 0}},
			}},