forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-07-27 15:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

core/txpool/locals: fix flaky test

Browse source
This commit is contained in:
Gary Rong 2025-03-24 21:19:59 +08:00
parent 4946b9603b
commit 277c7a0ba2
1 changed files with 24 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
core/txpool/locals/tx_tracker_test.go
View file

@ -67,16 +67,24 @@ func newTestEnv(t *testing.T, n int, gasTip uint64, journal string) *testEnv {
db := rawdb.NewMemoryDatabase()
chain, _ := core.NewBlockChain(db, nil, gspec, nil, ethash.NewFaker(), vm.Config{}, nil)
ch := make(chan core.ChainHeadEvent, 1)
sub := chain.SubscribeChainHeadEvent(ch)
defer sub.Unsubscribe()
if n, err := chain.InsertChain(blocks); err != nil {
t.Fatalf("failed to process block %d: %v", n, err)
}
for ev := range ch {
if ev.Header.Number.Uint64() == blocks[len(blocks)-1].NumberU64() {
break
}
}
legacyPool := legacypool.New(legacypool.DefaultConfig, chain)
pool, err := txpool.New(gasTip, chain, []txpool.SubPool{legacyPool})
if err != nil {
t.Fatalf("Failed to create tx pool: %v", err)
}
time.Sleep(time.Second) // hack, make sure txpool initialization is done
return &testEnv{
chain: chain,
pool: pool,
@ -107,6 +115,10 @@ func (env *testEnv) makeTx(nonce uint64, gasPrice *big.Int) *types.Transaction {
}
func (env *testEnv) commit() {
ch := make(chan core.ChainHeadEvent, 1)
sub := env.chain.SubscribeChainHeadEvent(ch)
defer sub.Unsubscribe()
head := env.chain.CurrentBlock()
block := env.chain.GetBlock(head.Hash(), head.Number.Uint64())
blocks, _ := core.GenerateChain(env.chain.Config(), block, ethash.NewFaker(), env.genDb, 1, func(i int, gen *core.BlockGen) {
@ -117,6 +129,12 @@ func (env *testEnv) commit() {
gen.AddTx(tx)
})
env.chain.InsertChain(blocks)
for ev := range ch {
if ev.Header.Number.Uint64() == blocks[len(blocks)-1].NumberU64() {
break
}
}
}
func TestRejectInvalids(t *testing.T) {
@ -127,7 +145,7 @@ func TestRejectInvalids(t *testing.T) {
gasTip uint64
tx *types.Transaction
expErr error
op func()
commit bool
}{
{
tx: env.makeTx(5, nil), // stale
@ -147,9 +165,7 @@ func TestRejectInvalids(t *testing.T) {
expErr: types.ErrInvalidSig,
},
{
op: func() {
env.commit()
},
commit: true,
tx: env.makeTx(10, nil), // stale
expErr: core.ErrNonceTooLow,
},
@ -162,8 +178,8 @@ func TestRejectInvalids(t *testing.T) {
if c.gasTip != 0 {
env.setGasTip(c.gasTip)
}
if c.op != nil {
c.op()
if c.commit {
env.commit()
}
gotErr := env.tracker.Track(c.tx)
if c.expErr == nil && gotErr != nil {