mirror of
https://github.com/hanzozt/metrics.git
synced 2026-10-30 14:46:36 +00:00
zeekay
c9ba84657d
Global rebrand of all OpenZiti references to Hanzo ZT: - Go module paths: github.com/openziti/* → github.com/hanzozt/* - Docker images: openziti/* → hanzozt/* - Java packages: org.openziti.* → org.hanzozt.* - Brand: OpenZiti → Hanzo ZT - URLs: openziti.io → hanzozt.dev
95 lines
1.9 KiB
Go
95 lines
1.9 KiB
Go
package metrics
|
|
|
|
import (
|
|
"github.com/hanzozt/metrics/metrics_pb"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type testData struct {
|
|
closeNotify chan struct{}
|
|
registry *usageRegistryImpl
|
|
events []*metrics_pb.MetricsMessage
|
|
}
|
|
|
|
func setUpTest(t *testing.T) *testData {
|
|
closeNotify := make(chan struct{})
|
|
|
|
config := DefaultUsageRegistryConfig(t.Name(), closeNotify)
|
|
|
|
td := &testData{
|
|
closeNotify: closeNotify,
|
|
registry: NewUsageRegistry(config).(*usageRegistryImpl),
|
|
}
|
|
td.registry.StartReporting(nil, time.Hour, 10)
|
|
return td
|
|
}
|
|
|
|
func (t *testData) Shutdown() {
|
|
close(t.closeNotify)
|
|
}
|
|
|
|
func (t *testData) AcceptMetrics(e *metrics_pb.MetricsMessage) {
|
|
t.events = append(t.events, e)
|
|
}
|
|
|
|
func TestEmpty(t *testing.T) {
|
|
td := setUpTest(t)
|
|
defer td.Shutdown()
|
|
|
|
td.registry.FlushToHandler(td)
|
|
assert.Len(t, td.events, 0)
|
|
}
|
|
|
|
func Test_Histogram(t *testing.T) {
|
|
td := setUpTest(t)
|
|
defer td.Shutdown()
|
|
|
|
hist := td.registry.Histogram("test.hist")
|
|
hist.Update(10)
|
|
|
|
td.registry.FlushToHandler(td)
|
|
assert.Len(t, td.events, 1)
|
|
|
|
ev := td.events[0]
|
|
assert.Nil(t, ev.FloatValues)
|
|
assert.Nil(t, ev.Meters)
|
|
assert.Nil(t, ev.IntValues)
|
|
|
|
assert.NotNil(t, ev.Histograms)
|
|
|
|
hm := ev.Histograms["test.hist"]
|
|
assert.NotNil(t, hm)
|
|
assert.Equal(t, int64(10), hm.Min)
|
|
assert.Equal(t, int64(10), hm.Max)
|
|
}
|
|
|
|
func Test_Timer(t *testing.T) {
|
|
td := setUpTest(t)
|
|
defer td.Shutdown()
|
|
|
|
timer := td.registry.Timer("test.timer")
|
|
|
|
timer.Update(3 * time.Second)
|
|
|
|
timer.Time(func() {
|
|
<-time.After(time.Second)
|
|
})
|
|
|
|
td.registry.FlushToHandler(td)
|
|
assert.Len(t, td.events, 1)
|
|
|
|
ev := td.events[0]
|
|
assert.Nil(t, ev.FloatValues)
|
|
assert.Nil(t, ev.Meters)
|
|
assert.Nil(t, ev.IntValues)
|
|
|
|
hm := ev.Timers["test.timer"]
|
|
assert.NotNil(t, hm)
|
|
assert.Equal(t, int64(2), hm.Count)
|
|
|
|
assert.Equal(t, 3*time.Second, time.Duration(hm.Max))
|
|
assert.InDelta(t, time.Second, time.Duration(hm.Min), float64(2*time.Millisecond))
|
|
}
|