prysm/beacon-chain/sync/initial-sync/fsm_benchmark_test.go

package initialsync

import (
	"testing"

	"github.com/OffchainLabs/prysm/v7/testing/require"
)

func BenchmarkStateMachine_trigger(b *testing.B) {
	sm := newStateMachineManager()

	handlerFn := func(m *stateMachine, in any) (id stateID, err error) {
		response, ok := in.(*fetchRequestParams)
		if !ok {
			return 0, errInputNotFetchRequestParams
		}
		_ = response.count
		return stateScheduled, nil
	}

	sm.addEventHandler(eventTick, stateNew, handlerFn)
	sm.addEventHandler(eventTick, stateScheduled, handlerFn)
	sm.addEventHandler(eventTick, stateDataParsed, handlerFn)
	sm.addEventHandler(eventTick, stateSkipped, handlerFn)
	sm.addEventHandler(eventTick, stateSent, handlerFn)
	sm.addStateMachine(64)

	b.ReportAllocs()

	for b.Loop() {
		data := &fetchRequestParams{
			start: 23,
			count: 32,
		}
		err := sm.machines[64].trigger(eventTick, data)
		require.NoError(b, err)
	}
}