mirror of
https://github.com/hanzoai/traces.git
synced 2026-10-30 18:20:41 +00:00
Patrick OyarzunandGitHub
d719849e13
* [Generator] Use prometheus labels instead of LabelValueCombo * Remove unused labels fields
48 lines
1.5 KiB
Go
48 lines
1.5 KiB
Go
package registry
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
|
|
"github.com/prometheus/prometheus/model/labels"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestTestRegistry_counter(t *testing.T) {
|
|
testRegistry := NewTestRegistry()
|
|
|
|
counter := testRegistry.NewCounter("counter")
|
|
|
|
labelValues := buildTestLabels([]string{"foo", "bar"}, []string{"foo-value", "bar-value"})
|
|
counter.Inc(labelValues, 1.0)
|
|
counter.Inc(labelValues, 2.0)
|
|
counter.Inc(labelValues, 1.5)
|
|
|
|
lbls := labels.FromMap(map[string]string{
|
|
"foo": "foo-value",
|
|
"bar": "bar-value",
|
|
})
|
|
assert.Equal(t, 4.5, testRegistry.Query("counter", lbls))
|
|
}
|
|
|
|
func TestTestRegistry_histogram(t *testing.T) {
|
|
testRegistry := NewTestRegistry()
|
|
|
|
histogram := testRegistry.NewHistogram("histogram", []float64{1.0, 2.0}, HistogramModeClassic)
|
|
|
|
labelValues := buildTestLabels([]string{"foo", "bar"}, []string{"foo-value", "bar-value"})
|
|
histogram.ObserveWithExemplar(labelValues, 1.0, "", 1.0)
|
|
histogram.ObserveWithExemplar(labelValues, 2.0, "", 1.0)
|
|
histogram.ObserveWithExemplar(labelValues, 2.5, "", 1.0)
|
|
|
|
lbls := labels.FromMap(map[string]string{
|
|
"foo": "foo-value",
|
|
"bar": "bar-value",
|
|
})
|
|
assert.Equal(t, 1.0, testRegistry.Query("histogram_bucket", withLe(lbls, 1.0)))
|
|
assert.Equal(t, 2.0, testRegistry.Query("histogram_bucket", withLe(lbls, 2.0)))
|
|
assert.Equal(t, 3.0, testRegistry.Query("histogram_bucket", withLe(lbls, math.Inf(1))))
|
|
assert.Equal(t, 3.0, testRegistry.Query("histogram_count", lbls))
|
|
assert.Equal(t, 5.5, testRegistry.Query("histogram_sum", lbls))
|
|
}
|