go-pulse/metrics/gauge_info_test.go

package metrics

import (
	"strconv"
	"testing"
)

func TestGaugeInfoJsonString(t *testing.T) {
	g := NewGaugeInfo()
	g.Update(GaugeInfoValue{
		"chain_id":   "5",
		"anotherKey": "any_string_value",
		"third_key":  "anything",
	},
	)
	want := `{"anotherKey":"any_string_value","chain_id":"5","third_key":"anything"}`
	if have := g.Value().String(); have != want {
		t.Errorf("\nhave: %v\nwant: %v\n", have, want)
	}
}

func TestGaugeInfoSnapshot(t *testing.T) {
	g := NewGaugeInfo()
	g.Update(GaugeInfoValue{"value": "original"})
	snapshot := g.Snapshot() // Snapshot @chainid 5
	g.Update(GaugeInfoValue{"value": "updated"})
	// The 'g' should be updated
	if have, want := g.Value().String(), `{"value":"updated"}`; have != want {
		t.Errorf("\nhave: %v\nwant: %v\n", have, want)
	}
	// Snapshot should be unupdated
	if have, want := snapshot.Value().String(), `{"value":"original"}`; have != want {
		t.Errorf("\nhave: %v\nwant: %v\n", have, want)
	}
}

func TestGetOrRegisterGaugeInfo(t *testing.T) {
	r := NewRegistry()
	NewRegisteredGaugeInfo("foo", r).Update(
		GaugeInfoValue{"chain_id": "5"})
	g := GetOrRegisterGaugeInfo("foo", r)
	if have, want := g.Value().String(), `{"chain_id":"5"}`; have != want {
		t.Errorf("have\n%v\nwant\n%v\n", have, want)
	}
}

func TestFunctionalGaugeInfo(t *testing.T) {
	info := GaugeInfoValue{"chain_id": "0"}
	counter := 1
	// A "functional" gauge invokes the method to obtain the value
	fg := NewFunctionalGaugeInfo(func() GaugeInfoValue {
		info["chain_id"] = strconv.Itoa(counter)
		counter++
		return info
	})
	fg.Value()
	fg.Value()
	if have, want := info["chain_id"], "2"; have != want {
		t.Errorf("have %v want %v", have, want)
	}
}

func TestGetOrRegisterFunctionalGaugeInfo(t *testing.T) {
	r := NewRegistry()
	NewRegisteredFunctionalGaugeInfo("foo", r, func() GaugeInfoValue {
		return GaugeInfoValue{
			"chain_id": "5",
		}
	})
	want := `{"chain_id":"5"}`
	have := GetOrRegisterGaugeInfo("foo", r).Value().String()
	if have != want {
		t.Errorf("have\n%v\nwant\n%v\n", have, want)
	}
}