strata-go/metrics_test.go at main · ctxsh/strata-go · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package strata

import (
	"fmt"
	"testing"

	"github.com/prometheus/client_golang/prometheus"
	"github.com/stretchr/testify/assert"
)

var labels = map[string]string{
	"region": "us-east-1",
}

func TestMetricsCounter(t *testing.T) {
	name := "test_total"
	m := testMetrics().WithLabels("region")

	m.CounterInc(name, "us-east-1")
	vec, err := getCounter(m, prefixedName(m.prefix, name, m.separator))
	assert.NoError(t, err)
	CollectAndCompare(t, vec, "strata_example_test_total", "counter", labels, 1.0)
	m.CounterAdd(name, 5.0, "us-east-1")
	CollectAndCompare(t, vec, "strata_example_test_total", "counter", labels, 6.0)

	m1 := m.WithPrefix("next")
	m1.CounterInc(name)
	vec, err = getCounter(m1, prefixedName(m1.prefix, name, m1.separator))
	assert.NoError(t, err)
	CollectAndCompare(t, vec, "strata_example_next_test_total", "counter", nil, 1.0)
	m1.CounterAdd(name, 5.0)
	CollectAndCompare(t, vec, "strata_example_next_test_total", "counter", nil, 6.0)
}

func TestMetricsGauge(t *testing.T) {
	name := "test_g"
	m := testMetrics().WithLabels("region")

	m.GaugeInc(name, "us-east-1")
	vec, err := getGauge(m, prefixedName(m.prefix, name, m.separator))
	assert.NoError(t, err)
	CollectAndCompare(t, vec, "strata_example_test_g", "gauge", labels, 1.0)
	m.GaugeAdd(name, 5.0, "us-east-1")
	CollectAndCompare(t, vec, "strata_example_test_g", "gauge", labels, 6.0)
	m.GaugeSet(name, 10.0, "us-east-1")
	CollectAndCompare(t, vec, "strata_example_test_g", "gauge", labels, 10.0)
	m.GaugeDec(name, "us-east-1")
	CollectAndCompare(t, vec, "strata_example_test_g", "gauge", labels, 9.0)
	m.GaugeSub(name, 9.0, "us-east-1")
	CollectAndCompare(t, vec, "strata_example_test_g", "gauge", labels, 0.0)

	m1 := m.WithPrefix("next")
	m1.GaugeInc(name)
	vec, err = getGauge(m, prefixedName(m1.prefix, name, m1.separator))
	assert.NoError(t, err)
	CollectAndCompare(t, vec, "strata_example_next_test_g", "gauge", nil, 1.0)
	m1.GaugeAdd(name, 5.0)
	CollectAndCompare(t, vec, "strata_example_next_test_g", "gauge", nil, 6.0)
	m1.GaugeSet(name, 10.0)
	CollectAndCompare(t, vec, "strata_example_next_test_g", "gauge", nil, 10.0)
	m1.GaugeDec(name)
	CollectAndCompare(t, vec, "strata_example_next_test_g", "gauge", nil, 9.0)
	m1.GaugeSub(name, 9.0)
	CollectAndCompare(t, vec, "strata_example_next_test_g", "gauge", nil, 0.0)
}

func getCounter(metrics *Metrics, n string) (MetricVec, error) {
	if v, ok := metrics.store.counters[n]; ok {
		return v, nil
	}
	return nil, fmt.Errorf("missing counter")
}

func getGauge(metrics *Metrics, n string) (MetricVec, error) {
	if v, ok := metrics.store.gauges[n]; ok {
		return v, nil
	}
	return nil, fmt.Errorf("missing gauge")
}

func testMetrics() *Metrics {
	registry := prometheus.NewPedanticRegistry()
	metrics := New(MetricsOpts{
		Separator:    '_',
		Registry:     registry,
		PanicOnError: true,
	}).WithPrefix("strata", "example")

	return metrics
}