summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/prometheus/client_golang/prometheus/vec_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/prometheus/client_golang/prometheus/vec_test.go')
-rw-r--r--vendor/github.com/prometheus/client_golang/prometheus/vec_test.go281
1 files changed, 252 insertions, 29 deletions
diff --git a/vendor/github.com/prometheus/client_golang/prometheus/vec_test.go b/vendor/github.com/prometheus/client_golang/prometheus/vec_test.go
index 445a6b39f..bd18a9f4e 100644
--- a/vendor/github.com/prometheus/client_golang/prometheus/vec_test.go
+++ b/vendor/github.com/prometheus/client_golang/prometheus/vec_test.go
@@ -21,8 +21,8 @@ import (
)
func TestDelete(t *testing.T) {
- vec := NewUntypedVec(
- UntypedOpts{
+ vec := NewGaugeVec(
+ GaugeOpts{
Name: "test",
Help: "helpless",
},
@@ -32,8 +32,8 @@ func TestDelete(t *testing.T) {
}
func TestDeleteWithCollisions(t *testing.T) {
- vec := NewUntypedVec(
- UntypedOpts{
+ vec := NewGaugeVec(
+ GaugeOpts{
Name: "test",
Help: "helpless",
},
@@ -44,12 +44,12 @@ func TestDeleteWithCollisions(t *testing.T) {
testDelete(t, vec)
}
-func testDelete(t *testing.T, vec *UntypedVec) {
+func testDelete(t *testing.T, vec *GaugeVec) {
if got, want := vec.Delete(Labels{"l1": "v1", "l2": "v2"}), false; got != want {
t.Errorf("got %v, want %v", got, want)
}
- vec.With(Labels{"l1": "v1", "l2": "v2"}).(Untyped).Set(42)
+ vec.With(Labels{"l1": "v1", "l2": "v2"}).(Gauge).Set(42)
if got, want := vec.Delete(Labels{"l1": "v1", "l2": "v2"}), true; got != want {
t.Errorf("got %v, want %v", got, want)
}
@@ -57,7 +57,7 @@ func testDelete(t *testing.T, vec *UntypedVec) {
t.Errorf("got %v, want %v", got, want)
}
- vec.With(Labels{"l1": "v1", "l2": "v2"}).(Untyped).Set(42)
+ vec.With(Labels{"l1": "v1", "l2": "v2"}).(Gauge).Set(42)
if got, want := vec.Delete(Labels{"l2": "v2", "l1": "v1"}), true; got != want {
t.Errorf("got %v, want %v", got, want)
}
@@ -65,7 +65,7 @@ func testDelete(t *testing.T, vec *UntypedVec) {
t.Errorf("got %v, want %v", got, want)
}
- vec.With(Labels{"l1": "v1", "l2": "v2"}).(Untyped).Set(42)
+ vec.With(Labels{"l1": "v1", "l2": "v2"}).(Gauge).Set(42)
if got, want := vec.Delete(Labels{"l2": "v1", "l1": "v2"}), false; got != want {
t.Errorf("got %v, want %v", got, want)
}
@@ -75,8 +75,8 @@ func testDelete(t *testing.T, vec *UntypedVec) {
}
func TestDeleteLabelValues(t *testing.T) {
- vec := NewUntypedVec(
- UntypedOpts{
+ vec := NewGaugeVec(
+ GaugeOpts{
Name: "test",
Help: "helpless",
},
@@ -86,8 +86,8 @@ func TestDeleteLabelValues(t *testing.T) {
}
func TestDeleteLabelValuesWithCollisions(t *testing.T) {
- vec := NewUntypedVec(
- UntypedOpts{
+ vec := NewGaugeVec(
+ GaugeOpts{
Name: "test",
Help: "helpless",
},
@@ -98,13 +98,13 @@ func TestDeleteLabelValuesWithCollisions(t *testing.T) {
testDeleteLabelValues(t, vec)
}
-func testDeleteLabelValues(t *testing.T, vec *UntypedVec) {
+func testDeleteLabelValues(t *testing.T, vec *GaugeVec) {
if got, want := vec.DeleteLabelValues("v1", "v2"), false; got != want {
t.Errorf("got %v, want %v", got, want)
}
- vec.With(Labels{"l1": "v1", "l2": "v2"}).(Untyped).Set(42)
- vec.With(Labels{"l1": "v1", "l2": "v3"}).(Untyped).Set(42) // Add junk data for collision.
+ vec.With(Labels{"l1": "v1", "l2": "v2"}).(Gauge).Set(42)
+ vec.With(Labels{"l1": "v1", "l2": "v3"}).(Gauge).Set(42) // Add junk data for collision.
if got, want := vec.DeleteLabelValues("v1", "v2"), true; got != want {
t.Errorf("got %v, want %v", got, want)
}
@@ -115,7 +115,7 @@ func testDeleteLabelValues(t *testing.T, vec *UntypedVec) {
t.Errorf("got %v, want %v", got, want)
}
- vec.With(Labels{"l1": "v1", "l2": "v2"}).(Untyped).Set(42)
+ vec.With(Labels{"l1": "v1", "l2": "v2"}).(Gauge).Set(42)
// Delete out of order.
if got, want := vec.DeleteLabelValues("v2", "v1"), false; got != want {
t.Errorf("got %v, want %v", got, want)
@@ -126,8 +126,8 @@ func testDeleteLabelValues(t *testing.T, vec *UntypedVec) {
}
func TestMetricVec(t *testing.T) {
- vec := NewUntypedVec(
- UntypedOpts{
+ vec := NewGaugeVec(
+ GaugeOpts{
Name: "test",
Help: "helpless",
},
@@ -137,8 +137,8 @@ func TestMetricVec(t *testing.T) {
}
func TestMetricVecWithCollisions(t *testing.T) {
- vec := NewUntypedVec(
- UntypedOpts{
+ vec := NewGaugeVec(
+ GaugeOpts{
Name: "test",
Help: "helpless",
},
@@ -149,7 +149,7 @@ func TestMetricVecWithCollisions(t *testing.T) {
testMetricVec(t, vec)
}
-func testMetricVec(t *testing.T, vec *UntypedVec) {
+func testMetricVec(t *testing.T, vec *GaugeVec) {
vec.Reset() // Actually test Reset now!
var pair [2]string
@@ -162,11 +162,11 @@ func testMetricVec(t *testing.T, vec *UntypedVec) {
vec.WithLabelValues(pair[0], pair[1]).Inc()
expected[[2]string{"v1", "v2"}]++
- vec.WithLabelValues("v1", "v2").(Untyped).Inc()
+ vec.WithLabelValues("v1", "v2").(Gauge).Inc()
}
var total int
- for _, metrics := range vec.children {
+ for _, metrics := range vec.metricMap.metrics {
for _, metric := range metrics {
total++
copy(pair[:], metric.values)
@@ -175,7 +175,7 @@ func testMetricVec(t *testing.T, vec *UntypedVec) {
if err := metric.metric.Write(&metricOut); err != nil {
t.Fatal(err)
}
- actual := *metricOut.Untyped.Value
+ actual := *metricOut.Gauge.Value
var actualPair [2]string
for i, label := range metricOut.Label {
@@ -201,7 +201,7 @@ func testMetricVec(t *testing.T, vec *UntypedVec) {
vec.Reset()
- if len(vec.children) > 0 {
+ if len(vec.metricMap.metrics) > 0 {
t.Fatalf("reset failed")
}
}
@@ -239,10 +239,233 @@ func TestCounterVecEndToEndWithCollision(t *testing.T) {
}
}
+func TestCurryVec(t *testing.T) {
+ vec := NewCounterVec(
+ CounterOpts{
+ Name: "test",
+ Help: "helpless",
+ },
+ []string{"one", "two", "three"},
+ )
+ testCurryVec(t, vec)
+}
+
+func TestCurryVecWithCollisions(t *testing.T) {
+ vec := NewCounterVec(
+ CounterOpts{
+ Name: "test",
+ Help: "helpless",
+ },
+ []string{"one", "two", "three"},
+ )
+ vec.hashAdd = func(h uint64, s string) uint64 { return 1 }
+ vec.hashAddByte = func(h uint64, b byte) uint64 { return 1 }
+ testCurryVec(t, vec)
+}
+
+func testCurryVec(t *testing.T, vec *CounterVec) {
+
+ assertMetrics := func(t *testing.T) {
+ n := 0
+ for _, m := range vec.metricMap.metrics {
+ n += len(m)
+ }
+ if n != 2 {
+ t.Error("expected two metrics, got", n)
+ }
+ m := &dto.Metric{}
+ c1, err := vec.GetMetricWithLabelValues("1", "2", "3")
+ if err != nil {
+ t.Fatal("unexpected error getting metric:", err)
+ }
+ c1.Write(m)
+ if want, got := 1., m.GetCounter().GetValue(); want != got {
+ t.Errorf("want %f as counter value, got %f", want, got)
+ }
+ m.Reset()
+ c2, err := vec.GetMetricWithLabelValues("11", "22", "33")
+ if err != nil {
+ t.Fatal("unexpected error getting metric:", err)
+ }
+ c2.Write(m)
+ if want, got := 1., m.GetCounter().GetValue(); want != got {
+ t.Errorf("want %f as counter value, got %f", want, got)
+ }
+ }
+
+ assertNoMetric := func(t *testing.T) {
+ if n := len(vec.metricMap.metrics); n != 0 {
+ t.Error("expected no metrics, got", n)
+ }
+ }
+
+ t.Run("zero labels", func(t *testing.T) {
+ c1 := vec.MustCurryWith(nil)
+ c2 := vec.MustCurryWith(nil)
+ c1.WithLabelValues("1", "2", "3").Inc()
+ c2.With(Labels{"one": "11", "two": "22", "three": "33"}).Inc()
+ assertMetrics(t)
+ if !c1.Delete(Labels{"one": "1", "two": "2", "three": "3"}) {
+ t.Error("deletion failed")
+ }
+ if !c2.DeleteLabelValues("11", "22", "33") {
+ t.Error("deletion failed")
+ }
+ assertNoMetric(t)
+ })
+ t.Run("first label", func(t *testing.T) {
+ c1 := vec.MustCurryWith(Labels{"one": "1"})
+ c2 := vec.MustCurryWith(Labels{"one": "11"})
+ c1.WithLabelValues("2", "3").Inc()
+ c2.With(Labels{"two": "22", "three": "33"}).Inc()
+ assertMetrics(t)
+ if c1.Delete(Labels{"two": "22", "three": "33"}) {
+ t.Error("deletion unexpectedly succeeded")
+ }
+ if c2.DeleteLabelValues("2", "3") {
+ t.Error("deletion unexpectedly succeeded")
+ }
+ if !c1.Delete(Labels{"two": "2", "three": "3"}) {
+ t.Error("deletion failed")
+ }
+ if !c2.DeleteLabelValues("22", "33") {
+ t.Error("deletion failed")
+ }
+ assertNoMetric(t)
+ })
+ t.Run("middle label", func(t *testing.T) {
+ c1 := vec.MustCurryWith(Labels{"two": "2"})
+ c2 := vec.MustCurryWith(Labels{"two": "22"})
+ c1.WithLabelValues("1", "3").Inc()
+ c2.With(Labels{"one": "11", "three": "33"}).Inc()
+ assertMetrics(t)
+ if c1.Delete(Labels{"one": "11", "three": "33"}) {
+ t.Error("deletion unexpectedly succeeded")
+ }
+ if c2.DeleteLabelValues("1", "3") {
+ t.Error("deletion unexpectedly succeeded")
+ }
+ if !c1.Delete(Labels{"one": "1", "three": "3"}) {
+ t.Error("deletion failed")
+ }
+ if !c2.DeleteLabelValues("11", "33") {
+ t.Error("deletion failed")
+ }
+ assertNoMetric(t)
+ })
+ t.Run("last label", func(t *testing.T) {
+ c1 := vec.MustCurryWith(Labels{"three": "3"})
+ c2 := vec.MustCurryWith(Labels{"three": "33"})
+ c1.WithLabelValues("1", "2").Inc()
+ c2.With(Labels{"one": "11", "two": "22"}).Inc()
+ assertMetrics(t)
+ if c1.Delete(Labels{"two": "22", "one": "11"}) {
+ t.Error("deletion unexpectedly succeeded")
+ }
+ if c2.DeleteLabelValues("1", "2") {
+ t.Error("deletion unexpectedly succeeded")
+ }
+ if !c1.Delete(Labels{"two": "2", "one": "1"}) {
+ t.Error("deletion failed")
+ }
+ if !c2.DeleteLabelValues("11", "22") {
+ t.Error("deletion failed")
+ }
+ assertNoMetric(t)
+ })
+ t.Run("two labels", func(t *testing.T) {
+ c1 := vec.MustCurryWith(Labels{"three": "3", "one": "1"})
+ c2 := vec.MustCurryWith(Labels{"three": "33", "one": "11"})
+ c1.WithLabelValues("2").Inc()
+ c2.With(Labels{"two": "22"}).Inc()
+ assertMetrics(t)
+ if c1.Delete(Labels{"two": "22"}) {
+ t.Error("deletion unexpectedly succeeded")
+ }
+ if c2.DeleteLabelValues("2") {
+ t.Error("deletion unexpectedly succeeded")
+ }
+ if !c1.Delete(Labels{"two": "2"}) {
+ t.Error("deletion failed")
+ }
+ if !c2.DeleteLabelValues("22") {
+ t.Error("deletion failed")
+ }
+ assertNoMetric(t)
+ })
+ t.Run("all labels", func(t *testing.T) {
+ c1 := vec.MustCurryWith(Labels{"three": "3", "two": "2", "one": "1"})
+ c2 := vec.MustCurryWith(Labels{"three": "33", "one": "11", "two": "22"})
+ c1.WithLabelValues().Inc()
+ c2.With(nil).Inc()
+ assertMetrics(t)
+ if !c1.Delete(Labels{}) {
+ t.Error("deletion failed")
+ }
+ if !c2.DeleteLabelValues() {
+ t.Error("deletion failed")
+ }
+ assertNoMetric(t)
+ })
+ t.Run("double curry", func(t *testing.T) {
+ c1 := vec.MustCurryWith(Labels{"three": "3"}).MustCurryWith(Labels{"one": "1"})
+ c2 := vec.MustCurryWith(Labels{"three": "33"}).MustCurryWith(Labels{"one": "11"})
+ c1.WithLabelValues("2").Inc()
+ c2.With(Labels{"two": "22"}).Inc()
+ assertMetrics(t)
+ if c1.Delete(Labels{"two": "22"}) {
+ t.Error("deletion unexpectedly succeeded")
+ }
+ if c2.DeleteLabelValues("2") {
+ t.Error("deletion unexpectedly succeeded")
+ }
+ if !c1.Delete(Labels{"two": "2"}) {
+ t.Error("deletion failed")
+ }
+ if !c2.DeleteLabelValues("22") {
+ t.Error("deletion failed")
+ }
+ assertNoMetric(t)
+ })
+ t.Run("use already curried label", func(t *testing.T) {
+ c1 := vec.MustCurryWith(Labels{"three": "3"})
+ if _, err := c1.GetMetricWithLabelValues("1", "2", "3"); err == nil {
+ t.Error("expected error when using already curried label")
+ }
+ if _, err := c1.GetMetricWith(Labels{"one": "1", "two": "2", "three": "3"}); err == nil {
+ t.Error("expected error when using already curried label")
+ }
+ assertNoMetric(t)
+ c1.WithLabelValues("1", "2").Inc()
+ if c1.Delete(Labels{"one": "1", "two": "2", "three": "3"}) {
+ t.Error("deletion unexpectedly succeeded")
+ }
+ if !c1.Delete(Labels{"one": "1", "two": "2"}) {
+ t.Error("deletion failed")
+ }
+ assertNoMetric(t)
+ })
+ t.Run("curry already curried label", func(t *testing.T) {
+ if _, err := vec.MustCurryWith(Labels{"three": "3"}).CurryWith(Labels{"three": "33"}); err == nil {
+ t.Error("currying unexpectedly succeeded")
+ } else if err.Error() != `label name "three" is already curried` {
+ t.Error("currying returned unexpected error:", err)
+ }
+
+ })
+ t.Run("unknown label", func(t *testing.T) {
+ if _, err := vec.CurryWith(Labels{"foo": "bar"}); err == nil {
+ t.Error("currying unexpectedly succeeded")
+ } else if err.Error() != "1 unknown label(s) found during currying" {
+ t.Error("currying returned unexpected error:", err)
+ }
+ })
+}
+
func BenchmarkMetricVecWithLabelValuesBasic(b *testing.B) {
benchmarkMetricVecWithLabelValues(b, map[string][]string{
- "l1": []string{"onevalue"},
- "l2": []string{"twovalue"},
+ "l1": {"onevalue"},
+ "l2": {"twovalue"},
})
}
@@ -290,8 +513,8 @@ func benchmarkMetricVecWithLabelValues(b *testing.B, labels map[string][]string)
}
values := make([]string, len(labels)) // Value cache for permutations.
- vec := NewUntypedVec(
- UntypedOpts{
+ vec := NewGaugeVec(
+ GaugeOpts{
Name: "test",
Help: "helpless",
},
generated by cgit v1.2.3-1-g7c22 (git 2.25.1) at 2024-06-02 23:57:54 +0000