Function TestManagedRegistry_limited

modules/generator/registry/registry_test.go:243–281

(t *testing.T)

}

func TestManagedRegistry_limited(t *testing.T) {
	appender := &capturingAppender{}

	overrides := &mockOverrides{}
	atLimit := false
	overflowLabels := labels.FromStrings("metric_overflow", "true")
	overflowHash := overflowLabels.Hash()
	limiter := &mockLimiter{
		onAddFunc: func(hash uint64, _ uint32, lbls labels.Labels) (labels.Labels, uint64) {
			if !atLimit {
				return lbls, hash
			}
			return overflowLabels, overflowHash
		},
	}
	registry := New(&Config{}, overrides, "test", appender, log.NewNopLogger(), limiter)
	defer registry.Close()

	counter1 := registry.NewCounter("metric_1")
	counter2 := registry.NewCounter("metric_2")

	counter1.Inc(buildTestLabels([]string{"label"}, []string{"value-1"}), 1.0)
	atLimit = true
	// these series should be mapped to overflow
	counter1.Inc(buildTestLabels([]string{"label"}, []string{"value-2"}), 1.0)
	counter2.Inc(labels.New(), 1.0)

	// 1 accepted series + 2 overflow series (one per metric) = 3 total
	assert.Equal(t, uint32(3), registry.activeSeries())
	expectedSamples := []sample{
		newSample(map[string]string{"__name__": "metric_1", "label": "value-1", "__metrics_gen_instance": mustGetHostname()}, 0, 0),
		newSample(map[string]string{"__name__": "metric_1", "label": "value-1", "__metrics_gen_instance": mustGetHostname()}, 0, 1),
		newSample(map[string]string{"__name__": "metric_1", "metric_overflow": "true", "__metrics_gen_instance": mustGetHostname()}, 0, 0),
		newSample(map[string]string{"__name__": "metric_1", "metric_overflow": "true", "__metrics_gen_instance": mustGetHostname()}, 0, 1),
		newSample(map[string]string{"__name__": "metric_2", "metric_overflow": "true", "__metrics_gen_instance": mustGetHostname()}, 0, 0),
		newSample(map[string]string{"__name__": "metric_2", "metric_overflow": "true", "__metrics_gen_instance": mustGetHostname()}, 0, 1),
	}
	collectRegistryMetricsAndAssert(t, registry, appender, expectedSamples)
}

func TestManagedRegistry_maxEntities(t *testing.T) {
	appender := &capturingAppender{}