roysc/plugeth
1
0
Fork 0
forked from cerc-io/plugeth
Code Pull Requests Activity

added meta blockProcessingError to test

Browse Source
This commit is contained in:
philip-morlier 2023-06-19 20:59:18 -07:00
parent b19d7f9f9e
commit 8a33d8e506
2 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
core/plugeth_injection_test.go
View File

@ -106,12 +106,16 @@ func TestPlugethInjections(t *testing.T) {
t.Run(fmt.Sprintf("test BlockProcessingError"), func(t *testing.T) { t.Run(fmt.Sprintf("test BlockProcessingError"), func(t *testing.T) {
called := false called := false
injectionCalled = &called injectionCalled = &called
metaInjectionCalled = &called
_, _, _, _ = sp.Process(block, statedb, vm.Config{}) _, _, _, _ = sp.Process(block, statedb, vm.Config{})
if *injectionCalled != true { if *injectionCalled != true {
t.Fatalf("pluginBlockProcessingError injection in stateProcessor.Process() not called") t.Fatalf("pluginBlockProcessingError injection in stateProcessor.Process() not called")
} }
if *metaInjectionCalled != true {
t.Fatalf("metaTracer.BlockProcessingError injection in stateProcessor.Process() not called")
}
}) })
t.Run(fmt.Sprintf("test Reorg"), func(t *testing.T) { t.Run(fmt.Sprintf("test Reorg"), func(t *testing.T) {
@ -134,17 +138,6 @@ func TestPlugethInjections(t *testing.T) {
} }
}) })
t.Run(fmt.Sprintf("test treiIntervarFlushClone"), func(t *testing.T) {
called := false
injectionCalled = &called
_ = blockchain.writeBlockWithState(block, []*types.Receipt{}, statedb)
if *injectionCalled != true {
t.Fatalf("pluginNewSideBlock injection in blockChain.writeBlockAndSetHead() not called")
}
})
t.Run(fmt.Sprintf("test NewSideBlock"), func(t *testing.T) { t.Run(fmt.Sprintf("test NewSideBlock"), func(t *testing.T) {
called := false called := false
injectionCalled = &called injectionCalled = &called

7
core/plugin_hooks.go
View File

@ -19,6 +19,7 @@ import (
) )
var injectionCalled *bool var injectionCalled *bool
var metaInjectionCalled *bool
func PluginPreProcessBlock(pl *plugins.PluginLoader, block *types.Block) { func PluginPreProcessBlock(pl *plugins.PluginLoader, block *types.Block) {
fnList := pl.Lookup("PreProcessBlock", func(item interface{}) bool { fnList := pl.Lookup("PreProcessBlock", func(item interface{}) bool {
@ -229,6 +230,12 @@ func (mt *metaTracer) PreProcessTransaction(tx *types.Transaction, block *types.
} }
} }
func (mt *metaTracer) BlockProcessingError(tx *types.Transaction, block *types.Block, err error) { func (mt *metaTracer) BlockProcessingError(tx *types.Transaction, block *types.Block, err error) {
if metaInjectionCalled != nil {
called := true
metaInjectionCalled = &called
}
if len(mt.tracers) == 0 { return } if len(mt.tracers) == 0 { return }
blockHash := core.Hash(block.Hash()) blockHash := core.Hash(block.Hash())
transactionHash := core.Hash(tx.Hash()) transactionHash := core.Hash(tx.Hash())