| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632 |
- // Copyright 2015 The go-ethereum Authors
- // This file is part of the go-ethereum library.
- //
- // The go-ethereum library is free software: you can redistribute it and/or modify
- // it under the terms of the GNU Lesser General Public License as published by
- // the Free Software Foundation, either version 3 of the License, or
- // (at your option) any later version.
- //
- // The go-ethereum library is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Lesser General Public License for more details.
- //
- // You should have received a copy of the GNU Lesser General Public License
- // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
- package core
- import (
- "crypto/ecdsa"
- "math/big"
- "testing"
- "github.com/ethereum/go-ethereum/common"
- "github.com/ethereum/go-ethereum/core/state"
- "github.com/ethereum/go-ethereum/core/types"
- "github.com/ethereum/go-ethereum/crypto"
- "github.com/ethereum/go-ethereum/ethdb"
- "github.com/ethereum/go-ethereum/event"
- )
- func transaction(nonce uint64, gaslimit *big.Int, key *ecdsa.PrivateKey) *types.Transaction {
- tx, _ := types.NewTransaction(nonce, common.Address{}, big.NewInt(100), gaslimit, big.NewInt(1), nil).SignECDSA(key)
- return tx
- }
- func setupTxPool() (*TxPool, *ecdsa.PrivateKey) {
- db, _ := ethdb.NewMemDatabase()
- statedb, _ := state.New(common.Hash{}, db)
- var m event.TypeMux
- key, _ := crypto.GenerateKey()
- newPool := NewTxPool(testChainConfig(), &m, func() (*state.StateDB, error) { return statedb, nil }, func() *big.Int { return big.NewInt(1000000) })
- newPool.resetState()
- return newPool, key
- }
- func TestInvalidTransactions(t *testing.T) {
- pool, key := setupTxPool()
- tx := transaction(0, big.NewInt(100), key)
- if err := pool.Add(tx); err != ErrNonExistentAccount {
- t.Error("expected", ErrNonExistentAccount)
- }
- from, _ := tx.From()
- currentState, _ := pool.currentState()
- currentState.AddBalance(from, big.NewInt(1))
- if err := pool.Add(tx); err != ErrInsufficientFunds {
- t.Error("expected", ErrInsufficientFunds)
- }
- balance := new(big.Int).Add(tx.Value(), new(big.Int).Mul(tx.Gas(), tx.GasPrice()))
- currentState.AddBalance(from, balance)
- if err := pool.Add(tx); err != ErrIntrinsicGas {
- t.Error("expected", ErrIntrinsicGas, "got", err)
- }
- currentState.SetNonce(from, 1)
- currentState.AddBalance(from, big.NewInt(0xffffffffffffff))
- tx = transaction(0, big.NewInt(100000), key)
- if err := pool.Add(tx); err != ErrNonce {
- t.Error("expected", ErrNonce)
- }
- tx = transaction(1, big.NewInt(100000), key)
- pool.minGasPrice = big.NewInt(1000)
- if err := pool.Add(tx); err != ErrCheap {
- t.Error("expected", ErrCheap, "got", err)
- }
- pool.SetLocal(tx)
- if err := pool.Add(tx); err != nil {
- t.Error("expected", nil, "got", err)
- }
- }
- func TestTransactionQueue(t *testing.T) {
- pool, key := setupTxPool()
- tx := transaction(0, big.NewInt(100), key)
- from, _ := tx.From()
- currentState, _ := pool.currentState()
- currentState.AddBalance(from, big.NewInt(1000))
- pool.enqueueTx(tx.Hash(), tx)
- pool.promoteExecutables()
- if len(pool.pending) != 1 {
- t.Error("expected valid txs to be 1 is", len(pool.pending))
- }
- tx = transaction(1, big.NewInt(100), key)
- from, _ = tx.From()
- currentState.SetNonce(from, 2)
- pool.enqueueTx(tx.Hash(), tx)
- pool.promoteExecutables()
- if _, ok := pool.pending[from].items[tx.Nonce()]; ok {
- t.Error("expected transaction to be in tx pool")
- }
- if len(pool.queue) > 0 {
- t.Error("expected transaction queue to be empty. is", len(pool.queue))
- }
- pool, key = setupTxPool()
- tx1 := transaction(0, big.NewInt(100), key)
- tx2 := transaction(10, big.NewInt(100), key)
- tx3 := transaction(11, big.NewInt(100), key)
- from, _ = tx1.From()
- currentState, _ = pool.currentState()
- currentState.AddBalance(from, big.NewInt(1000))
- pool.enqueueTx(tx1.Hash(), tx1)
- pool.enqueueTx(tx2.Hash(), tx2)
- pool.enqueueTx(tx3.Hash(), tx3)
- pool.promoteExecutables()
- if len(pool.pending) != 1 {
- t.Error("expected tx pool to be 1, got", len(pool.pending))
- }
- if pool.queue[from].Len() != 2 {
- t.Error("expected len(queue) == 2, got", pool.queue[from].Len())
- }
- }
- func TestRemoveTx(t *testing.T) {
- pool, key := setupTxPool()
- tx := transaction(0, big.NewInt(100), key)
- from, _ := tx.From()
- currentState, _ := pool.currentState()
- currentState.AddBalance(from, big.NewInt(1))
- pool.enqueueTx(tx.Hash(), tx)
- pool.promoteTx(from, tx.Hash(), tx)
- if len(pool.queue) != 1 {
- t.Error("expected queue to be 1, got", len(pool.queue))
- }
- if len(pool.pending) != 1 {
- t.Error("expected pending to be 1, got", len(pool.pending))
- }
- pool.Remove(tx.Hash())
- if len(pool.queue) > 0 {
- t.Error("expected queue to be 0, got", len(pool.queue))
- }
- if len(pool.pending) > 0 {
- t.Error("expected pending to be 0, got", len(pool.pending))
- }
- }
- func TestNegativeValue(t *testing.T) {
- pool, key := setupTxPool()
- tx, _ := types.NewTransaction(0, common.Address{}, big.NewInt(-1), big.NewInt(100), big.NewInt(1), nil).SignECDSA(key)
- from, _ := tx.From()
- currentState, _ := pool.currentState()
- currentState.AddBalance(from, big.NewInt(1))
- if err := pool.Add(tx); err != ErrNegativeValue {
- t.Error("expected", ErrNegativeValue, "got", err)
- }
- }
- func TestTransactionChainFork(t *testing.T) {
- pool, key := setupTxPool()
- addr := crypto.PubkeyToAddress(key.PublicKey)
- resetState := func() {
- db, _ := ethdb.NewMemDatabase()
- statedb, _ := state.New(common.Hash{}, db)
- pool.currentState = func() (*state.StateDB, error) { return statedb, nil }
- currentState, _ := pool.currentState()
- currentState.AddBalance(addr, big.NewInt(100000000000000))
- pool.resetState()
- }
- resetState()
- tx := transaction(0, big.NewInt(100000), key)
- if err := pool.add(tx); err != nil {
- t.Error("didn't expect error", err)
- }
- pool.RemoveBatch([]*types.Transaction{tx})
- // reset the pool's internal state
- resetState()
- if err := pool.add(tx); err != nil {
- t.Error("didn't expect error", err)
- }
- }
- func TestTransactionDoubleNonce(t *testing.T) {
- pool, key := setupTxPool()
- addr := crypto.PubkeyToAddress(key.PublicKey)
- resetState := func() {
- db, _ := ethdb.NewMemDatabase()
- statedb, _ := state.New(common.Hash{}, db)
- pool.currentState = func() (*state.StateDB, error) { return statedb, nil }
- currentState, _ := pool.currentState()
- currentState.AddBalance(addr, big.NewInt(100000000000000))
- pool.resetState()
- }
- resetState()
- tx1, _ := types.NewTransaction(0, common.Address{}, big.NewInt(100), big.NewInt(100000), big.NewInt(1), nil).SignECDSA(key)
- tx2, _ := types.NewTransaction(0, common.Address{}, big.NewInt(100), big.NewInt(1000000), big.NewInt(2), nil).SignECDSA(key)
- tx3, _ := types.NewTransaction(0, common.Address{}, big.NewInt(100), big.NewInt(1000000), big.NewInt(1), nil).SignECDSA(key)
- // Add the first two transaction, ensure higher priced stays only
- if err := pool.add(tx1); err != nil {
- t.Error("didn't expect error", err)
- }
- if err := pool.add(tx2); err != nil {
- t.Error("didn't expect error", err)
- }
- pool.promoteExecutables()
- if pool.pending[addr].Len() != 1 {
- t.Error("expected 1 pending transactions, got", pool.pending[addr].Len())
- }
- if tx := pool.pending[addr].items[0]; tx.Hash() != tx2.Hash() {
- t.Errorf("transaction mismatch: have %x, want %x", tx.Hash(), tx2.Hash())
- }
- // Add the thid transaction and ensure it's not saved (smaller price)
- if err := pool.add(tx3); err != nil {
- t.Error("didn't expect error", err)
- }
- pool.promoteExecutables()
- if pool.pending[addr].Len() != 1 {
- t.Error("expected 1 pending transactions, got", pool.pending[addr].Len())
- }
- if tx := pool.pending[addr].items[0]; tx.Hash() != tx2.Hash() {
- t.Errorf("transaction mismatch: have %x, want %x", tx.Hash(), tx2.Hash())
- }
- // Ensure the total transaction count is correct
- if len(pool.all) != 1 {
- t.Error("expected 1 total transactions, got", len(pool.all))
- }
- }
- func TestMissingNonce(t *testing.T) {
- pool, key := setupTxPool()
- addr := crypto.PubkeyToAddress(key.PublicKey)
- currentState, _ := pool.currentState()
- currentState.AddBalance(addr, big.NewInt(100000000000000))
- tx := transaction(1, big.NewInt(100000), key)
- if err := pool.add(tx); err != nil {
- t.Error("didn't expect error", err)
- }
- if len(pool.pending) != 0 {
- t.Error("expected 0 pending transactions, got", len(pool.pending))
- }
- if pool.queue[addr].Len() != 1 {
- t.Error("expected 1 queued transaction, got", pool.queue[addr].Len())
- }
- if len(pool.all) != 1 {
- t.Error("expected 1 total transactions, got", len(pool.all))
- }
- }
- func TestNonceRecovery(t *testing.T) {
- const n = 10
- pool, key := setupTxPool()
- addr := crypto.PubkeyToAddress(key.PublicKey)
- currentState, _ := pool.currentState()
- currentState.SetNonce(addr, n)
- currentState.AddBalance(addr, big.NewInt(100000000000000))
- pool.resetState()
- tx := transaction(n, big.NewInt(100000), key)
- if err := pool.Add(tx); err != nil {
- t.Error(err)
- }
- // simulate some weird re-order of transactions and missing nonce(s)
- currentState.SetNonce(addr, n-1)
- pool.resetState()
- if fn := pool.pendingState.GetNonce(addr); fn != n+1 {
- t.Errorf("expected nonce to be %d, got %d", n+1, fn)
- }
- }
- func TestRemovedTxEvent(t *testing.T) {
- pool, key := setupTxPool()
- tx := transaction(0, big.NewInt(1000000), key)
- from, _ := tx.From()
- currentState, _ := pool.currentState()
- currentState.AddBalance(from, big.NewInt(1000000000000))
- pool.eventMux.Post(RemovedTransactionEvent{types.Transactions{tx}})
- pool.eventMux.Post(ChainHeadEvent{nil})
- if pool.pending[from].Len() != 1 {
- t.Error("expected 1 pending tx, got", pool.pending[from].Len())
- }
- if len(pool.all) != 1 {
- t.Error("expected 1 total transactions, got", len(pool.all))
- }
- }
- // Tests that if an account runs out of funds, any pending and queued transactions
- // are dropped.
- func TestTransactionDropping(t *testing.T) {
- // Create a test account and fund it
- pool, key := setupTxPool()
- account, _ := transaction(0, big.NewInt(0), key).From()
- state, _ := pool.currentState()
- state.AddBalance(account, big.NewInt(1000))
- // Add some pending and some queued transactions
- var (
- tx0 = transaction(0, big.NewInt(100), key)
- tx1 = transaction(1, big.NewInt(200), key)
- tx10 = transaction(10, big.NewInt(100), key)
- tx11 = transaction(11, big.NewInt(200), key)
- )
- pool.promoteTx(account, tx0.Hash(), tx0)
- pool.promoteTx(account, tx1.Hash(), tx1)
- pool.enqueueTx(tx10.Hash(), tx10)
- pool.enqueueTx(tx11.Hash(), tx11)
- // Check that pre and post validations leave the pool as is
- if pool.pending[account].Len() != 2 {
- t.Errorf("pending transaction mismatch: have %d, want %d", pool.pending[account].Len(), 2)
- }
- if pool.queue[account].Len() != 2 {
- t.Errorf("queued transaction mismatch: have %d, want %d", pool.queue[account].Len(), 2)
- }
- if len(pool.all) != 4 {
- t.Errorf("total transaction mismatch: have %d, want %d", len(pool.all), 4)
- }
- pool.resetState()
- if pool.pending[account].Len() != 2 {
- t.Errorf("pending transaction mismatch: have %d, want %d", pool.pending[account].Len(), 2)
- }
- if pool.queue[account].Len() != 2 {
- t.Errorf("queued transaction mismatch: have %d, want %d", pool.queue[account].Len(), 2)
- }
- if len(pool.all) != 4 {
- t.Errorf("total transaction mismatch: have %d, want %d", len(pool.all), 4)
- }
- // Reduce the balance of the account, and check that invalidated transactions are dropped
- state.AddBalance(account, big.NewInt(-750))
- pool.resetState()
- if _, ok := pool.pending[account].items[tx0.Nonce()]; !ok {
- t.Errorf("funded pending transaction missing: %v", tx0)
- }
- if _, ok := pool.pending[account].items[tx1.Nonce()]; ok {
- t.Errorf("out-of-fund pending transaction present: %v", tx1)
- }
- if _, ok := pool.queue[account].items[tx10.Nonce()]; !ok {
- t.Errorf("funded queued transaction missing: %v", tx10)
- }
- if _, ok := pool.queue[account].items[tx11.Nonce()]; ok {
- t.Errorf("out-of-fund queued transaction present: %v", tx11)
- }
- if len(pool.all) != 2 {
- t.Errorf("total transaction mismatch: have %d, want %d", len(pool.all), 2)
- }
- }
- // Tests that if a transaction is dropped from the current pending pool (e.g. out
- // of fund), all consecutive (still valid, but not executable) transactions are
- // postponed back into the future queue to prevent broadcasting them.
- func TestTransactionPostponing(t *testing.T) {
- // Create a test account and fund it
- pool, key := setupTxPool()
- account, _ := transaction(0, big.NewInt(0), key).From()
- state, _ := pool.currentState()
- state.AddBalance(account, big.NewInt(1000))
- // Add a batch consecutive pending transactions for validation
- txns := []*types.Transaction{}
- for i := 0; i < 100; i++ {
- var tx *types.Transaction
- if i%2 == 0 {
- tx = transaction(uint64(i), big.NewInt(100), key)
- } else {
- tx = transaction(uint64(i), big.NewInt(500), key)
- }
- pool.promoteTx(account, tx.Hash(), tx)
- txns = append(txns, tx)
- }
- // Check that pre and post validations leave the pool as is
- if pool.pending[account].Len() != len(txns) {
- t.Errorf("pending transaction mismatch: have %d, want %d", pool.pending[account].Len(), len(txns))
- }
- if len(pool.queue) != 0 {
- t.Errorf("queued transaction mismatch: have %d, want %d", pool.queue[account].Len(), 0)
- }
- if len(pool.all) != len(txns) {
- t.Errorf("total transaction mismatch: have %d, want %d", len(pool.all), len(txns))
- }
- pool.resetState()
- if pool.pending[account].Len() != len(txns) {
- t.Errorf("pending transaction mismatch: have %d, want %d", pool.pending[account].Len(), len(txns))
- }
- if len(pool.queue) != 0 {
- t.Errorf("queued transaction mismatch: have %d, want %d", pool.queue[account].Len(), 0)
- }
- if len(pool.all) != len(txns) {
- t.Errorf("total transaction mismatch: have %d, want %d", len(pool.all), len(txns))
- }
- // Reduce the balance of the account, and check that transactions are reorganised
- state.AddBalance(account, big.NewInt(-750))
- pool.resetState()
- if _, ok := pool.pending[account].items[txns[0].Nonce()]; !ok {
- t.Errorf("tx %d: valid and funded transaction missing from pending pool: %v", 0, txns[0])
- }
- if _, ok := pool.queue[account].items[txns[0].Nonce()]; ok {
- t.Errorf("tx %d: valid and funded transaction present in future queue: %v", 0, txns[0])
- }
- for i, tx := range txns[1:] {
- if i%2 == 1 {
- if _, ok := pool.pending[account].items[tx.Nonce()]; ok {
- t.Errorf("tx %d: valid but future transaction present in pending pool: %v", i+1, tx)
- }
- if _, ok := pool.queue[account].items[tx.Nonce()]; !ok {
- t.Errorf("tx %d: valid but future transaction missing from future queue: %v", i+1, tx)
- }
- } else {
- if _, ok := pool.pending[account].items[tx.Nonce()]; ok {
- t.Errorf("tx %d: out-of-fund transaction present in pending pool: %v", i+1, tx)
- }
- if _, ok := pool.queue[account].items[tx.Nonce()]; ok {
- t.Errorf("tx %d: out-of-fund transaction present in future queue: %v", i+1, tx)
- }
- }
- }
- if len(pool.all) != len(txns)/2 {
- t.Errorf("total transaction mismatch: have %d, want %d", len(pool.all), len(txns)/2)
- }
- }
- // Tests that if the transaction count belonging to a single account goes above
- // some threshold, the higher transactions are dropped to prevent DOS attacks.
- func TestTransactionQueueLimiting(t *testing.T) {
- // Create a test account and fund it
- pool, key := setupTxPool()
- account, _ := transaction(0, big.NewInt(0), key).From()
- state, _ := pool.currentState()
- state.AddBalance(account, big.NewInt(1000000))
- // Keep queuing up transactions and make sure all above a limit are dropped
- for i := uint64(1); i <= maxQueued+5; i++ {
- if err := pool.Add(transaction(i, big.NewInt(100000), key)); err != nil {
- t.Fatalf("tx %d: failed to add transaction: %v", i, err)
- }
- if len(pool.pending) != 0 {
- t.Errorf("tx %d: pending pool size mismatch: have %d, want %d", i, len(pool.pending), 0)
- }
- if i <= maxQueued {
- if pool.queue[account].Len() != int(i) {
- t.Errorf("tx %d: queue size mismatch: have %d, want %d", i, pool.queue[account].Len(), i)
- }
- } else {
- if pool.queue[account].Len() != maxQueued {
- t.Errorf("tx %d: queue limit mismatch: have %d, want %d", i, pool.queue[account].Len(), maxQueued)
- }
- }
- }
- if len(pool.all) != maxQueued {
- t.Errorf("total transaction mismatch: have %d, want %d", len(pool.all), maxQueued)
- }
- }
- // Tests that even if the transaction count belonging to a single account goes
- // above some threshold, as long as the transactions are executable, they are
- // accepted.
- func TestTransactionPendingLimiting(t *testing.T) {
- // Create a test account and fund it
- pool, key := setupTxPool()
- account, _ := transaction(0, big.NewInt(0), key).From()
- state, _ := pool.currentState()
- state.AddBalance(account, big.NewInt(1000000))
- // Keep queuing up transactions and make sure all above a limit are dropped
- for i := uint64(0); i < maxQueued+5; i++ {
- if err := pool.Add(transaction(i, big.NewInt(100000), key)); err != nil {
- t.Fatalf("tx %d: failed to add transaction: %v", i, err)
- }
- if pool.pending[account].Len() != int(i)+1 {
- t.Errorf("tx %d: pending pool size mismatch: have %d, want %d", i, pool.pending[account].Len(), i+1)
- }
- if len(pool.queue) != 0 {
- t.Errorf("tx %d: queue size mismatch: have %d, want %d", i, pool.queue[account].Len(), 0)
- }
- }
- if len(pool.all) != maxQueued+5 {
- t.Errorf("total transaction mismatch: have %d, want %d", len(pool.all), maxQueued+5)
- }
- }
- // Tests that the transaction limits are enforced the same way irrelevant whether
- // the transactions are added one by one or in batches.
- func TestTransactionQueueLimitingEquivalency(t *testing.T) { testTransactionLimitingEquivalency(t, 1) }
- func TestTransactionPendingLimitingEquivalency(t *testing.T) { testTransactionLimitingEquivalency(t, 0) }
- func testTransactionLimitingEquivalency(t *testing.T, origin uint64) {
- // Add a batch of transactions to a pool one by one
- pool1, key1 := setupTxPool()
- account1, _ := transaction(0, big.NewInt(0), key1).From()
- state1, _ := pool1.currentState()
- state1.AddBalance(account1, big.NewInt(1000000))
- for i := uint64(0); i < maxQueued+5; i++ {
- if err := pool1.Add(transaction(origin+i, big.NewInt(100000), key1)); err != nil {
- t.Fatalf("tx %d: failed to add transaction: %v", i, err)
- }
- }
- // Add a batch of transactions to a pool in one big batch
- pool2, key2 := setupTxPool()
- account2, _ := transaction(0, big.NewInt(0), key2).From()
- state2, _ := pool2.currentState()
- state2.AddBalance(account2, big.NewInt(1000000))
- txns := []*types.Transaction{}
- for i := uint64(0); i < maxQueued+5; i++ {
- txns = append(txns, transaction(origin+i, big.NewInt(100000), key2))
- }
- pool2.AddBatch(txns)
- // Ensure the batch optimization honors the same pool mechanics
- if len(pool1.pending) != len(pool2.pending) {
- t.Errorf("pending transaction count mismatch: one-by-one algo: %d, batch algo: %d", len(pool1.pending), len(pool2.pending))
- }
- if len(pool1.queue) != len(pool2.queue) {
- t.Errorf("queued transaction count mismatch: one-by-one algo: %d, batch algo: %d", len(pool1.queue), len(pool2.queue))
- }
- if len(pool1.all) != len(pool2.all) {
- t.Errorf("total transaction count mismatch: one-by-one algo %d, batch algo %d", len(pool1.all), len(pool2.all))
- }
- }
- // Benchmarks the speed of validating the contents of the pending queue of the
- // transaction pool.
- func BenchmarkPendingDemotion100(b *testing.B) { benchmarkPendingDemotion(b, 100) }
- func BenchmarkPendingDemotion1000(b *testing.B) { benchmarkPendingDemotion(b, 1000) }
- func BenchmarkPendingDemotion10000(b *testing.B) { benchmarkPendingDemotion(b, 10000) }
- func benchmarkPendingDemotion(b *testing.B, size int) {
- // Add a batch of transactions to a pool one by one
- pool, key := setupTxPool()
- account, _ := transaction(0, big.NewInt(0), key).From()
- state, _ := pool.currentState()
- state.AddBalance(account, big.NewInt(1000000))
- for i := 0; i < size; i++ {
- tx := transaction(uint64(i), big.NewInt(100000), key)
- pool.promoteTx(account, tx.Hash(), tx)
- }
- // Benchmark the speed of pool validation
- b.ResetTimer()
- for i := 0; i < b.N; i++ {
- pool.demoteUnexecutables()
- }
- }
- // Benchmarks the speed of scheduling the contents of the future queue of the
- // transaction pool.
- func BenchmarkFuturePromotion100(b *testing.B) { benchmarkFuturePromotion(b, 100) }
- func BenchmarkFuturePromotion1000(b *testing.B) { benchmarkFuturePromotion(b, 1000) }
- func BenchmarkFuturePromotion10000(b *testing.B) { benchmarkFuturePromotion(b, 10000) }
- func benchmarkFuturePromotion(b *testing.B, size int) {
- // Add a batch of transactions to a pool one by one
- pool, key := setupTxPool()
- account, _ := transaction(0, big.NewInt(0), key).From()
- state, _ := pool.currentState()
- state.AddBalance(account, big.NewInt(1000000))
- for i := 0; i < size; i++ {
- tx := transaction(uint64(1+i), big.NewInt(100000), key)
- pool.enqueueTx(tx.Hash(), tx)
- }
- // Benchmark the speed of pool validation
- b.ResetTimer()
- for i := 0; i < b.N; i++ {
- pool.promoteExecutables()
- }
- }
- // Benchmarks the speed of iterative transaction insertion.
- func BenchmarkPoolInsert(b *testing.B) {
- // Generate a batch of transactions to enqueue into the pool
- pool, key := setupTxPool()
- account, _ := transaction(0, big.NewInt(0), key).From()
- state, _ := pool.currentState()
- state.AddBalance(account, big.NewInt(1000000))
- txs := make(types.Transactions, b.N)
- for i := 0; i < b.N; i++ {
- txs[i] = transaction(uint64(i), big.NewInt(100000), key)
- }
- // Benchmark importing the transactions into the queue
- b.ResetTimer()
- for _, tx := range txs {
- pool.Add(tx)
- }
- }
- // Benchmarks the speed of batched transaction insertion.
- func BenchmarkPoolBatchInsert100(b *testing.B) { benchmarkPoolBatchInsert(b, 100) }
- func BenchmarkPoolBatchInsert1000(b *testing.B) { benchmarkPoolBatchInsert(b, 1000) }
- func BenchmarkPoolBatchInsert10000(b *testing.B) { benchmarkPoolBatchInsert(b, 10000) }
- func benchmarkPoolBatchInsert(b *testing.B, size int) {
- // Generate a batch of transactions to enqueue into the pool
- pool, key := setupTxPool()
- account, _ := transaction(0, big.NewInt(0), key).From()
- state, _ := pool.currentState()
- state.AddBalance(account, big.NewInt(1000000))
- batches := make([]types.Transactions, b.N)
- for i := 0; i < b.N; i++ {
- batches[i] = make(types.Transactions, size)
- for j := 0; j < size; j++ {
- batches[i][j] = transaction(uint64(size*i+j), big.NewInt(100000), key)
- }
- }
- // Benchmark importing the transactions into the queue
- b.ResetTimer()
- for _, batch := range batches {
- pool.AddBatch(batch)
- }
- }
|