fix rpc modules mocks

ChainSafe · Jan 9, 2023 · a5924b7 · a5924b7
1 parent 3aba5ae
commit a5924b7
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 0 deletions.
diff --git a/dot/rpc/modules/api_mocks.go b/dot/rpc/modules/api_mocks.go
@@ -10,6 +10,7 @@ import (
 	"github.com/ChainSafe/gossamer/dot/types"
 	"github.com/ChainSafe/gossamer/lib/common"
 	"github.com/ChainSafe/gossamer/lib/runtime"
+	mocksruntime "github.com/ChainSafe/gossamer/lib/runtime/mocks"
 	"github.com/stretchr/testify/mock"
 )
 
@@ -49,6 +50,8 @@ func NewMockeryBlockAPI(t *testing.T) *modulesmocks.BlockAPI {
 		Return(make([]common.Hash, 0), nil).Maybe()
 	m.On("RegisterRuntimeUpdatedChannel", mock.AnythingOfType("chan<- runtime.Version")).
 		Return(uint32(0), nil).Maybe()
+	//	GetRuntime(blockHash common.Hash) (instance state.Runtime, err error)
+	m.On("GetRuntime", mock.AnythingOfType("common.Hash")).Return(mocksruntime.NewInstance(t), nil).Maybe()
 
 	return m
 }

diff --git a/lib/runtime/mocks/instance.go b/lib/runtime/mocks/instance.go