Added psudoCreateEngine function to test CreateEngine call

This commit is contained in:
philip-morlier 2023-12-01 21:06:25 -08:00
parent e591342994
commit 892f98ad6a
3 changed files with 17 additions and 0 deletions

View File

@ -17,6 +17,7 @@ var (
backend restricted.Backend backend restricted.Backend
log core.Logger log core.Logger
events core.Feed events core.Feed
createEngineCalled bool
) )
var httpApiFlagName = "http.api" var httpApiFlagName = "http.api"
@ -107,6 +108,7 @@ func (e *engine) Close() error {
} }
func CreateEngine(chainConfig *params.ChainConfig, db restricted.Database) consensus.Engine { func CreateEngine(chainConfig *params.ChainConfig, db restricted.Database) consensus.Engine {
createEngineCalled = true
return &engine{} return &engine{}
} }

View File

@ -187,6 +187,17 @@ func OpCodeSelect() []int {
return nil return nil
} }
// eth/ethconfig
func pseudoCreateEngine() {
if createEngineCalled {
m := map[string]struct{}{
"CreateEngine":struct{}{},
}
hookChan <- m
}
}
// rpc/ // rpc/
@ -239,6 +250,7 @@ func Is160(num *big.Int) bool {
} }
var plugins map[string]struct{} = map[string]struct{}{ var plugins map[string]struct{} = map[string]struct{}{
"CreateEngine":struct{}{},
"OnShutdown": struct{}{}, "OnShutdown": struct{}{},
"SetTrieFlushIntervalClone":struct{}{}, "SetTrieFlushIntervalClone":struct{}{},
"StateUpdate": struct{}{}, "StateUpdate": struct{}{},

View File

@ -53,6 +53,8 @@ func BlockChain() {
var ok bool var ok bool
f := func(key string) bool {_, ok = m[key]; return ok} f := func(key string) bool {_, ok = m[key]; return ok}
switch { switch {
case f("CreateEngine"):
delete(plugins, "CreateEngine")
case f("OnShutdown"): case f("OnShutdown"):
delete(plugins, "OnShutdown") delete(plugins, "OnShutdown")
case f("StateUpdate"): case f("StateUpdate"):
@ -139,6 +141,7 @@ func BlockChain() {
} }
}() }()
pseudoCreateEngine()
txFactory() txFactory()
txTracer() txTracer()
} }