From 510edb18bb41245e8b72867c4cb3bfed9fbcf46b Mon Sep 17 00:00:00 2001
From: Ethan Frey <ethanfrey@users.noreply.github.com>
Date: Thu, 7 Oct 2021 21:40:10 +0200
Subject: [PATCH] Copy over basic bank benchmark from cosmos-sdk

---
 benchmarks/bench_test.go | 71 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 benchmarks/bench_test.go

diff --git a/benchmarks/bench_test.go b/benchmarks/bench_test.go
new file mode 100644
index 00000000..512a0bde
--- /dev/null
+++ b/benchmarks/bench_test.go
@@ -0,0 +1,71 @@
+package benchmarks
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	abci "github.com/tendermint/tendermint/abci/types"
+	tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
+
+	"github.com/cosmos/cosmos-sdk/crypto/keys/secp256k1"
+	"github.com/cosmos/cosmos-sdk/simapp"
+	simappparams "github.com/cosmos/cosmos-sdk/simapp/params"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
+	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
+	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
+)
+
+var moduleAccAddr = authtypes.NewModuleAddress(stakingtypes.BondedPoolName)
+
+var (
+	priv1 = secp256k1.GenPrivKey()
+	addr1 = sdk.AccAddress(priv1.PubKey().Address())
+	priv2 = secp256k1.GenPrivKey()
+	addr2 = sdk.AccAddress(priv2.PubKey().Address())
+
+	coins    = sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}
+	sendMsg1 = banktypes.NewMsgSend(addr1, addr2, coins)
+)
+
+func BenchmarkOneBankSendTxPerBlock(b *testing.B) {
+	// Add an account at genesis
+	acc := authtypes.BaseAccount{
+		Address: addr1.String(),
+	}
+
+	// construct genesis state
+	genAccs := []authtypes.GenesisAccount{&acc}
+	benchmarkApp := simapp.SetupWithGenesisAccounts(genAccs)
+	ctx := benchmarkApp.BaseApp.NewContext(false, tmproto.Header{})
+
+	// some value conceivably higher than the benchmarks would ever go
+	err := benchmarkApp.BankKeeper.SetBalances(ctx, addr1, sdk.NewCoins(sdk.NewInt64Coin("foocoin", 100000000000)))
+	require.NoError(b, err)
+
+	benchmarkApp.Commit()
+	txGen := simappparams.MakeTestEncodingConfig().TxConfig
+
+	// Precompute all txs
+	txs, err := simapp.GenSequenceOfTxs(txGen, []sdk.Msg{sendMsg1}, []uint64{0}, []uint64{uint64(0)}, b.N, priv1)
+	require.NoError(b, err)
+	b.ResetTimer()
+
+	height := int64(3)
+
+	// Run this with a profiler, so its easy to distinguish what time comes from
+	// Committing, and what time comes from Check/Deliver Tx.
+	for i := 0; i < b.N; i++ {
+		benchmarkApp.BeginBlock(abci.RequestBeginBlock{Header: tmproto.Header{Height: height}})
+		_, _, err := benchmarkApp.Check(txGen.TxEncoder(), txs[i])
+		if err != nil {
+			panic("something is broken in checking transaction")
+		}
+
+		_, _, err = benchmarkApp.Deliver(txGen.TxEncoder(), txs[i])
+		require.NoError(b, err)
+		benchmarkApp.EndBlock(abci.RequestEndBlock{Height: height})
+		benchmarkApp.Commit()
+		height++
+	}
+}