client_golang/prometheus/gauge_test.go

// Copyright (c) 2013, Prometheus Team
// All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package prometheus

import (
	"encoding/json"
	"testing"
)

func testGauge(t 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.Reset(map[string]string{"handler": "/bar"})
					},
				},
			},
			out: output{
				value: `{"type":"gauge","value":[{"labels":{"handler":"/foo"},"value":13}]}`,
			},
		},
		{
			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)
		}

		bytes, err := json.Marshal(gauge)
		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)
	}
}