138 lines
2.5 KiB
Go
138 lines
2.5 KiB
Go
// Copyright (c) 2012, Matt T. Proud
|
|
// All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package metrics
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/prometheus/client_golang/utility/test"
|
|
"testing"
|
|
)
|
|
|
|
func testGauge(t test.Tester) {
|
|
type input struct {
|
|
steps []func(g Gauge)
|
|
}
|
|
type output struct {
|
|
value string
|
|
}
|
|
|
|
var scenarios = []struct {
|
|
in input
|
|
out output
|
|
}{
|
|
{
|
|
in: input{
|
|
steps: []func(g Gauge){},
|
|
},
|
|
out: output{
|
|
value: "{\"type\":\"gauge\",\"value\":[]}",
|
|
},
|
|
},
|
|
{
|
|
in: input{
|
|
steps: []func(g Gauge){
|
|
func(g Gauge) {
|
|
g.Set(nil, 1)
|
|
},
|
|
},
|
|
},
|
|
out: output{
|
|
value: "{\"type\":\"gauge\",\"value\":[{\"labels\":{},\"value\":1}]}",
|
|
},
|
|
},
|
|
{
|
|
in: input{
|
|
steps: []func(g Gauge){
|
|
func(g Gauge) {
|
|
g.Set(map[string]string{}, 2)
|
|
},
|
|
},
|
|
},
|
|
out: output{
|
|
value: "{\"type\":\"gauge\",\"value\":[{\"labels\":{},\"value\":2}]}",
|
|
},
|
|
},
|
|
{
|
|
in: input{
|
|
steps: []func(g Gauge){
|
|
func(g Gauge) {
|
|
g.Set(map[string]string{}, 3)
|
|
},
|
|
func(g Gauge) {
|
|
g.Set(map[string]string{}, 5)
|
|
},
|
|
},
|
|
},
|
|
out: output{
|
|
value: "{\"type\":\"gauge\",\"value\":[{\"labels\":{},\"value\":5}]}",
|
|
},
|
|
},
|
|
{
|
|
in: input{
|
|
steps: []func(g Gauge){
|
|
func(g Gauge) {
|
|
g.Set(map[string]string{"handler": "/foo"}, 13)
|
|
},
|
|
func(g Gauge) {
|
|
g.Set(map[string]string{"handler": "/bar"}, 17)
|
|
},
|
|
func(g Gauge) {
|
|
g.ResetAll()
|
|
},
|
|
},
|
|
},
|
|
out: output{
|
|
value: "{\"type\":\"gauge\",\"value\":[]}",
|
|
},
|
|
},
|
|
{
|
|
in: input{
|
|
steps: []func(g Gauge){
|
|
func(g Gauge) {
|
|
g.Set(map[string]string{"handler": "/foo"}, 19)
|
|
},
|
|
},
|
|
},
|
|
out: output{
|
|
value: "{\"type\":\"gauge\",\"value\":[{\"labels\":{\"handler\":\"/foo\"},\"value\":19}]}",
|
|
},
|
|
},
|
|
}
|
|
|
|
for i, scenario := range scenarios {
|
|
gauge := NewGauge()
|
|
|
|
for _, step := range scenario.in.steps {
|
|
step(gauge)
|
|
}
|
|
|
|
marshallable := gauge.AsMarshallable()
|
|
|
|
bytes, err := json.Marshal(marshallable)
|
|
if err != nil {
|
|
t.Errorf("%d. could not marshal into JSON %s", i, err)
|
|
continue
|
|
}
|
|
|
|
asString := string(bytes)
|
|
|
|
if scenario.out.value != asString {
|
|
t.Errorf("%d. expected %q, got %q", i, scenario.out.value, asString)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGauge(t *testing.T) {
|
|
testGauge(t)
|
|
}
|
|
|
|
func BenchmarkGauge(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
testGauge(b)
|
|
}
|
|
}
|