tendermint/internal/mempool/v1/mempool_bench_test.go

package v1

import (
	"context"
	"fmt"
	"math/rand"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
	"github.com/tendermint/tendermint/internal/mempool"
)

func BenchmarkTxMempool_CheckTx(b *testing.B) {
	txmp := setup(b, 10000)
	rng := rand.New(rand.NewSource(time.Now().UnixNano()))

	b.ResetTimer()

	for n := 0; n < b.N; n++ {
		b.StopTimer()
		prefix := make([]byte, 20)
		_, err := rng.Read(prefix)
		require.NoError(b, err)

		priority := int64(rng.Intn(9999-1000) + 1000)
		tx := []byte(fmt.Sprintf("%X=%d", prefix, priority))
		b.StartTimer()

		require.NoError(b, txmp.CheckTx(context.Background(), tx, nil, mempool.TxInfo{}))
	}
}