/
mocked_module.go
41 lines (34 loc) · 1.38 KB
/
mocked_module.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
package utility
import (
"github.com/golang/mock/gomock"
"github.com/pokt-network/pocket/shared/config"
"github.com/pokt-network/pocket/shared/modules"
modulesMock "github.com/pokt-network/pocket/shared/modules/mocks"
)
var maxTxBytes = 90000
var emptyByzValidators = make([][]byte, 0)
var appHash []byte
func CreateMockedModule(_ *config.Config) (modules.UtilityModule, error) {
ctrl := gomock.NewController(nil)
utilityMock := modulesMock.NewMockUtilityModule(ctrl)
utilityContextMock := modulesMock.NewMockUtilityContext(ctrl)
persistenceContextMock := modulesMock.NewMockPersistenceContext(ctrl)
utilityMock.EXPECT().Start().Return(nil).AnyTimes()
utilityMock.EXPECT().SetBus(gomock.Any()).Do(func(modules.Bus) {}).AnyTimes()
utilityMock.EXPECT().
NewContext(gomock.Any()).
Return(utilityContextMock, nil).
AnyTimes()
utilityContextMock.EXPECT().GetPersistenceContext().Return(persistenceContextMock).AnyTimes()
utilityContextMock.EXPECT().ReleaseContext().Return().AnyTimes()
utilityContextMock.EXPECT().
GetTransactionsForProposal(gomock.Any(), maxTxBytes, gomock.AssignableToTypeOf(emptyByzValidators)).
Return(make([][]byte, 0), nil).
AnyTimes()
utilityContextMock.EXPECT().
ApplyBlock(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).
Return(appHash, nil).
AnyTimes()
persistenceContextMock.EXPECT().Commit().Return(nil).AnyTimes()
return utilityMock, nil
}