ceremonyclient/go-libp2p/p2p/host/eventbus/basic_metrics_test.go
2023-08-20 23:07:43 -05:00

67 lines
1.9 KiB
Go

//go:build nocover
package eventbus
import (
"math/rand"
"reflect"
"testing"
"github.com/libp2p/go-libp2p/core/event"
)
func BenchmarkEventEmitted(b *testing.B) {
b.ReportAllocs()
types := []reflect.Type{
reflect.TypeOf(new(event.EvtLocalAddressesUpdated)),
reflect.TypeOf(new(event.EvtNATDeviceTypeChanged)),
reflect.TypeOf(new(event.EvtLocalProtocolsUpdated)),
}
mt := NewMetricsTracer()
for i := 0; i < b.N; i++ {
mt.EventEmitted(types[i%len(types)])
}
}
func BenchmarkSubscriberQueueLength(b *testing.B) {
b.ReportAllocs()
names := []string{"s1", "s2", "s3", "s4"}
mt := NewMetricsTracer()
for i := 0; i < b.N; i++ {
mt.SubscriberQueueLength(names[i%len(names)], i)
}
}
var eventTypes = []reflect.Type{
reflect.TypeOf(new(event.EvtLocalAddressesUpdated)),
reflect.TypeOf(new(event.EvtNATDeviceTypeChanged)),
reflect.TypeOf(new(event.EvtLocalProtocolsUpdated)),
reflect.TypeOf(new(event.EvtPeerIdentificationCompleted)),
}
var names = []string{
"one",
"two",
"three",
"four",
"five",
}
func TestMetricsNoAllocNoCover(t *testing.T) {
mt := NewMetricsTracer()
tests := map[string]func(){
"EventEmitted": func() { mt.EventEmitted(eventTypes[rand.Intn(len(eventTypes))]) },
"AddSubscriber": func() { mt.AddSubscriber(eventTypes[rand.Intn(len(eventTypes))]) },
"RemoveSubscriber": func() { mt.RemoveSubscriber(eventTypes[rand.Intn(len(eventTypes))]) },
"SubscriberQueueLength": func() { mt.SubscriberQueueLength(names[rand.Intn(len(names))], rand.Intn(100)) },
"SubscriberQueueFull": func() { mt.SubscriberQueueFull(names[rand.Intn(len(names))], rand.Intn(2) == 1) },
"SubscriberEventQueued": func() { mt.SubscriberEventQueued(names[rand.Intn(len(names))]) },
}
for method, f := range tests {
allocs := testing.AllocsPerRun(1000, f)
if allocs > 0 {
t.Fatalf("Alloc Test: %s, got: %0.2f, expected: 0 allocs", method, allocs)
}
}
}