render_allocation_test.go RawHistoryFolder

1package daemon
2
3import (
4	"runtime"
5	"testing"
6
7	"github.com/stretchr/testify/require"
8)
9
10// Measure bytes with the same bounded warmup/sampling strategy as
11// testing.AllocsPerRun. These sequential allocation tests need a byte budget,
12// not testing.Benchmark's default one-second timing calibration per assertion.
13func renderAllocatedBytesPerRun(run func()) int64 {
14	defer runtime.GOMAXPROCS(runtime.GOMAXPROCS(1))
15	run()
16	var before, after runtime.MemStats
17	runtime.ReadMemStats(&before)
18	const samples = 20
19	for range samples {
20		run()
21	}
22	runtime.ReadMemStats(&after)
23	return int64((after.TotalAlloc - before.TotalAlloc) / samples)
24}
25
26// Setup and warmup remain outside the measured operation; existing byte
27// ceilings are unchanged. Like AllocsPerRun, do not call from parallel tests.
28func assertRenderByteBudget(t *testing.T, run func(), maximum int64) {
29	t.Helper()
30	require.LessOrEqual(t, renderAllocatedBytesPerRun(run), maximum, "rendering must preserve the compact byte-allocation benefit")
31}