Function TestIngesterPerLabelsetLimitExceeded

pkg/ingester/ingester_test.go:259–664

(t *testing.T)

}

func TestIngesterPerLabelsetLimitExceeded(t *testing.T) {
	limits := defaultLimitsTestConfig()
	limits.EnableNativeHistograms = true
	userID := "1"
	registry := prometheus.NewRegistry()

	limits.LimitsPerLabelSet = []validation.LimitsPerLabelSet{
		{
			LabelSet: labels.FromMap(map[string]string{
				"label1": "value1",
			}),
			Limits: validation.LimitsPerLabelSetEntry{
				MaxSeries: 3,
			},
		},
		{
			LabelSet: labels.FromMap(map[string]string{
				"label2": "value2",
			}),
			Limits: validation.LimitsPerLabelSetEntry{
				MaxSeries: 2,
			},
		},
	}
	tenantLimits := newMockTenantLimits(map[string]*validation.Limits{userID: &limits})

	b, err := json.Marshal(limits)
	require.NoError(t, err)
	require.NoError(t, limits.UnmarshalJSON(b))

	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))

	ing, err := prepareIngesterWithBlocksStorageAndLimits(t, defaultIngesterTestConfig(t), limits, tenantLimits, blocksDir, registry)
	require.NoError(t, err)
	require.NoError(t, services.StartAndAwaitRunning(context.Background(), ing))
	// Wait until it's ACTIVE
	test.Poll(t, time.Second, ring.ACTIVE, func() any {
		return ing.lifecycler.GetState()
	})

	ctx := user.InjectOrgID(context.Background(), userID)
	samples := []cortexpb.Sample{{Value: 2, TimestampMs: 10}}

	// Create first series within the limits
	for _, set := range limits.LimitsPerLabelSet {
		lbls := []string{labels.MetricName, "metric_name"}
		set.LabelSet.Range(func(l labels.Label) {
			lbls = append(lbls, l.Name, l.Value)
		})
		for i := 0; i < set.Limits.MaxSeries; i++ {
			_, err = ing.Push(ctx, cortexpb.ToWriteRequest(
				[]labels.Labels{labels.FromStrings(append(lbls, "extraLabel", fmt.Sprintf("extraValue%v", i))...)}, samples, nil, nil, cortexpb.API))
			require.NoError(t, err)
		}