You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

65 lines
1.4 KiB

package prometheus
import (
"bytes"
"fmt"
"testing"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/common/expfmt"
)
func gatherLatest(reg *prometheus.Registry) (result string, err error) {
mfs, err := reg.Gather()
if err != nil {
return "", err
}
buf := &bytes.Buffer{}
enc := expfmt.NewEncoder(buf, expfmt.FmtText)
for _, mf := range mfs {
if err = enc.Encode(mf); err != nil {
return "", err
}
}
return buf.String(), nil
}
func TestCounter(t *testing.T) {
expect := `# HELP test_request_test_metric test
# TYPE test_request_test_metric counter
test_request_test_metric{code="test",kind="test",operation="test",reason="test"} %d
`
counterVec := prometheus.NewCounterVec(prometheus.CounterOpts{
Namespace: "test",
Name: "test_metric",
Subsystem: "request",
Help: "test",
}, []string{"kind", "operation", "code", "reason"})
counter := NewCounter(counterVec)
counter.With("test", "test", "test", "test").Inc()
reg := prometheus.NewRegistry()
reg.MustRegister(counterVec)
result, err := gatherLatest(reg)
if err != nil {
t.Fatal(err)
}
if result != fmt.Sprintf(expect, 1) {
t.Fatal("metrics error")
}
counter.With("test", "test", "test", "test").Add(10)
result, err = gatherLatest(reg)
if err != nil {
t.Fatal(err)
}
if result != fmt.Sprintf(expect, 11) {
t.Fatal("metrics error")
}
}