Function TestIngesterUserLimitExceeded

pkg/ingester/ingester_test.go:754–871

(t *testing.T)

}

func TestIngesterUserLimitExceeded(t *testing.T) {
	limits := defaultLimitsTestConfig()
	limits.EnableNativeHistograms = true
	limits.MaxLocalSeriesPerUser = 1
	limits.MaxLocalNativeHistogramSeriesPerUser = 1
	limits.MaxLocalMetricsWithMetadataPerUser = 1

	userID := "1"
	// Series
	labels1 := labels.FromStrings(labels.MetricName, "testmetric", "foo", "bar")
	sample1 := cortexpb.Sample{
		TimestampMs: 0,
		Value:       1,
	}
	sample2 := cortexpb.Sample{
		TimestampMs: 1,
		Value:       2,
	}
	labels3 := labels.FromStrings(labels.MetricName, "testmetric", "foo", "biz")
	sample3 := cortexpb.Sample{
		TimestampMs: 1,
		Value:       3,
	}
	// Metadata
	metadata1 := &cortexpb.MetricMetadata{MetricFamilyName: "testmetric", Help: "a help for testmetric", Type: cortexpb.COUNTER}
	metadata2 := &cortexpb.MetricMetadata{MetricFamilyName: "testmetric2", Help: "a help for testmetric2", Type: cortexpb.COUNTER}

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

	blocksIngesterGenerator := func(reg prometheus.Registerer) *Ingester {
		ing, err := prepareIngesterWithBlocksStorageAndLimits(t, defaultIngesterTestConfig(t), limits, nil, blocksDir, reg)
		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()
		})

		return ing
	}

	tests := []string{"blocks"}
	for i, ingGenerator := range []func(reg prometheus.Registerer) *Ingester{blocksIngesterGenerator} {
		t.Run(tests[i], func(t *testing.T) {
			reg := prometheus.NewRegistry()
			ing := ingGenerator(reg)

			// Append only one series and one metadata first, expect no error.
			ctx := user.InjectOrgID(context.Background(), userID)
			_, err := ing.Push(ctx, cortexpb.ToWriteRequest([]labels.Labels{labels1}, []cortexpb.Sample{sample1}, []*cortexpb.MetricMetadata{metadata1}, nil, cortexpb.API))
			require.NoError(t, err)

			testLimits := func(reg prometheus.Gatherer) {