ceremonyclient/go-libp2p/p2p/host/resource-manager/noalloc_test.go
2023-08-20 23:07:43 -05:00

112 lines
2.2 KiB
Go

//go:build nocover
package rcmgr
import (
"math/rand"
"sync"
"testing"
"time"
"github.com/prometheus/client_golang/prometheus"
"github.com/stretchr/testify/require"
)
func randomTraceEvt(rng *rand.Rand) TraceEvt {
// Possibly non-sensical
typs := []TraceEvtTyp{
TraceStartEvt,
TraceCreateScopeEvt,
TraceDestroyScopeEvt,
TraceReserveMemoryEvt,
TraceBlockReserveMemoryEvt,
TraceReleaseMemoryEvt,
TraceAddStreamEvt,
TraceBlockAddStreamEvt,
TraceRemoveStreamEvt,
TraceAddConnEvt,
TraceBlockAddConnEvt,
TraceRemoveConnEvt,
}
names := []string{
"conn-1",
"stream-2",
"peer:abc",
"system",
"transient",
"peer:12D3Koo",
"protocol:/libp2p/autonat/1.0.0",
"protocol:/libp2p/autonat/1.0.0.peer:12D3Koo",
"service:libp2p.autonat",
"service:libp2p.autonat.peer:12D3Koo",
}
return TraceEvt{
Type: typs[rng.Intn(len(typs))],
Name: names[rng.Intn(len(names))],
DeltaOut: rng.Intn(5),
DeltaIn: rng.Intn(5),
Delta: int64(rng.Intn(5)),
Memory: int64(rng.Intn(10000)),
StreamsIn: rng.Intn(100),
StreamsOut: rng.Intn(100),
ConnsIn: rng.Intn(100),
ConnsOut: rng.Intn(100),
FD: rng.Intn(100),
Time: time.Now().Format(time.RFC3339Nano),
}
}
var regOnce sync.Once
func BenchmarkMetricsRecording(b *testing.B) {
b.ReportAllocs()
registerOnce.Do(func() {
MustRegisterWith(prometheus.DefaultRegisterer)
})
evtCount := 10000
evts := make([]TraceEvt, evtCount)
rng := rand.New(rand.NewSource(int64(b.N)))
for i := 0; i < evtCount; i++ {
evts[i] = randomTraceEvt(rng)
}
str, err := NewStatsTraceReporter()
if err != nil {
b.Fatal(err)
}
b.ResetTimer()
for i := 0; i < b.N; i++ {
str.ConsumeEvent(evts[i%len(evts)])
}
}
func TestNoAllocsNoCover(t *testing.T) {
str, err := NewStatsTraceReporter()
require.NoError(t, err)
evtCount := 10_000
evts := make([]TraceEvt, 0, evtCount)
rng := rand.New(rand.NewSource(1))
for i := 0; i < evtCount; i++ {
evts = append(evts, randomTraceEvt(rng))
}
tagSlice := make([]string, 0, 10)
allocs := testing.AllocsPerRun(100, func() {
for i := 0; i < evtCount; i++ {
str.consumeEventWithLabelSlice(evts[i], &tagSlice)
}
})
if allocs > 10 {
t.Fatalf("expected less than 10 heap bytes, got %f", allocs)
}
}