Function Test_counter

modules/generator/registry/counter_test.go:15–56

(t *testing.T)

)

func Test_counter(t *testing.T) {
	var seriesAdded int
	lifecycler := &mockLimiter{
		onAddFunc: func(hash uint64, _ uint32, lbls labels.Labels) (labels.Labels, uint64) {
			seriesAdded++
			return lbls, hash
		},
	}

	c := newCounter("my_counter", lifecycler, map[string]string{}, 15*time.Minute)

	c.Inc(buildTestLabels([]string{"label"}, []string{"value-1"}), 1.0)
	c.Inc(buildTestLabels([]string{"label"}, []string{"value-2"}), 2.0)

	assert.Equal(t, 2, seriesAdded)

	collectionTimeMs := time.Now().UnixMilli()
	endOfLastMinuteMs := getEndOfLastMinuteMs(collectionTimeMs)
	expectedSamples := []sample{
		newSample(map[string]string{"__name__": "my_counter", "label": "value-1"}, endOfLastMinuteMs, 0),
		newSample(map[string]string{"__name__": "my_counter", "label": "value-1"}, collectionTimeMs, 1),
		newSample(map[string]string{"__name__": "my_counter", "label": "value-2"}, endOfLastMinuteMs, 0),
		newSample(map[string]string{"__name__": "my_counter", "label": "value-2"}, collectionTimeMs, 2),
	}
	collectMetricAndAssert(t, c, collectionTimeMs, 2, expectedSamples, nil)

	c.Inc(buildTestLabels([]string{"label"}, []string{"value-2"}), 2.0)
	c.Inc(buildTestLabels([]string{"label"}, []string{"value-3"}), 3.0)

	assert.Equal(t, 3, seriesAdded)

	collectionTimeMs = time.Now().UnixMilli()
	endOfLastMinuteMs = getEndOfLastMinuteMs(collectionTimeMs)
	expectedSamples = []sample{
		newSample(map[string]string{"__name__": "my_counter", "label": "value-1"}, collectionTimeMs, 1),
		newSample(map[string]string{"__name__": "my_counter", "label": "value-2"}, collectionTimeMs, 4),
		newSample(map[string]string{"__name__": "my_counter", "label": "value-3"}, endOfLastMinuteMs, 0),
		newSample(map[string]string{"__name__": "my_counter", "label": "value-3"}, collectionTimeMs, 3),
	}

	collectMetricAndAssert(t, c, collectionTimeMs, 3, expectedSamples, nil)
}

func TestCounterDifferentLabels(t *testing.T) {
	var seriesAdded int