fixes after merge 3

multiversx · May 15, 2024 · 041f166 · 041f166
1 parent 61bb52f
commit 041f166
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/testscommon/genesisMocks/runTypeComponentsStub.go b/testscommon/genesisMocks/runTypeComponentsStub.go
@@ -14,6 +14,8 @@ import (
 	"github.com/multiversx/mx-chain-go/testscommon/enableEpochsHandlerMock"
 	"github.com/multiversx/mx-chain-go/testscommon/hashingMocks"
 	"github.com/multiversx/mx-chain-go/testscommon/marshallerMock"
+	"github.com/multiversx/mx-chain-go/testscommon/vmContext"
+	"github.com/multiversx/mx-chain-go/vm/systemSmartContracts"
 )
 
 // RunTypeComponentsStub -
@@ -24,6 +26,7 @@ type RunTypeComponentsStub struct {
 	SCProcessorFactory            scrCommon.SCProcessorCreator
 	AccountParser                 genesis.AccountsParser
 	AccountCreator                state.AccountFactory
+	VMContextCreatorHandler       systemSmartContracts.VMContextCreatorHandler
 	ShardCoordinatorFactory       sharding.ShardCoordinatorFactory
 	TxPreProcessorFactory         preprocess.TxPreProcessorCreator
 }
@@ -47,6 +50,7 @@ func NewRunTypeComponentsStub() *RunTypeComponentsStub {
 		SCProcessorFactory:            scProcessorCreator,
 		AccountParser:                 &AccountsParserStub{},
 		AccountCreator:                accountsCreator,
+		VMContextCreatorHandler:       systemSmartContracts.NewVMContextCreator(),
 		ShardCoordinatorFactory:       sharding.NewMultiShardCoordinatorFactory(),
 		TxPreProcessorFactory:         preprocess.NewTxPreProcessorCreator(),
 	}
@@ -76,6 +80,7 @@ func NewSovereignRunTypeComponentsStub() *RunTypeComponentsStub {
 		SCProcessorFactory:            scProcessorCreator,
 		AccountParser:                 &AccountsParserStub{},
 		AccountCreator:                accountsCreator,
+		VMContextCreatorHandler:       &vmContext.VMContextCreatorStub{},
 		ShardCoordinatorFactory:       sharding.NewSovereignShardCoordinatorFactory(),
 		TxPreProcessorFactory:         preprocess.NewSovereignTxPreProcessorCreator(),
 	}
@@ -111,6 +116,11 @@ func (r *RunTypeComponentsStub) AccountsCreator() state.AccountFactory {
 	return r.AccountCreator
 }
 
+// VMContextCreator -
+func (r *RunTypeComponentsStub) VMContextCreator() systemSmartContracts.VMContextCreatorHandler {
+	return r.VMContextCreatorHandler
+}
+
 // ShardCoordinatorCreator -
 func (r *RunTypeComponentsStub) ShardCoordinatorCreator() sharding.ShardCoordinatorFactory {
 	return r.ShardCoordinatorFactory